search

google-deepmind / optax

Optax is a gradient processing and optimization library for JAX.
https://optax.readthedocs.io
Apache License 2.0
1.68k stars 190 forks source link

Add missing examples to contrib examples section. #794

Closed fabianp closed 8 months ago

fabianp commented 8 months ago

Also add flax as dependency to examples, which should fix the error in

https://optax.readthedocs.io/en/latest/_collections/examples/mlp_mnist.html

remove dangling comment

Deepmind -> Google Deepmind

remove dangling spaces