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