Last released Jul 23, 2023
Fine-grained, dynamic control of neural network topology in JAX.
Supported by