ak.jax.register_and_check
=========================

Defined in `awkward.jax <https://github.com/scikit-hep/awkward/blob/5aaab0916caf6d2910fdd118ae61b148aa2e0869/src/awkward/jax.py>`__ on `line 32 <https://github.com/scikit-hep/awkward/blob/5aaab0916caf6d2910fdd118ae61b148aa2e0869/src/awkward/jax.py#L32>`__.

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

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