Efficient, honest AI β small models trained from scratch, with the full recipe published.
We build compact language models that punch above their weight, and we release everything needed to trust and reproduce them: code, weights, evaluations, and reports.
Tiny language models pretrained from scratch (jugnu β ΰ€ΰ₯ΰ€ΰ€¨ΰ₯ β means "firefly": small, but it glows). All under 150M parameters, all trained on 4Γ Blackwell GPUs, all Apache-2.0.
| Model | Params | BLiMP | ARC-Easy | WikiText-2 (byte-ppl) |
|---|---|---|---|---|
| JugnuLM-53M | 53.5M | 78.1 | 51.4 | 2.04 |
| JugnuLM-110M (R0) | 109.7M | 81.3 | 52.5 | 1.95 |
| JugnuLM-110M-R1 β + value residuals | 109.7M | 81.1 | 54.7 | 1.94 |
| JugnuLM-110M-R2 β + Muon | 109.7M | 80.8 | 56.1 | 1.93 |
| JugnuLM-110M-R3 β + data blend | 109.7M | 81.8 | 53.6 | 1.91 |
| JugnuLM-110M-R4a β + logit KD (heavy) | 109.7M | 80.4 | 57.0 | 2.18 |
| JugnuLM-110M-R4b β + logit KD (light) | 109.7M | 79.3 | 55.5 | 1.92 |
| π JugnuLM-110M-R2+ β R2 recipe, scaled | 109.7M | 82.5 | 55.1 | 1.87 |
JugnuLM-110M-R2+ is our flagship β the kept R2 recipe (value residuals + Muon) scaled to 25B tokens under a WSD schedule. It posts the family's best BLiMP and perplexity, beats the baseline on every metric, and ranks #1 on the Tiny-ML Leaderboard efficiency score β a hair ahead of GPT-X2-125M, at a smaller size. (It's a value-residual model, so load it with trust_remote_code=True.)
JugnuLM-110M's 81.3% BLiMP β GPT-X2-125M at ~12% fewer parameters and ~9Γ fewer training tokens.
Starting from an honest 110M baseline (R0), we add one lever at a time and keep only what beats the previous rung on the metric that matters:
Training code & full recipe: github.com/AltSlate-Labs/jugnu