def METHOD_NAME(
    params: networks_lib.Params, observations: networks_lib.Observation,
    initial_state: networks_lib.RecurrentState
) -> Tuple[networks_lib.NetworkOutput, networks_lib.RecurrentState]:
  unused_rng = jax.random.PRNGKey(0)
  return networks.unroll(params, unused_rng, observations, initial_state)