code: def _get_eps(p: jnp.ndarray) -> float:
      dtype = p.dtype
      if jnp.issubdtype(dtype, jnp.inexact):
          return jnp.finfo(dtype).eps
      else:
          return 0
  
imports: import copy
  import dataclasses
  import re
  from typing import Any, Callable, Generator, List, Optional, Tuple, Union
  from absl import logging
  import flax
  import jax
  import jax.numpy
  import ml_collections
  import numpy
  import optax
label_desc: Use image node_id 1 for calling a global function with example usage: _get_eps(p) and returns: jnp, int