ak.jax._register

Defined in awkward.jax on line 90.

ak.jax._register()

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