optax
optax copied to clipboard
Remove multi_normal from docs (in utilities.rst) or add dosctring to multi_normal (in utils.py).