jax.dlpack module

from_dlpack(external_array[, device, copy])

Returns a Array representation of a DLPack tensor.