optax 0.2.2

A gradient processing and optimisation library in JAX.

Stars: 1487, Watchers: 1487, Forks: 151, Open Issues: 55

The google-deepmind/optax repo was created 3 years ago and the last code push was Yesterday.
The project is very popular with an impressive 1487 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 55 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