DEV Community

#pytorch

Posts

đź‘‹ Sign in for the ability to sort posts by relevant, latest, or top.
nanochat: Train Your Own GPT-2 Grade ChatGPT in About Two Hours

nanochat: Train Your Own GPT-2 Grade ChatGPT in About Two Hours

Comments
9 min read
How many GPUs do you need to train an AI model?

How many GPUs do you need to train an AI model?

Comments
8 min read
Warmup Did Nothing for My Learning Rate — Then It Fully Rescued the Wrong One

Warmup Did Nothing for My Learning Rate — Then It Fully Rescued the Wrong One

Comments
4 min read
My Gradient Clipping Was Never Clipping. It Was Secretly Multiplying My Weight Decay

My Gradient Clipping Was Never Clipping. It Was Secretly Multiplying My Weight Decay

Comments
4 min read
Saving and Loading Models in PyTorch: state_dict, Checkpoints and Resuming Training

Saving and Loading Models in PyTorch: state_dict, Checkpoints and Resuming Training

Comments
13 min read
Weight Decay 0.05: Catastrophe in Adam, Optimum in AdamW

Weight Decay 0.05: Catastrophe in Adam, Optimum in AdamW

Comments
4 min read
Distillation didn't help our small model until we took the labels away

Distillation didn't help our small model until we took the labels away

Comments
3 min read
Running Captum on Cloud TPUs with TorchTPU: a work log

Running Captum on Cloud TPUs with TorchTPU: a work log

Comments
19 min read
Gemma 4 in Rust on TorchTPU: Hands On With Colab

Gemma 4 in Rust on TorchTPU: Hands On With Colab

8
Comments
9 min read
TorchTitan's October update is out. I fixed 2 bugs in its quick-start script

TorchTitan's October update is out. I fixed 2 bugs in its quick-start script

Comments
6 min read
Model soups failed until we fixed one line: our 26M grammar tagger from 11% to 1.9% false fixes

Model soups failed until we fixed one line: our 26M grammar tagger from 11% to 1.9% false fixes

Comments
3 min read
PyTorch Custom Dataset and DataLoader: __getitem__, __len__, collate_fn, and num_workers

PyTorch Custom Dataset and DataLoader: __getitem__, __len__, collate_fn, and num_workers

Comments
9 min read
How PyTorch Became the Socket Every AI Chip Plugs Into

How PyTorch Became the Socket Every AI Chip Plugs Into

5
Comments
4 min read
LoRA on a 16 GB card: what fits and what does not

LoRA on a 16 GB card: what fits and what does not

Comments
5 min read
Blackwell quantisation on my own card

Blackwell quantisation on my own card

Comments
5 min read
đź‘‹ Sign in for the ability to sort posts by relevant, latest, or top.