ak.jax.assert_registered
------------------------

.. py:module: ak.jax.assert_registered

Defined in `awkward.jax <https://github.com/scikit-hep/awkward/blob/ecb41df1a0678f2c38387f6f91203f0122fd0906/src/awkward/jax.py>`__ on `line 118 <https://github.com/scikit-hep/awkward/blob/ecb41df1a0678f2c38387f6f91203f0122fd0906/src/awkward/jax.py#L118>`__.

.. py:function:: ak.jax.assert_registered()

Ensure that JAX integration is registered. Raise a RuntimeError if not.