ak.jax._register
================

Defined in `awkward.jax <https://github.com/scikit-hep/awkward/blob/80c3e39415a29a29ae2ffcc089f09e6de9e17fa2/src/awkward/jax.py>`__ on `line 90 <https://github.com/scikit-hep/awkward/blob/80c3e39415a29a29ae2ffcc089f09e6de9e17fa2/src/awkward/jax.py#L90>`__.

.. py:function:: ak.jax._register()

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