class_docstr: Simple convolutional stack commonly used for Atari.
code: def __init__(self):
      super().__init__(name="atari_torso")
      self._network = hk.Sequential(
          [
              hk.Conv2D(32, [8, 8], 4),
              jax.nn.relu,
              hk.Conv2D(64, [4, 4], 2),
              jax.nn.relu,
              hk.Conv2D(64, [3, 3], 1),
              jax.nn.relu,
          ]
      )
  
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 to create a new AtariTorso object from inherited base classes: hk with example: obj = AtariTorso()