CMD + K
conda-forge/optax

optax

Community

A gradient processing and optimisation library in JAX.

Copied fromcf-post-staging / optax

Installation

To install this package, run one of the following:

Conda
$conda install conda-forge::optax

Usage Tracking

0.2.8
0.2.7
0.2.6
0.2.5
0.2.4
5 / 8 versions selected
Downloads (Last 6 months): 0

About

Summary

A gradient processing and optimisation library in JAX.

Last Updated

Mar 21, 2026 at 03:03

License

Apache-2.0

Supported Platforms

noarch

Home

https://github.com/deepmind/optax