sequince.utils.pytrees
======================

.. py:module:: sequince.utils.pytrees


Attributes
----------

.. autoapisummary::

   sequince.utils.pytrees.OPTREE_NAMESPACE


Functions
---------

.. autoapisummary::

   sequince.utils.pytrees.pytree_dataclass


Module Contents
---------------

.. py:data:: OPTREE_NAMESPACE
   :value: 'sequince'


.. py:function:: pytree_dataclass(*, data_fields, meta_fields=(), drop_fields=())

   Register a dataclass as an Optree and, when available, JAX pytree.