code: def get_initial_params(
      model: models.Seq2seq, rng: PRNGKey, ctable: CTable
  ) -> Dict[str, Any]:
      """Returns the initial parameters of a seq2seq model."""
      rng1, rng2 = jax.random.split(rng)
      variables = model.init(
          {"params": rng1, "lstm": rng2},
          jnp.ones(ctable.encoder_input_shape, jnp.float32),
          jnp.ones(ctable.decoder_input_shape, jnp.float32),
      )
      return variables["params"]
  
imports: import functools
  from typing import Any, Dict, Tuple
  from absl import app
  from absl import flags
  from absl import logging
  from clu import metric_writers
  from flax import linen
  from flax.training import train_state
  import jax
  import jax.numpy
  import optax
  import models
  from input_pipeline import CharacterTable
  from input_pipeline import get_sequence_lengths
  from input_pipeline import mask_sequences
label_desc: Use image node_id 2 for calling a global function with example usage: get_initial_params(model, rng, ctable) and returns: variables