tinygrad/docs/nn.md at cc33947fa563912400a060031a78796b1a5a2b09

mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00

Files

George Hotz d325be2540 update docs (#4356 )

* update docs

* nn.md

* mnist cleanups

* rhip test is very slow

2024-04-30 16:51:42 +09:00

665 B

Raw Blame History

Neural Network classes

::: tinygrad.nn.BatchNorm2d ::: tinygrad.nn.Conv1d ::: tinygrad.nn.Conv2d ::: tinygrad.nn.ConvTranspose2d ::: tinygrad.nn.Linear ::: tinygrad.nn.GroupNorm ::: tinygrad.nn.InstanceNorm ::: tinygrad.nn.LayerNorm ::: tinygrad.nn.LayerNorm2d ::: tinygrad.nn.Embedding

Optimizers

::: tinygrad.nn.optim.SGD ::: tinygrad.nn.optim.LARS ::: tinygrad.nn.optim.AdamW ::: tinygrad.nn.optim.Adam ::: tinygrad.nn.optim.LAMB

Load/Save

::: tinygrad.nn.state.safe_load ::: tinygrad.nn.state.safe_save ::: tinygrad.nn.state.get_state_dict ::: tinygrad.nn.state.get_parameters ::: tinygrad.nn.state.load_state_dict ::: tinygrad.nn.state.torch_load

665 B Raw Blame History

Neural Network classes

Optimizers

Load/Save

665 B

Raw Blame History