Flax is a neural network ecosystem for JAX that is designed for flexibility.
Latest commits.
Builders behind this project.
