ak.jax.register_behavior_class
------------------------------

.. py:module: ak.jax.register_behavior_class

Defined in `awkward.jax <https://github.com/scikit-hep/awkward/blob/36da52cfa8846355c390beb6555eac1d31c27c26/src/awkward/jax.py>`__ on `line 64 <https://github.com/scikit-hep/awkward/blob/36da52cfa8846355c390beb6555eac1d31c27c26/src/awkward/jax.py#L64>`__.

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


    :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.