treevalue.tree.integration
=======================================

.. py:currentmodule:: treevalue.tree.integration


.. _apidoc_tree_integration_register_for_jax:

register_for_jax
------------------------

.. autofunction:: register_for_jax