- ak.from_jax(array, *, regulararray=False, highlevel=True, behavior=None)#
array (jax.numpy.DeviceArray) – The JAX DeviceArray to convert into an Awkward Array.
regulararray (bool) – If True and the array is multidimensional, the dimensions are represented by nested
ak.contents.RegularArraynodes; if False and the array is multidimensional, the dimensions are represented by a multivalued
ak.contents.NumpyArray.shape. If the array is one-dimensional, this has no effect.
Converts a JAX DeviceArray array into an Awkward Array.
The resulting layout may involve the following