code: def check_valid_dtype(dtype: DType) -> None:
      if dtype not in _jax_dtype_set:
          raise TypeError(
              f"Dtype {dtype} is not a valid JAX array "
              "type. Only arrays of numeric types are supported by JAX."
          )
  
imports: from __future__ import annotations
  import abc
  import builtins
  import functools
  from typing import cast, overload, Any, Literal, Union
  import warnings
  import ml_dtypes
  import numpy
  from jax._src import config
  from jax._src.typing import DType, DTypeLike
  from jax._src.util import set_module
  from jax._src import traceback_util
label_desc: Use image node_id 22 for calling a global function with example usage: check_valid_dtype(dtype) without return types