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