``jax.tree`` module
===================

.. currentmodule:: jax.tree

.. automodule:: jax.tree

List of Functions
-----------------

.. autosummary::
   :toctree: _autosummary

   all
   flatten
   flatten_with_path
   leaves
   leaves_with_path
   map
   map_with_path
   reduce
   structure
   transpose
   unflatten