mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-02-07 21:26:21 -05:00
29 lines
665 B
Markdown
29 lines
665 B
Markdown
## 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 |