AIAny
AI Train2024
Icon for item

minimind

Trains a sub-100M-parameter LLM from scratch — pretraining, SFT, LoRA, DPO/RLHF, and distillation, sized from ~26M up to ~100M-plus dense and MoE. Headline figure: the ~64M minimind-3 variant's SFT stage runs 1 epoch in ~2h and ~3 RMB on one NVIDIA 3090.

Introduction

Most "learn how LLMs work" resources stop at toy notebooks or hand you a frozen checkpoint to fine-tune. The interesting bet here is the opposite: the entire pipeline that produces a real, chatting language model fits on one consumer GPU. The project's headline data point is that the SFT stage of its ~64M minimind-3 variant runs 1 epoch in roughly 2 hours and about 3 RMB of GPU rental on a single NVIDIA 3090 — cheap enough that the architecture and every training stage become things you can actually rerun and break.

What Sets It Apart
  • It builds the model from absolute zero rather than fine-tuning an existing base — pretraining, SFT, LoRA, DPO/RLHF, and knowledge distillation are all present end to end. So the parts usually treated as black boxes become editable code.
  • The full modern training ladder lets you trace how a base model turns into something that follows instructions, not just read about it.
  • Sizes scale from ~26M up through ~100M-plus dense and MoE variants. So the same codebase covers both "smallest thing that talks" and experiments with mixture-of-experts routing.
  • A concrete cost figure is attached to the ~64M minimind-3 SFT run (~3 RMB, ~2 hours, one 3090). So reproducing that result is a planned afternoon, not an open-ended GPU bill.
  • A vocabulary-training example (train_tokenizer.py) is included for the curious, though the project deliberately ships and recommends its own minimind_tokenizer and advises against retraining it, since changing the vocab breaks weight, data, and ecosystem compatibility.
Who It's For

Great fit if you want to understand transformer LLMs by building one — students, engineers ramping into ML, or anyone who learns by rerunning the whole loop on hardware they own. Look elsewhere if you need a deployable assistant: these models are deliberately tiny and meant for study and experimentation, so their factual range and reasoning depth are far below production-grade models.

Information

  • Websitegithub.com
  • OrganizationsIndependent
  • Authorsjingyaogong
  • Published date2024/08/27

Categories

More Items

GitHub
AI Train2025

An asynchronous, high-throughput framework for large-scale reinforcement learning and agentic training that scales to 1T+ MoE models and 1000+ GPUs, with native verifiers integration, end-to-end SFT/RL/evals, and Slurm/Kubernetes deployment; requires NVIDIA GPUs.

GitHub
AI Agent2026

Trains reusable natural-language 'skills' for frozen LLM agents by optimizing the skill document in text-space — using trajectory-driven edits, validation-gated updates, and deployable best_skill.md artifacts. Multi-backend, zero inference-time cost at deployment, designed for iterative, validation-led skill improvement.

GitHub
AI Train2023

Modular PyTorch-based framework for building, training, and deploying physics-informed ML models (neural operators, PINNs, GNNs, diffusion). Provides GPU‑optimized training, domain-specific datapipes for meshes/point clouds, distributed scaling and a model zoo.