IQ.Pilot Release Commit @ bec7652
This commit is contained in:
40
artifacts/package_sources/tinygrad/docs/nn.md
Normal file
40
artifacts/package_sources/tinygrad/docs/nn.md
Normal file
@@ -0,0 +1,40 @@
|
||||
## Neural Network classes
|
||||
|
||||
::: tinygrad.nn.BatchNorm
|
||||
::: tinygrad.nn.Conv1d
|
||||
::: tinygrad.nn.Conv2d
|
||||
::: tinygrad.nn.ConvTranspose1d
|
||||
::: tinygrad.nn.ConvTranspose2d
|
||||
::: tinygrad.nn.Linear
|
||||
::: tinygrad.nn.GroupNorm
|
||||
::: tinygrad.nn.InstanceNorm
|
||||
::: tinygrad.nn.LayerNorm
|
||||
::: tinygrad.nn.LayerNorm2d
|
||||
::: tinygrad.nn.RMSNorm
|
||||
::: tinygrad.nn.Embedding
|
||||
::: tinygrad.nn.LSTMCell
|
||||
|
||||
## 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.tar_extract
|
||||
options:
|
||||
show_signature: false
|
||||
separate_signature: false
|
||||
::: tinygrad.nn.state.torch_load
|
||||
options:
|
||||
show_signature: false
|
||||
separate_signature: false
|
||||
::: tinygrad.llm.gguf.gguf_load
|
||||
Reference in New Issue
Block a user