ak.jax.register_and_check

Defined in awkward.jax on line 31.

ak.jax.register_and_check()

Register Awkward Array node types with JAX’s tree mechanism.