Optax

https://github.com/google-deepmind/optax

Optax is a gradient processing and optimization library for JAX, designed to facilitate research by...

Tokens:207,588
Snippets:1,759
Trust Score:7.9
License:Apache-2.0
Update:2 weeks ago
Tokens:
Raw