# {py:mod}`evorl.algorithms.meta.pbt_sac.param_sac`

```{py:module} evorl.algorithms.meta.pbt_sac.param_sac
```

```{autodoc2-docstring} evorl.algorithms.meta.pbt_sac.param_sac
:parser: autodoc2_docstrings_parser
:allowtitles:
```

## Module Contents

### Classes

````{list-table}
:class: autosummary longtable
:align: left

* - {py:obj}`ParamSACAgent <evorl.algorithms.meta.pbt_sac.param_sac.ParamSACAgent>`
  - ```{autodoc2-docstring} evorl.algorithms.meta.pbt_sac.param_sac.ParamSACAgent
    :parser: autodoc2_docstrings_parser
    :summary:
    ```
* - {py:obj}`ParamSACTrainMetric <evorl.algorithms.meta.pbt_sac.param_sac.ParamSACTrainMetric>`
  -
* - {py:obj}`ParamSACWorkflow <evorl.algorithms.meta.pbt_sac.param_sac.ParamSACWorkflow>`
  - ```{autodoc2-docstring} evorl.algorithms.meta.pbt_sac.param_sac.ParamSACWorkflow
    :parser: autodoc2_docstrings_parser
    :summary:
    ```
````

### Functions

````{list-table}
:class: autosummary longtable
:align: left

* - {py:obj}`make_mlp_sac_agent <evorl.algorithms.meta.pbt_sac.param_sac.make_mlp_sac_agent>`
  - ```{autodoc2-docstring} evorl.algorithms.meta.pbt_sac.param_sac.make_mlp_sac_agent
    :parser: autodoc2_docstrings_parser
    :summary:
    ```
````

### API

`````{py:class} ParamSACAgent
:canonical: evorl.algorithms.meta.pbt_sac.param_sac.ParamSACAgent

Bases: {py:obj}`evorl.algorithms.sac.SACAgent`

```{autodoc2-docstring} evorl.algorithms.meta.pbt_sac.param_sac.ParamSACAgent
:parser: autodoc2_docstrings_parser
```

````{py:method} critic_loss(agent_state: evorl.agent.AgentState, sample_batch: evorl.sample_batch.SampleBatch, key: chex.PRNGKey) -> evorl.types.LossDict
:canonical: evorl.algorithms.meta.pbt_sac.param_sac.ParamSACAgent.critic_loss

```{autodoc2-docstring} evorl.algorithms.meta.pbt_sac.param_sac.ParamSACAgent.critic_loss
:parser: autodoc2_docstrings_parser
```

````

````{py:method} init(obs_space: evorl.envs.Space, action_space: evorl.envs.Space, key: chex.PRNGKey) -> evorl.agent.AgentState
:canonical: evorl.algorithms.meta.pbt_sac.param_sac.ParamSACAgent.init

```{autodoc2-docstring} evorl.algorithms.meta.pbt_sac.param_sac.ParamSACAgent.init
:parser: autodoc2_docstrings_parser
```

````

`````

`````{py:class} ParamSACTrainMetric
:canonical: evorl.algorithms.meta.pbt_sac.param_sac.ParamSACTrainMetric

Bases: {py:obj}`evorl.algorithms.sac.SACTrainMetric`

````{py:attribute} trajectory
:canonical: evorl.algorithms.meta.pbt_sac.param_sac.ParamSACTrainMetric.trajectory
:type: evorl.sample_batch.SampleBatch
:value: >
   None

```{autodoc2-docstring} evorl.algorithms.meta.pbt_sac.param_sac.ParamSACTrainMetric.trajectory
:parser: autodoc2_docstrings_parser
```

````

`````

`````{py:class} ParamSACWorkflow(env: evorl.envs.Env, agent: evorl.agent.Agent, optimizer: optax.GradientTransformation, evaluator: evorl.evaluators.Evaluator, replay_buffer: evorl.replay_buffers.AbstractReplayBuffer, config: omegaconf.DictConfig)
:canonical: evorl.algorithms.meta.pbt_sac.param_sac.ParamSACWorkflow

Bases: {py:obj}`evorl.algorithms.offpolicy_utils.OffPolicyWorkflowTemplate`

```{autodoc2-docstring} evorl.algorithms.meta.pbt_sac.param_sac.ParamSACWorkflow
:parser: autodoc2_docstrings_parser
```

````{py:method} name()
:canonical: evorl.algorithms.meta.pbt_sac.param_sac.ParamSACWorkflow.name
:classmethod:

````

````{py:method} setup(key: chex.PRNGKey) -> evorl.types.State
:canonical: evorl.algorithms.meta.pbt_sac.param_sac.ParamSACWorkflow.setup

```{autodoc2-docstring} evorl.algorithms.meta.pbt_sac.param_sac.ParamSACWorkflow.setup
:parser: autodoc2_docstrings_parser
```

````

````{py:method} step(state: evorl.types.State) -> tuple[evorl.metrics.MetricBase, evorl.types.State]
:canonical: evorl.algorithms.meta.pbt_sac.param_sac.ParamSACWorkflow.step

````

`````

````{py:function} make_mlp_sac_agent(action_space: evorl.envs.Space, critic_hidden_layer_sizes: tuple[int] = (256, 256), actor_hidden_layer_sizes: tuple[int] = (256, 256), init_alpha: float = 1.0, discount: float = 0.99, normalize_obs: bool = False)
:canonical: evorl.algorithms.meta.pbt_sac.param_sac.make_mlp_sac_agent

```{autodoc2-docstring} evorl.algorithms.meta.pbt_sac.param_sac.make_mlp_sac_agent
:parser: autodoc2_docstrings_parser
```
````