Update flax.optim.rst - SemanticDiff

Commit

4 years ago

Update flax.optim.rst Update documentation for AdaBelief optimizer, which was recently added to the codebase.