Skip to main content

Module adam

teeny_kernels::nn::optim

Module adam 

Source

Structs§

AdamStep
Adam step.
AdamwStep
AdamW step (decoupled weight decay).

Functions§

adam_step
Adam step.
adamw_step
AdamW step (decoupled weight decay).