ak.from_jax
-----------

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

.. py:function:: ak.from_jax(array, regulararray=False, highlevel=True, behavior=None)


    :param array: The ``jax.numpy.array`` to convert into an Awkward Array.
    :type array: jax.numpy.array
    :param regulararray: If True and the array is multidimensional,
                     the dimensions are represented by nested :py:obj:`ak.layout.RegularArray`
                     nodes; if False and the array is multidimensional, the dimensions
                     are represented by a multivalued :py:obj:`ak.layout.NumpyArray.shape`.
                     If the array is one-dimensional, this has no effect.
    :type regulararray: bool
    :param highlevel: If True, return an :py:obj:`ak.Array`; otherwise, return
                  a low-level :py:obj:`ak.layout.Content` subclass.
    :type highlevel: bool
    :param behavior: Custom :py:obj:`ak.behavior` for the output array, if
                 high-level.
    :type behavior: None or dict

Converts a JAX array into an Awkward Array.

The resulting layout may involve the following :py:obj:`ak.layout.Content` types
(only):

   * :py:obj:`ak.layout.NumpyArray`
   * :py:obj:`ak.layout.RegularArray` if ``regulararray=True``.

See also :py:obj:`ak.from_cupy` and :py:obj:`ak.from_numpy`.

