.. module:: thunder.core.pytree

PyTree
------

.. list-table:: pytree API map
   :widths: 50 50
   :header-rows: 1

   * - thunder.core.pytree
     - pytorch pytree
   * - :func:`tree_map`
     - See https://optree.readthedocs.io/en/latest/ops.html#optree.tree_map
   * - :func:`tree_flatten`
     - See https://optree.readthedocs.io/en/latest/ops.html#optree.tree_flatten
   * - :func:`tree_unflatten`
     - See https://optree.readthedocs.io/en/latest/ops.html#optree.tree_unflatten