Improve doc-strings and formatting of popular optimiser instances - Part 1
Improve doc-strings and formatting of popular optimiser instances - Part 1

PiperOrigin-RevId: 364776564

deepmind/optax,deepmind/optax