ak.to_jax
---------

Defined in `awkward.operations.convert <https://github.com/scikit-hep/awkward-1.0/blob/80bbef0738a6b7928333d7c705ee1b359991de5b/src/awkward/operations/convert.py>`__ on `line 595 <https://github.com/scikit-hep/awkward-1.0/blob/80bbef0738a6b7928333d7c705ee1b359991de5b/src/awkward/operations/convert.py#L595>`__.

.. py:function:: ak.to_jax(array)

Converts ``array`` (many types supported) into a JAX array, if possible.

If the data are numerical and regular (nested lists have equal lengths
in each dimension, as described by the #type), they can be losslessly
converted to a CuPy array and this function returns without an error.

Otherwise, the function raises an error.

If ``array`` is a scalar, it is converted into a JAX scalar.

See also :py:obj:`ak.to_cupy`, :py:obj:`ak.from_jax` and :py:obj:`ak.to_numpy`.

