code: def _scenic_optimizer_args_to_optax_args(
      config: ml_collections.ConfigDict,
  ) -> ml_collections.ConfigDict:
      """Transform original scenic arguments to optax arguments."""
      if "beta1" in config:
          config.b1 = config.beta1
          del config.beta1
      if "beta2" in config:
          config.b2 = config.beta2
          del config.beta2
      if "epsilon" in config:
          config.eps = config.epsilon
          del config.epsilon
      return config
  
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 5 for calling a global function with example usage: _scenic_optimizer_args_to_optax_args(config) and returns: config