google-deepmind/optax

@google-deepmind

Optax is a gradient processing and optimization library for JAX.

↗ Open on GitHub

Stars

2,329

Forks

363

Language

Python

License

Apache-2.0

Last push

1 week ago

Python machine-learning optimization

No related intel yet

This repo has not appeared in any of the sources the radar tracks. The collector runs on a schedule — check back once it covers this repo.