AI & ML interests

Small Models and Inference Cascades

Recent Activity

rajpdusΒ  updated a model about 1 hour ago
altslate/JugnuLM-110M-R2plus
rajpdusΒ  updated a Space about 9 hours ago
altslate/README
rajpdusΒ  published a model about 9 hours ago
altslate/JugnuLM-110M-R2plus
View all activity

Organization Card

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

datasets 0

None public yet