code: def compute_eps(params: Any) -> float:
      return jax.tree_util.tree_reduce(
          lambda p1, p2: max(_get_eps(p1), _get_eps(p2)), params
      )
  
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 2 for calling a global function with example usage: compute_eps(params) and returns: jax