optax.with\_extra\_args\_support
================================

.. currentmodule:: optax

.. autofunction:: with_extra_args_support