jax.dlpack.is_supported_dtype

jax.dlpack.is_supported_dtype(dtype)

Check if dtype is supported by jax.dlpack.

Parameters:

dtype (DTypeLike)

Return type:

bool