ak.jax.register_and_check#

Defined in awkward.jax on line 30.

ak.jax.register_and_check()#

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