Package Usage: pypi: optax
A gradient processing and optimisation library in JAX.
21 versions
Latest release: about 1 year ago
227 dependent packages
913,485 downloads last month
View more package details: https://packages.ecosystem.code.gouv.fr/registries/pypi.org/packages/optax