ak.jax.import_jax
=================

Defined in `awkward.jax <https://github.com/scikit-hep/awkward/blob/f5805c67847ec9574491cfb8976cb800b685b3ad/src/awkward/jax.py>`__ on `line 161 <https://github.com/scikit-hep/awkward/blob/f5805c67847ec9574491cfb8976cb800b685b3ad/src/awkward/jax.py#L161>`__.

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

   Ensure that JAX integration is registered, and return the JAX module. Raise a RuntimeError if not.