warp.dtype_from_jax

warp.dtype_from_jax(jax_dtype)

Return the Warp dtype corresponding to a Jax dtype.

Raises:

TypeError – Unable to find a corresponding Warp data type.