Flax is a neural network library for JAX that is designed for flexibility.
Repository Index
47,133 repositories indexed
1–1 of 1 · Most stars
47,133 repositories indexed
1–1 of 1 · Most stars
Flax is a neural network library for JAX that is designed for flexibility.