ak.jax.register_and_check
-------------------------

.. py:module: ak.jax.register_and_check

Defined in `awkward.jax <https://github.com/scikit-hep/awkward/blob/e9dd2694cf5c6ccb1717358fe926816d8ecbfb9c/src/awkward/jax.py>`__ on `line 31 <https://github.com/scikit-hep/awkward/blob/e9dd2694cf5c6ccb1717358fe926816d8ecbfb9c/src/awkward/jax.py#L31>`__.

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

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