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
0 / 8 versions selected
Downloads (Last 6 months): 0
About
Summary
Flax: A neural network library for JAX designed for flexibility
Last Updated
Jan 14, 2026 at 18:59
License
Apache-2.0
Total Downloads
0
GitHub Repository
https://github.com/google/flaxDocumentation
http://flax.readthedocs.io