Last released Jun 29, 2026
Fine-grained, dynamic control of neural network topology in JAX.
Supported by