
Parameter Golf
https://github.com/openai/parameter-golf- Category
- AI Tools
- Rank
- No. 946Tools index
- Listed in
- #39 Find AI benchmarks
- Pricing
- Open Source
- Type
- TOOL
- Builder
- openai
- GitHub
- 5.2k stars
- Date
About
OpenAI's challenge: train the smallest language model that fits in 16MB. Best model wins — a benchmark for tiny-LM engineering.
What it does
Parameter Golf provides runnable training baselines, dataset preparation utilities, evaluation logic, and an archive of competitive experiments. Participants can change architecture, tokenizer, optimization, quantization, and test-time methods. Submissions are compared by FineWeb validation compression in bits per byte, while the official track enforces artifact-size and training-time limits.
Why it's ranked here
This is unusually useful competition infrastructure, not merely a problem statement. The repository pairs readable starting points with reproducible data tooling and preserved experiment lineages. Its tokenizer-neutral metric also leaves meaningful room for model and compression research. The main reservation is the official track's substantial hardware requirement.
What's good
The baseline exposes model shape, optimizer settings, validation cadence, and time limits through environment variables, making controlled experiments straightforward. Evaluation counts bytes rather than assuming a fixed tokenizer. Separate PyTorch and MLX implementations broaden experimentation options. Archived submissions document concrete techniques including parameter tying, recurrence, quantization, fused kernels, and test-time training.
Tradeoffs
Official comparisons require eight H100 GPUs, so faithful participation remains expensive despite the strict time cap. The supplied baselines are explicitly starting points, not competitive configurations. Stronger entries can combine specialized CUDA kernels, compression tooling, custom tokenizers, and lengthy test-time procedures, which raises reproduction complexity considerably.
How to use it well
Use it for compact-language-model research, architecture experiments, tokenizer studies, and compression benchmarking. Start from the readable baseline, download matched FineWeb shards, change one controlled setting at a time, and compare bits per byte. Study archived submissions when the baseline stops teaching you. It does not address production serving, application integration, or general model deployment.
Technical notes+
train_gpt.py implements a PyTorch baseline with DistributedDataParallel, CUDA bfloat16 execution, Muon and Adam-style parameter groups, environment-driven hyperparameters, SentencePiece byte lookup tables, and full-validation BPB calculation. train_gpt_mlx.py mirrors the baseline in MLX with gradient accumulation and memory-oriented microbatch chunking. requirements.txt lists PyTorch, NumPy, Hugging Face Hub, datasets, tiktoken, and SentencePiece, but does not list MLX. data/cached_challenge_fineweb.py downloads manifest-selected shards and tokenizer artifacts, while data/download_hf_docs_and_tokenize.py can build pure-byte or SentencePiece datasets. measure_variance_ratio.py is a CLI analysis utility for comparing attention payload variance. scripts/run_1868_rerun.py orchestrates seeded eight-GPU compliance reruns. results/pr1868_rerun_compliance/train_gpt.py shows a competitive configuration using Triton fused softcapped cross-entropy and FlashAttention.
Observed
- Primary language
- Python
- Install surface
- A pip requirements file lists the core PyTorch and data-processing dependencies.
- Interfaces
- Command-line scripts cover training, dataset acquisition, tokenization, analysis, and compliance reruns.
- Compute backends
- The repository supplies distributed PyTorch CUDA and separate MLX training implementations.
- Evaluation
- FineWeb validation compression is measured in tokenizer-agnostic bits per byte.
- Repository structure
- Readable newcomer baselines are separated from archived competitive submissions.
Read from README.md, requirements.txt, train_gpt.py, train_gpt_mlx.py, measure_variance_ratio.py, scripts/run_1868_rerun.py, data/cached_challenge_fineweb.py, data/download_hf_docs_and_tokenize.py, results/pr1868_rerun_compliance/train_gpt.py, records/track_10min_16mb/2026-03-18_LowerLR/train_gpt.py.
What it can do
Train compact language models under size constraints
Training data and model architecture parameters → Language model under 16MB
Benchmark model performance against size limits
Trained language model → Performance metrics and size validation
Optimize model parameters for minimal footprint
Base language model and compression settings → Size-optimized model parameters
Compare models in competitive ranking
Multiple submitted models → Leaderboard with best performing tiny models
Validate model size compliance
Language model file → Size verification and compliance status
Intel on Parameter Golf
Tags
Tech Stack
Comments (0)
No comments yet
Editorially curated, with community endorsements as a secondary signal. Corrections welcome.