flax
Flax: A neural network library for JAX designed for flexibility
Installation
To install this package, run one of the following:
Usage Tracking
0.10.4
0.10.3
0.10.2
0.10.1
0.10.0
5 / 8 versions selected
Downloads (Last 6 months): 0
About
Summary
Flax: A neural network library for JAX designed for flexibility
Last Updated
Oct 27, 2025 at 15:24
License
Apache-2.0
Total Downloads
143.3K
Supported Platforms
noarch
GitHub Repository
https://github.com/google/flaxDocumentation
http://flax.readthedocs.io