IQ.Pilot Prebuilt Release @ ab07000

tinygrad_repo/docs/nn.md

@@ -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