jax.dtypes.canonicalize_dtype# jax.dtypes.canonicalize_dtype(dtype, allow_extended_dtype=False, allow_opaque_dtype=None)# Convert from a dtype to a canonical dtype based on config.x64_enabled. Parameters: dtype (Any) – allow_extended_dtype (bool) – allow_opaque_dtype (Any) – Return type: Union[dtype, ExtendedDType]