Skip to content

AI Developer Tool · Python

mini-llm-lab

Single-GPU decoder-only LLM pretraining in PyTorch. Built for study and experimentation with modular architectures and Hydra configs: components are written from scratch to modify in source, with toggles for BF16 AMP, SDPA, compile, and Liger FLCE for speed.

What it is

mini-llm-lab is a configurable framework for pretraining decoder-only language models from scratch, featuring custom implementations of model components, a two-stage warmup-stable-decay training schedule, bit-exact resume capability, and an integrated chat interface for interacting with trained checkpoints.

Scored at4558ef911 Nov 2026Changes since

Merit describes the code at this commit. Later pushes count once the owner rescans. How merit is scored

Where it counts

  • NLP & LLMs95%
  • Machine Learning90%
  • Data Science40%
  • MLOps30%
  • AI Agents20%
  • Computer Vision10%

How much of the project is about each field, as scoring counts it. It ranks in the field it is most about.

What the judge found

Read from the project's code at its scored commit.

Strengths4

  • Highly configurable model architecture with reference and optimized implementations
  • Advanced training features: WSD schedule, bit-exact resume, memmap data pipeline, on-GPU metrics
  • Comprehensive testing and documentation
  • Demonstrated capability to train a 116M model on a single consumer GPU

Limitations

The owner keeps this project's limitations private.

Around it on the NLP & LLMs board

Whole board
  1. 1ROCmForgeInference System · Rust77.9/100ExpertLuiz Cesar Spies
  2. 2mini-llm-labAI Developer Tool · Python77.0/100AdvancedMohamed Khayat
  3. 3regescore-benchmarkBenchmarking Project · Python76.5/100AdvancedMaster Mind
  4. 04quarryML Application · TypeScript75.3/100AdvancedMurat Genç

More from Mohamed Khayat