jax.dtypes.issubdtype

jax.dtypes.issubdtype(a, b)

Returns True if first argument is a typecode lower/equal in type hierarchy.

This is like numpy.issubdtype(), but can handle dtype extensions such as jax.dtypes.bfloat16 and jax.dtypes.prng_key.

Parameters:

• a (DTypeLike | None)

• b (DTypeLike | None)

Return type:

bool