.. _PyTree Node Registration:

PyTree Node Registration
========================

.. currentmodule:: optree

.. autosummary::

    register_pytree_node
    register_pytree_node_class
    unregister_pytree_node

.. autofunction:: register_pytree_node
.. autofunction:: register_pytree_node_class
.. autofunction:: unregister_pytree_node