code: def dtype_max_value(dtype):
      if dtype.kind == "f":
          return jnp.inf
      elif dtype.kind == "i":
          return jnp.iinfo(dtype).max
      elif dtype.kind == "b":
          return True
      else:
          raise ValueError(f"Invalid data type {dtype.kind!r}.")
  
imports: import functools
  from typing import Any, Callable, Generator, Iterable, Iterator, List, Mapping, Optional, Sequence, Union
  import jax
  from jax import lax
  import jax.numpy
  import jax.tree_util
  from jraph._src import graph
  import numpy
label_desc: Use image node_id 28 for calling a global function with example usage: dtype_max_value(dtype) and returns: jnp, jnp, True