Last released Aug 27, 2024
Flax: A neural network library for JAX designed for flexibility
Supported by