search

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

Move linear algebra operations into utilities #831

Closed fabianp closed 5 months ago

fabianp commented 6 months ago

In the Reference section, move https://optax.readthedocs.io/en/latest/api/linear_algebra_operators.html into https://optax.readthedocs.io/en/latest/api/utilities.html