skills.homes logoskills.homescapability registry
home/categories/machine-learning/benchflow-ai-skillsbench-tasks-mhc-layer-impl-environment-skills-nanogpt-training-skill-md
cd ..
machine-learningdata-ai

nanogpt-training

Train GPT-2 scale models (~124M parameters) efficiently on a single GPU. Covers GPT-124M architecture, tokenized dataset loading (e.g., HuggingFace Hub shards), modern optimizers (Muon, AdamW), mixed precision training, and training loop implementation.

Посмотреть исходный кодmachine-learning
benchflow-ai
maintainer
benchflow-ai
Обновлено 1/19/2026
Звёзды
946
Форки
244
quick start

Installation and usage

Train GPT-2 scale models (~124M parameters) efficiently on a single GPU. Covers GPT-124M architecture, tokenized dataset loading (e.g., HuggingFace Hub shards), modern optimizers (Muon, AdamW), mixed precision training, and training loop implementation.

Установка
$ install --globalskills.sh
Использование

После установки вы можете использовать этот skill, выполнив следующую команду в терминале:

skills use nanogpt-training