class_docstr: A general purpose multi-layer perceptron model.
class_docstr_tok: ["A","general","purpose","multi-layer","perceptron","model","."]
code: def get_actions(
      self,
      t_states: TaskState,
      params: jnp.ndarray,
      p_states: PolicyState,
  ) -> Tuple[jnp.ndarray, PolicyState]:
      params = self._format_params_fn(params)
      return self._forward_fn(params, t_states.obs), p_states
  
file_path: evojax/evojax/policy/mlp.py
imports: import logging
  from typing import Sequence
  from typing import Tuple
  import jax
  import jax.numpy
  from jax import random
  from flax import linen
  from evojax.policy.base import PolicyNetwork
  from evojax.policy.base import PolicyState
  from evojax.task.base import TaskState
  from evojax.util import create_logger
  from evojax.util import get_params_format_fn
label_desc: Use image node_id 2 for calling the MLPPolicy obj's underlying member method code with example usage: obj.get_actions(t_states, params, p_states) and returns: self,  p_states