ak.jax.import_jax

Defined in awkward.jax on line 135.

ak.jax.import_jax()

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