optax
A gradient processing and optimisation library in JAX.
Installation
To install this package, run one of the following:
Usage Tracking
0.2.7
0.2.6
0.2.5
0.2.4
0.2.3
5 / 8 versions selected
Downloads (Last 6 months): 0
About
Summary
A gradient processing and optimisation library in JAX.
Last Updated
Feb 6, 2026 at 01:15
License
Apache-2.0
Total Downloads
178.1K
Version Downloads
3.2K
Supported Platforms
noarch