ak.jax.import_jax#

Defined in awkward.jax on line 140.

ak.jax.import_jax()#

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