integration.rst

Integration with Third-Party Libraries

Integration for JAX

.. currentmodule:: optree.integration.jax

.. autosummary::

    tree_ravel

.. autofunction:: tree_ravel

---

Integration for NumPy

.. currentmodule:: optree.integration.numpy

.. autosummary::

    tree_ravel

.. autofunction:: tree_ravel

---

Integration for PyTorch

.. currentmodule:: optree.integration.torch

.. autosummary::

    tree_ravel

.. autofunction:: tree_ravel