jax.tree\_util.tree\_map\_with\_path
====================================

.. currentmodule:: jax.tree_util

.. autofunction:: tree_map_with_path