AI & ML interests
Small Models and Inference Cascades
Recent Activity
AltSlate Labs
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.
πͺ° The Jugnu family
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.
The ablation ladder
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:
- R1 β value residuals (ResFormer): ARC-Easy +2.2 β kept.
- R2 β Muon optimizer: ARC-Easy +1.4, lower perplexity β kept.
- R3 β diverse data blend (+DCLM +FineMath): best BLiMP & perplexity, but ARC-Easy β2.5 β dropped. An honest negative β at this scale, FineWeb-Edu's educational filtering feeds ARC-Easy more than raw diversity does.
- R4 β logit distillation (from a 1.7B teacher): brackets a narrow operating point β R4a (heavy KD) lifts ARC-Easy to a family-best 57.0 but hurts perplexity (2.18); R4b (light KD) recovers perplexity but loses the ARC gain β both dropped. Offline KD across a ~15Γ teacherβstudent gap isn't a clean win at this scale.
- R2+ β the competitive run: take the strongest kept stack (R2) and scale it β 25B tokens + a WSD schedule β best-in-family BLiMP and perplexity, and #1 on the leaderboard efficiency score.
Training code & full recipe: github.com/AltSlate-Labs/jugnu
What we care about
- Efficiency per parameter β the sub-150M regime: on-device, low-latency, cheap-to-serve models.
- Reproducibility β public code, open weights, and one-command evals.
- Honest reporting β clean baselines and measured ablations over unvalidated clever tricks. We publish what didn't work (see R3 and R4), not just what did.
Links
- π» Code β https://github.com/AltSlate-Labs
- π Web β https://altslate.com
- βοΈ hello@altslate.com