optax

Community

A gradient processing and optimisation library in JAX.

Versions

To install this package, run one of the following:

Installation commands are not available for this package.

Version

1 / 8 versions selected

Downloads (Last 6 months): 0

Summary

A gradient processing and optimisation library in JAX.

Last Updated

Mar 21, 2026 at 03:03

License

Apache-2.0

Home