optax.stateless\_with\_tree\_map
================================

.. currentmodule:: optax

.. autofunction:: stateless_with_tree_map