optax.adam
==========

.. currentmodule:: optax

.. autofunction:: adam