jax.dlpack module# from_dlpack(external_array) Returns a Array representation of a DLPack tensor. to_dlpack(x[, take_ownership, stream]) Returns a DLPack tensor that encapsulates a Array x.