code: def optimiser() -> optax.GradientTransformation:
      return optax.chain(
          optax.clip_by_global_norm(GRAD_CLIP_VALUE),
          optax.adam(LEARNING_RATE, b1=0.9, b2=0.99),
      )
  
imports: from collections.abc import MutableMapping
  import time
  from typing import Any, NamedTuple, Union
  from absl import app
  from absl import flags
  from absl import logging
  import haiku
  from examples.transformer import dataset
  from examples.transformer import model
  import jax
  import jax.numpy
  import numpy
  import optax
label_desc: Use image node_id 2 for calling a global function with example usage: optimiser() and returns: optax