ak.jax.register_and_check

Defined in awkward.jax on line 28.

ak.jax.register_and_check()

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