flax
Flax: A neural network library for JAX designed for flexibility
Installation
To install this package, run one of the following:
Installation commands are not available for this package.
Usage Tracking
latest
1 / 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
GitHub Repository
https://github.com/google/flaxDocumentation
http://flax.readthedocs.io