Repository info

flax

• Summary: a summary of flax
• Description: Flax is a neural network library for JAX that is designed for flexibility.
• Stars: 6678
• Number of forks: 718