# kd.optim

[[[Source]]](https://github.com/google-research/kauldron/tree/main/kauldron/optim/__init__.py)

```{eval-rst}
.. automodule:: kauldron.optim
  :no-members:
```

## Symbols


### Class

|  |  |
--- | ---
[kd.optim.UseEmaParams](UseEmaParams) | Use the EMA parameters stored by the `ema_params` transform.

### Function

|  |  |
--- | ---
[kd.optim.decay_to_init](decay_to_init) | Add (params - init_params) scaled by `weight_decay`.
[kd.optim.ema_params](ema_params) | Store an EMA version of model parameters.
[kd.optim.exclude](exclude) | Create a mask which selects all nodes except the ones matching the pattern.
[kd.optim.named_chain](named_chain) | Wraps optax.named_chain and allows passing transformations as kwargs.
[kd.optim.partial_updates](partial_updates) | Applies the optimizer to a subset of the parameters.
[kd.optim.select](select) | Create a mask which selects only the sub-pytree matching the pattern.

```{toctree}
:hidden:

UseEmaParams
decay_to_init
ema_params
exclude
named_chain
partial_updates
select
```