issues
search
google-deepmind
/
optax
Optax is a gradient processing and optimization library for JAX.
https://optax.readthedocs.io
Apache License 2.0
1.65k
stars
181
forks
source link
docstring cosmetic improvements
#879
Closed
fabianp
closed
6 months ago