code: def dqn_atari_network(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
  
imports: from typing import Any, Optional, Sequence, Tuple
  from acme.jax.networks import base
  from acme.jax.networks import duelling
  from acme.jax.networks import embedding
  from acme.jax.networks import policy_value
  from acme.jax.networks import resnet
  from acme.wrappers import observation_action_reward
  import haiku
  import jax
  import jax.numpy
label_desc: Use image node_id 1 for calling a global function with example usage: dqn_atari_network(num_actions) and returns: network, model