# {py:mod}`evox.triton_kernels.kernels.lora_noise`

```{py:module} evox.triton_kernels.kernels.lora_noise
```

```{autodoc2-docstring} evox.triton_kernels.kernels.lora_noise
:allowtitles:
```

## Module Contents

### Functions

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

* - {py:obj}`_flatten_to_2d <evox.triton_kernels.kernels.lora_noise._flatten_to_2d>`
  - ```{autodoc2-docstring} evox.triton_kernels.kernels.lora_noise._flatten_to_2d
    :summary:
    ```
* - {py:obj}`_ceil_div4 <evox.triton_kernels.kernels.lora_noise._ceil_div4>`
  - ```{autodoc2-docstring} evox.triton_kernels.kernels.lora_noise._ceil_div4
    :summary:
    ```
* - {py:obj}`compute_counter_offsets <evox.triton_kernels.kernels.lora_noise.compute_counter_offsets>`
  - ```{autodoc2-docstring} evox.triton_kernels.kernels.lora_noise.compute_counter_offsets
    :summary:
    ```
* - {py:obj}`generate_lora_factors <evox.triton_kernels.kernels.lora_noise.generate_lora_factors>`
  - ```{autodoc2-docstring} evox.triton_kernels.kernels.lora_noise.generate_lora_factors
    :summary:
    ```
* - {py:obj}`lora_delta_output <evox.triton_kernels.kernels.lora_noise.lora_delta_output>`
  - ```{autodoc2-docstring} evox.triton_kernels.kernels.lora_noise.lora_delta_output
    :summary:
    ```
* - {py:obj}`lora_gradient <evox.triton_kernels.kernels.lora_noise.lora_gradient>`
  - ```{autodoc2-docstring} evox.triton_kernels.kernels.lora_noise.lora_gradient
    :summary:
    ```
````

### API

````{py:function} _flatten_to_2d(weight_shape: tuple) -> tuple[int, int]
:canonical: evox.triton_kernels.kernels.lora_noise._flatten_to_2d

```{autodoc2-docstring} evox.triton_kernels.kernels.lora_noise._flatten_to_2d
```
````

````{py:function} _ceil_div4(x: int) -> int
:canonical: evox.triton_kernels.kernels.lora_noise._ceil_div4

```{autodoc2-docstring} evox.triton_kernels.kernels.lora_noise._ceil_div4
```
````

````{py:function} compute_counter_offsets(param_shapes: list[tuple], lora_rank: int) -> list[int]
:canonical: evox.triton_kernels.kernels.lora_noise.compute_counter_offsets

```{autodoc2-docstring} evox.triton_kernels.kernels.lora_noise.compute_counter_offsets
```
````

````{py:function} generate_lora_factors(seeds: torch.Tensor, weight_shape: tuple, rank: int, counter: int) -> torch.Tensor | tuple[torch.Tensor, torch.Tensor]
:canonical: evox.triton_kernels.kernels.lora_noise.generate_lora_factors

```{autodoc2-docstring} evox.triton_kernels.kernels.lora_noise.generate_lora_factors
```
````

````{py:function} lora_delta_output(x: torch.Tensor, A: torch.Tensor, B: torch.Tensor, sigma: float) -> torch.Tensor
:canonical: evox.triton_kernels.kernels.lora_noise.lora_delta_output

```{autodoc2-docstring} evox.triton_kernels.kernels.lora_noise.lora_delta_output
```
````

````{py:function} lora_gradient(fitness: torch.Tensor, A: torch.Tensor, B: torch.Tensor | None, pop_size: int, sigma: float, weight_shape: tuple) -> torch.Tensor
:canonical: evox.triton_kernels.kernels.lora_noise.lora_gradient

```{autodoc2-docstring} evox.triton_kernels.kernels.lora_noise.lora_gradient
```
````