CMD + K

optax

Community

A gradient processing and optimisation library in JAX.

Installation

To install this package, run one of the following:

Conda
$conda install conda-forge::optax

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