optax 0.2.3

A gradient processing and optimisation library in JAX.

Stars: 1657, Watchers: 1657, Forks: 183, Open Issues: 51

The google-deepmind/optax repo was created 4 years ago and the last code push was 2 days ago.
The project is very popular with an impressive 1657 github stars!

How to Install optax

You can install optax using pip

pip install optax

or add it to a project with poetry

poetry add optax

Package Details

Author

None

License

None

Homepage

None

PyPi:

https://pypi.org/project/optax/

GitHub Repo:

https://github.com/deepmind/optax

Classifiers

• Scientific/Engineering/Artificial Intelligence
• Software Development/Libraries/Python Modules

Related Packages

Errors

A list of common optax errors.

Code Examples

Here are some optax code examples and snippets.

GitHub Issues

The optax package has 51 open issues on GitHub

• Using optax with partially complex models
• Bump ipython from 7.16.1 to 7.16.3 in /requirements
• Clarifies optax.adamw(mask) parameter.
• add_noise scales by variance, not its root
• Implement complex norm in optimizers
• Fix EMA bias correction
• Nesting 'masked' wrappers gives a confusing error
• Add online newton optimizer
• Missing type hints
• [RFC] Proposal for complex-valued optimization in Optax

See more issues on GitHub