code: def _infer_dtype(dtype: jnp.dtype):
      default_dtype = ivy.infer_default_dtype(dtype)
      if ivy.dtype_bits(dtype) < ivy.dtype_bits(default_dtype):
          return default_dtype
      return dtype
  
file_path: ivy/ivy/functional/backends/jax/statistical.py
imports: import jax.numpy
  from typing import Union, Optional, Sequence
  import ivy
  from ivy.func_wrapper import with_unsupported_dtypes
  from ivy.functional.backends.jax import JaxArray
  from .None import backend_version
label_desc: Use image node_id 4 for calling a global function with example usage: _infer_dtype(dtype) and returns: dtype, default_dtype