optax.snapshot
==============

.. currentmodule:: optax

.. autofunction:: snapshot