warmup_constant_schedule was defined but not displayed on the documentation
added mathematical formula for polynomial_schedule , and fixed some RST display issues, where the second doctest was not displaying properly due to missing empty line.
add more precise documentation to linear_onecycle_schedule
Ran format document which fixed a bunch of minor formatting.
misc improvements to optax's documentation
warmup_constant_schedulewas defined but not displayed on the documentationadded mathematical formula for
polynomial_schedule, and fixed some RST display issues, where the second doctest was not displaying properly due to missing empty line.add more precise documentation to
linear_onecycle_scheduleRan
format documentwhich fixed a bunch of minor formatting.