jax.tree\_util.register\_pytree\_with\_keys\_class
==================================================

.. currentmodule:: jax.tree_util

.. autofunction:: register_pytree_with_keys_class