jax.dtypes.scalar_type_of

jax.dtypes.scalar_type_of(x)

Return the scalar type associated with a JAX value.

Parameters:

x (Any)

Return type:

type