ak.jax.register_behavior_class ------------------------------ .. py:module: ak.jax.register_behavior_class Defined in `awkward.jax `__ on `line 61 `__. .. 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.