ak.jax.register_behavior_class
==============================

Defined in `awkward.jax <https://github.com/scikit-hep/awkward/blob/08439425ed1a9162f8c0eb6ee8d33df375616b71/src/awkward/jax.py>`__ on `line 87 <https://github.com/scikit-hep/awkward/blob/08439425ed1a9162f8c0eb6ee8d33df375616b71/src/awkward/jax.py#L87>`__.

.. py:function:: ak.jax.register_behavior_class(cls: HighLevelType)

   :param cls: behavior class to register with JAX

   Register the behavior class with JAX, if JAX integration is enabled. Otherwise,
   queue the type for subsequent registration when/if JAX is registered.