class_docstr: A simple network.
class_docstr_tok: ["A","simple","network","."]
code: def __init__(self, num_actions: int):
      super().__init__()
      self._num_actions = num_actions
  
imports: import functools
  import queue
  import threading
  from typing import Any, Callable, NamedTuple
  from absl import app
  from absl import logging
  from bsuite.environments import catch
  import dm_env
  import haiku
  import jax
  import jax.numpy
  import numpy
  import optax
  import rlax
label_desc: Use image node_id 1 to create a new SimpleNet object from inherited base classes: hk with example: obj = SimpleNet(num_actions)