flax
Flax: A neural network library for JAX designed for flexibility
Installation
To install this package, run one of the following:
Usage Tracking
0.12.6
0.12.5
0.12.4
0.12.3
0.12.2
5 / 8 versions selected
Downloads (Last 6 months): 0
About
Summary
Flax: A neural network library for JAX designed for flexibility
Last Updated
Mar 26, 2026 at 20:29
License
Apache-2.0
Supported Platforms
noarch
GitHub Repository
https://github.com/google/flaxDocumentation
http://flax.readthedocs.io/