Moved linear algebra operations into utilities

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

Closed hmludwig closed 6 months ago

hmludwig commented 6 months ago

See #831

fabianp commented 6 months ago

looks great to me, thanks @hmludwig !

vroulet commented 6 months ago

I would either

Currently the doc of multi_normal is uninformative. I let you choose!

fabianp commented 6 months ago

huh, I hadn't realized about this function. I wonder why we need the function multi_normal in the first place

vroulet commented 6 months ago

I let it merged and opened a different issue concerning multi_normal. Thank you @hmludwig for this PR!