def METHOD_NAME(num_actions: int) -> base.QNetwork:
  """A feed-forward network for use with Ape-X DQN."""
  def network(inputs: Images) -> base.QValues:
    model = hk.Sequential([
        AtariTorso(),
        duelling.DuellingMLP(num_actions, hidden_sizes=[512]),
    ])
    return model(inputs)
  return network