google / flax
Flax is a neural network library for JAX that is designed for flexibility.
- Repository stars
- 7,365All-time #7,772
- Fork
- 849
- Last push
- updated 10 days ago
- Repository created
- since 2020
About
Summarised by AI from the README; the original prevails
Author description: Flax is a neural network library for JAX that is designed for flexibility.
Basic listing based on official public information; functionality has not been independently evaluated.
- Platform
- Not verified
- Pricing
- Pricing not verifiedCheck current official software, hosting, model and infrastructure pricing.
- Maintenance notes
- Not verified