jax.dlpack 模块

from_dlpack(external_array[, device, copy])	返回 DLPack 张量的 Array 表示。
is_supported_dtype(dtype)	检查 dtype 是否被 jax.dlpack 支持。