Documentation about optax HOT 6 CLOSED

We now have documentation on read the docs!
https://optax.readthedocs.io/en/latest/

from optax.

Hi, thanks for reaching out.

Yes I can confirm that 'optax' is the supported solution going forward. It is already widely used internally at DeepMind and I encourage you to also use 'optax' rather than 'optix', as the latter will be deprecated soon.

We will be adding more public documentation for 'optax' in the coming weeks.

from optax.

I just wanted to chime in and say that this would be nice to have. I'm working on a project that used to reference some optix functions through intersphinx.. now I don't know where to point it to.

from optax.

Thanks! Will update Elegy then :D

from optax.

switched. nice library!

from optax.

Thanks @mtthss

It's so nice to walk through the API without being distracted by the code itself. (I just saw periodic-update for the first time, awesome!)

One minor thing.. Some of the sections aren't showing up in the correct hierarchy in the sidebar. For instance, "Apply Updates" is listed as a subsubsection (under its predecessor item).

I changed some ---- sections to ==== in #79

from optax.