google-deepmind / optax

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

Remove multi_normal from docs (in utilities.rst) or add dosctring to multi_normal (in utils.py). #842

Closed vroulet closed 1 month ago