home/categories/machine-learning/benchflow-ai-skillsbench-tasks-mhc-layer-impl-environment-skills-nanogpt-training-skill-md
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.
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
使用
安装后，您可以通过在终端运行以下命令来使用此技能：
skills use nanogpt-training