optax.microbatching.micro\_grad
===============================

.. currentmodule:: optax.microbatching

.. autofunction:: micro_grad