ak.jax.assert_registered ======================== Defined in `awkward.jax `__ on `line 144 `__. .. py:function:: ak.jax.assert_registered() Ensure that JAX integration is registered. Raise a RuntimeError if not.