jax.dtypes.scalar_type_of# jax.dtypes.scalar_type_of(x)[source]# Return the scalar type associated with a JAX value. Parameters: x (Any) Return type: type
