jax

f4aa5150 - Move internal type-related functions into a new (internal) jax.types … (#1695)

Move internal type-related functions into a new (internal) jax.types … (#1695) * Move internal type-related functions into a new (internal) jax.types module. Avoid calling onp type functions in lieu of the wrappers in jax.types. Currently these do the same thing, but future changes will make the behavior of the jax type functions diverge from the classic NumPy versions in some cases. Move xla_bridge.canonicalize_dtype into jax.types, since it fits there more naturally. * Rename jax.types to jax.dtypes. * s/types/dtypes/ in tests.