# {py:mod}`evox.problems.neuroevolution.utils`

```{py:module} evox.problems.neuroevolution.utils
```

```{autodoc2-docstring} evox.problems.neuroevolution.utils
:allowtitles:
```

## Module Contents

### Classes

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

* - {py:obj}`ModelStateForwardResult <evox.problems.neuroevolution.utils.ModelStateForwardResult>`
  - ```{autodoc2-docstring} evox.problems.neuroevolution.utils.ModelStateForwardResult
    :summary:
    ```
````

### Functions

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

* - {py:obj}`get_vmap_model_state_forward <evox.problems.neuroevolution.utils.get_vmap_model_state_forward>`
  - ```{autodoc2-docstring} evox.problems.neuroevolution.utils.get_vmap_model_state_forward
    :summary:
    ```
````

### API

`````{py:class} ModelStateForwardResult
:canonical: evox.problems.neuroevolution.utils.ModelStateForwardResult

Bases: {py:obj}`typing.NamedTuple`

```{autodoc2-docstring} evox.problems.neuroevolution.utils.ModelStateForwardResult
```

````{py:attribute} init_state
:canonical: evox.problems.neuroevolution.utils.ModelStateForwardResult.init_state
:type: typing.Tuple[typing.Dict[str, torch.Tensor], typing.Dict[str, torch.Tensor]]
:value: >
   None

```{autodoc2-docstring} evox.problems.neuroevolution.utils.ModelStateForwardResult.init_state
```

````

````{py:attribute} state_forward
:canonical: evox.problems.neuroevolution.utils.ModelStateForwardResult.state_forward
:type: typing.Callable
:value: >
   None

```{autodoc2-docstring} evox.problems.neuroevolution.utils.ModelStateForwardResult.state_forward
```

````

`````

````{py:function} get_vmap_model_state_forward(model: torch.nn.Module, pop_size: int, device: torch.device, in_dims=(0, None), randomness='different') -> evox.problems.neuroevolution.utils.ModelStateForwardResult
:canonical: evox.problems.neuroevolution.utils.get_vmap_model_state_forward

```{autodoc2-docstring} evox.problems.neuroevolution.utils.get_vmap_model_state_forward
```
````