Added mathematical description to AdaBelief docstring

google-deepmind / optax

Optax is a gradient processing and optimization library for JAX.

https://optax.readthedocs.io

Apache License 2.0

1.57k stars 169 forks source link

Closed hmludwig closed 4 months ago

hmludwig commented 4 months ago

I took the Adam docstring and adjusted it to match the implementation of AdaBelief.

The resulting doc entry