Marin: Week of May 25th summary

Milestone: May: Prepping scaling training recipe + data mix + GPU training
Contents
  1. Data
  2. Summary
  3. Production data-mix swarm goes live; Will pre-registers a vibe-mixed candidate
  4. Reconcile becomes Iris's only task-lifecycle channel; preempt races closed
  5. Inference-distributed library blocked from main; Kueue gang admission for preemptible evals
  6. Prompt-format sensitivity ruled out; a real code-interpretation gap found
  7. Epic closes on a blind intruder test of v0 quality and domain taggers
  8. isoFLOP results land: May Recipe ~2.12x faster than v16, June compute set
  9. FA4 THD attention beats CuTe 2x on B200; d5120 MoE still 4-6x behind Megatron
  10. June MoE run scoped into 10T-token plan; DeepEP backend lands in draft
  11. Other Changes
  12. Community Pulse
  13. Agent MoE
  14. Runs
GitHub
60 merged 39 opened 38 issues closed 16 contributors 8 epics 229 comments this week
Compute
GCP TPU 1.38e24 HW FLOPs (1.45e23 reserved) W&B 1.65e22 HW FLOPs (6.30e21 model FLOPs)
Compute calculations should be taken with a large grain of salt.
Infra
Discord
392 messages 112 authors 18 new members 16 channels active 12 threads
Tokens
19.0T tokens 0 32.9% synthetic 113 datasets 🤗 collection
web 12.6T (66.3%) multilingual 4.1T (21.3%) code 1.6T (8.3%) specialized 384.0B (2.0%) math 377.1B (2.0%)
Milestones

The May milestone effectively landed this week, on two fronts at once. On the recipe side, Larry Dial closed out the four-budget isoFLOP sweep #6074 and tied the per-budget optima into a scaling law carrying the same exponent as the v16 README curve — implying a uniform ~2.12× equal-TPS speedup for the combined MuonH May Recipe, reaching the v16 1e23 loss at roughly half the compute. With the refit LR law #5951 now in heuristic.py and a per-dim compute-optimal table published, the June run finally has a forecastable target shape; the optimizer and architecture probes still running under agent autonomy (MuonEq, AMUSE, parallel layouts, sandwich post-norm) all failed gate 1, which is exactly what locking a recipe looks like. On the data side, the #5360 quality/dedup epic closed on May 29 with its long-promised independent metric — a blind LLM intruder study where v0 taggers beat the Dolma3 baselines on both domain (0.710 vs 0.490) and quality (0.292 vs 0.214) — and the production data-mix swarm that had been code-only last week went live as an 840-candidate Fisher DSP design #5958, with Will Held pre-registering a hand-steered "vibe mix" #6063 before any training. The recurring finding across the swarm work was that there is no free lunch in the mixture — quality cannot be modeled as a scalar discount, and gains on code/math come as explicit trade-offs against news and fiction.

With May's recipe and data both converging, the center of gravity shifted to the June GPU hero run — and that is where the risk now sits. We scoped the run as a 10T-token, 120B-A2B MoE targeting ~June 19 conditional on CoreWeave #6044, raising the bar well past the epic's original 16B-A2B target. But David Hall's first fair full-node B200 profile told a sobering story: at d1024 Grug crushes Megatron (373k vs 166k tok/s), yet at the June-relevant d5120 it collapses to 14–24k against Megatron's 91.6k — a 4–6× gap driven mostly by MoE routing and ragged_all_to_all, not attention. The FA4 THD attention probe #6059 roughly halved attention-core time, and a clean DeepEP backend #5982 landed in draft, but closing the wide-model throughput gap is now squarely on the critical path. Underneath, the Iris core had a heavy week: Reconcile became the sole controller-to-worker task-lifecycle channel #5964 (−1500 LOC), the SetTaskStatusText→finelog migration finished #5988, and a batch of preemption-correctness fixes finally stopped spot-capacity churn from burning the failure budget — Larry reported one v5p-8 job preempted 32 times in three days.

#5359 Production data-mix swarm goes live; Will pre-registers a vibe-mixed candidate

Epic title: Determine data mixture for pre- and mid-training for June model


Progress: 75% Risk: ● On track
The production swarm is live as #5958 (#5365 confirms liftoff) and on track to finish before the June launch. Method and metric de-risking is active — the no-free-lunch quality finding and the pre-registered vibe mix #6063 — with the 10T June datamix #6045 following once the swarm resolves; remaining work is running it out, not a blocker.

Summary: Must Have May Target: An active swarm launched over all sources `datakit/sources.py`. The method and metric used for this swarm must have been successfully de-risked using the existing swarm from #2345 to improve results on: UncheatableEval, HumanEval, MMLU, GPQA, and all of David's PPL sets.

1/4 sub-issues closed

The production swarm launched this week, clearing the central deliverable that slipped last week. Will Held reported on #5365 that the swarm went up the prior day, a few days behind schedule, with two operational lessons logged for next time: ingestion onto the datakit store should be budgeted at roughly a week and given more lead time, and Iris should be stress-tested under full-ingestion load earlier, especially on CoreWeave. The swarm itself lands in PR #5958 as an 840-candidate Fisher DSP-aligned two-phase design — D512 MoE on ~100B tokens each over the us-central2 datakit store's 167 mixable domains plus a tail aggregate, with phase weights drawn from an offline D-optimal design and an 80/20 phase split. It ran live as iris-run-swarm_fisher_dsp-20260526-063341 with ~250 sub-tasks at submission. Will Held flagged the PR as not yet review-ready and noted the swarm is capped at 256 concurrent runs, so the D-optimal candidates are scheduled first and the proportional and per-domain baselines that Calvin Xu asked about will follow in a later batch.

Most of the week's thinking was on the metric-and-method side the epic requires de-risking before trusting the swarm, and the recurring conclusion was that there is no free lunch in the mixture. Working from the existing 240-run swarm, Calvin Xu showed sizable eval-loss crossovers — rank Spearman between the first eval and the final of only ~0.31 at 60M and ~0.23 at 300M — which undercut a tempting shortcut Will had floated of training larger models to a small fraction of the run instead of small models to completion scratch that idea then. When Percy Liang pressed on whether downweighting low-quality CC ought to give a free improvement, Will Held showed both HellaSwag and Paloma macro BPB are maxed within the swarm by proportional mixing, so any gains on code/math come as explicit tradeoffs. Calvin Xu separately found that reallocating low-quality budget to the matching high-quality bucket did not uniformly help HellaSwag, which convinced him quality cannot be modeled as a scalar discount.

That "no free lunch" framing motivated a tooling side-quest that produced this week's most concrete artifact. Will Held built a swarm explorer that, in its DSP-driven form, turns each benchmark slider into a steering wheel: dragging a task takes a minimum-norm step on the mixture along the analytical gradient of Calvin's DSP fit, so the mix only drifts as far as the demanded improvements force it make tradeoffs explicitly rather than implicitly. Using it, he hand-steered from token-proportional toward HumanEval, GSM8K, and (more gently) MMLU, and pre-registered the resulting mixture in #6063 before any training — five confident wins (humaneval, github_python/cpp, gsm8k, arxiv_physics), two confident losses on uncheatable news/fiction sets, and 17 tasks inside fit noise, with an explicit reservation that the confident-loss list is the shakiest part. The delta is concentrated in phase 1, which swaps general and high-quality CC for synthetic code, synthetic math, stack-edu, and arxiv. The stated point of pre-registering is the same discipline as the scaling-law work: keep the prediction honest and build confidence that capabilities can be controlled through the mix.

The downstream 10T June datamix issue #6045 remains open and unstarted, gated on this swarm resolving, while the metric-selection sub-issue #5362 is effectively the live debate above. On #2403, Calvin Xu backfilled the Grug-MoE v4 endpoint evals to full 20/20-task coverage at every hidden dim and added a standardized effect-size view showing the v4-optimized direction is positive on average but not Pareto-safe — its largest single regression (boolq, -1.16σ) exceeds its largest gain (medmcqa, +0.85σ).

1 PR this week, 3 new comments, and 0 new issues (4 total)
Sort:
3 autocategorized

#5369 Reconcile becomes Iris's only task-lifecycle channel; preempt races closed

Epic title: Infra Tune up - unified queries, zero-trust proxy, GH integration to Iris


Progress: 100% Risk: ● On track
As an ongoing catch-all for the Iris core, this milestone’s slice is complete: #5964 made Reconcile the sole task-lifecycle channel, the finelog status migration finished (#5988, #6012), and the preemption races were closed (#6004, #6000). The remaining named tracks (zero-trust proxy, GitHub-to-Iris) are continuous-improvement carry-forward, not gaps in this milestone.

Summary: Non-critical catch-al to make our lives better.

The week's centerpiece is making Reconcile the single controller-to-worker task-lifecycle channel and then hardening its shape. #5964 deleted the legacy StartTasks / StopTasks / PollTasks RPCs, the UpdateTaskStatus push, and the rollout gate now that every worker is on the new wire — roughly -1500 LOC, with routing now purely by attempt_uid #5921. #5954 trims the Reconcile observation payload to the desired set on both ends, and the still-open #5987 refactor splits the state machine into snapshot reads, a mutation writer, and pure transitions.py logic, with an AST purity guard that forbids cur.execute and attribute reads across the seventeen hot-path functions. #5965 stood up a synthetic benchmark_reconcile that drives the full snapshot → apply tick against an in-process fake worker and surfaced the real costs at zephyr scale — a 4.3s dispatch tick dominated by the apply path, and a 478 MB wire payload at 5000 tasks — giving the refactor a target rather than a guess. Russell Power rolled the reconcile-based scheduler out mid-week with a few controller restarts, confirming the path that terminates unknown slices behaved in production.

The second thread is preemption correctness, which had been quietly burning the failure budget. #6000 (open) has workers poll the GCP metadata server for the preempt notice, latch a drain bit, and SIGTERM local attempts; the controller treats that bit as authoritative and atomically marks the worker's tasks plus coscheduled siblings PREEMPTED, eliminating the JAX peer-loss races that previously looked like genuine failures #5872. Around it, #6004 gives worker-restart a failure budget and classifies "slice not found in zone" as skipped rather than fatal so one TPU reclaim no longer tanks a rollout, and #5977 raises the BackoffDetector short-lived threshold so spot capacity that lives ten to twenty-five minutes decays group health via AIMD instead of reading as normal preemption. This lines up with what users were seeing on the ground — Larry reported a batch v5p-8 job preempted 32 times in three days, making about one-ninth forward progress — and with Will Held's #5953, which lets the v4 reservation back small slices (sizes 8 and 16) with chip-budget-derived slice caps instead of stalling at whatever happened to be pre-warmed.

The finelog migration that began last week with #5912 landed and then got tuned. #5988 drops SetTaskStatusText — the highest-volume controller RPC at ~125/s — and routes status text into an iris.task_status finelog namespace read back by the dashboard with a ten-minute retention filter, while #6003 adds per-namespace storage policies so finished-task rows roll off at 1h / 100MB. Rafal Wojdylabot's #6012 then closed the last hot spot: in-task clients had still been writing those rows through the controller's StatsServiceProxy on its 64-thread pool, so a slow finelog backend could starve controller RPCs and fail jobs; the fix resolves the log server directly from the cluster registry. That change also flushed out a pyqwest bug on short-lived RPC sessions, which @rav confirmed the patched client resolves. Operationally, #6020 reaps hung py-spy profilers (and clears the ptrace group-stop a killed tracer leaves behind) that had been freezing zephyr shards for hours on both docker and CoreWeave, and #6078 stamps started_at_ms on the BUILDING transition so wedged image pulls finally fall into the execution-timeout scan instead of holding chips indefinitely.

Smaller hardening rounded out the week: #6018 lifts the RPC handler pool from 64 to 1024 threads so slow sync handlers can't head-of-line-block heartbeats, #5961 and #5952 move the worker-restart driver off a controller RPC and onto the CLI (fixing an AttributeError that broke every restart), and #5966 advertises the controller's SSH auth mode via a GCE label so marin.yaml can flip to OS Login without lockstep client upgrades. On the dashboard, #6017 adds a back-to-parent-job link. Notably, the epic's other named tracks — the zero-trust proxy and a GitHub-to-Iris integration — saw no movement this week; the energy went entirely into the controller core.

25 autocategorized
9 potentially related in Other Changes

#5368 Inference-distributed library blocked from main; Kueue gang admission for preemptible evals

Epic title: Inference service for evals


Progress: 75% Risk: ● On track
The eval-serving work is effectively complete and just needs to land: the brokered vLLM path #5887 and the Kueue gang-admission for preemption-resilient startup #6058 (validated on kind) are both written and pending merge. The marin.inference.distributed integration cascade is now understood #5985; on track once these merge.

Summary: Def of done: Selected evals run on Iris, in a worker pre-emption resilient fashion.

The brokered vLLM eval-serving path from last week, #5887, stayed open and unmerged through the week, still waiting on review from Russell Power. Meanwhile Ahmed Ahmed opened #5985 after trying to adopt the separate marin.inference.distributed library — the Zephyr-backed multi-region inference path on the inference/distributed-library fork — by vendoring only its subtree onto main. All cross-package imports resolve at static-check time, but at runtime regional_job._build_context passes a worker_environment kwarg that main's ZephyrContext does not accept. Running it end-to-end at 1e21 (3.4B) and 1e22 (9.7B) showed the cascade is three layers deep, not two: past a one-line lib/zephyr/ patch the worker actor then crashes on ZephyrWorker.__init__() got an unexpected keyword argument 'environment', which the fork only fixes via coordinated lib/fray/ changes that thread environment= through the actor-creation path. Ahmed Ahmed updated the definition-of-done to require those supporting zephyr and fray deltas land in main before the library is usable.

On the reliability side, Russell Power opened #6058, which gives the Iris K8s direct provider atomic gang admission via Kueue plain pod groups — the missing piece for preemption-resilient multi-worker jobs that the epic's def-of-done leans on. Previously the direct path had no atomic startup: a 64×8 job could bind 63 pods and strand the 64th behind the autoscaler, blocking the startup collective, with gang guarantees only on teardown. The PR promotes coscheduled gangs all-or-none on a single pod-group generation, deletes the backing Kueue Workload on teardown so a requeue does not deadlock behind a parked WaitingForReplacementPods quota hold, and bills a vanished coscheduled pod against the preemption budget rather than as a hard failure. It was validated end-to-end on a kind cluster — a gang that fits quota admits whole, a gang that does not stays fully gated with zero partial admission, and tearing down the first frees quota for the second to run.

Two scoping issues opened to shape the eval side of the service. Romain Yon filed #6043 for eval UX including a leaderboard, and we filed #6050 to enumerate and hook up the evals of interest (AA II evals, possibly SWE-bench) ahead of post-training; both still carry placeholder definitions of done. Separately David Hall's draft #5959 adds a Levanter labeled-LM-eval training callback for periodic pretraining probes — adjacent to this epic but a different code path from the served-eval service.

On Discord the epic drew new interest. In #evals Percy Liang connected a prospective contributor interested in helping add evals, and in #infra the same contributor asked what the happy path is for standing up an HTTP endpoint over a multi-host JAX mesh without Ray — exactly the serving shape this epic is converging on. The week's controller churn that the serving path runs under was also visible in #infra, where @rjpower rolled out the new reconcile-based Iris scheduler.

5 autocategorized

#5367 Prompt-format sensitivity ruled out; a real code-interpretation gap found

Epic title: Identifying perplexity gaps for eval and training


Progress: 50% Risk: ● Off track
The epic’s definition of done — an aggregate PPL pool that statistically correlates with post-trained performance across the Qwen model pool — is still not demonstrated, and the first dashboard verdicts mostly surfaced eval-construction artifacts rather than validated signal. New supervised target-only suites for prompt-format sensitivity #6067 and code interpretation #6070 landed via #6080, with BPB accounting hardened in #6011 and #6055 and Grug checkpoints wired in via #6056 — but the defining correlation remains unproven with the June deadline approaching.

Summary: Def of done: We should have an aggregate pool of PPL evaluations which (to relatively low bar standard, just positively correlate with P<0.05) correlate with the corresponding post-trained performance for a set of models. Pool of models to use would be Qwen 2.5 base models -> Qwen 2.5 coder...

The week turned from building out the gap pool to reading the first real dashboard verdicts on it. David Hall stood up a prompt-format sensitivity suite in #6067, holding task semantics fixed while rendering the same 5-shot examples through 156 surface templates, and after a v2 renderer fix for JSON/JSONL continuations and record-extraction targets the readout was reassuring: Marin 32B sits roughly on par with Qwen3 32B across the slices and is often better, so broad wrapper/template fragility is no longer treated as a worry. The exception was Python REPL/reasoning continuations, which were split out into their own code-interpretation family under #6070 for static target-only records like len('prompt') * 7 producing a final output line, deliberately avoiding famous functions and keeping the scored span to the REPL/doctest result. In #evals on Discord, David Hall flagged this as a real gap area on the function-definition-calls heatmap and asked whether the CWM work might close it.

Underneath the new probes, the scoring machinery got more trustworthy. #6079 added supervised target-only scoring so datasets with separate input and target fields score only the continuation after a prompt, rather than treating each document as one span and contaminating metrics with support-prompt bytes; that machinery landed on main, and the actual prompt-format and code-interpretation suites were retargeted directly to main in #6080. Two BPB-accounting fixes under #5821 hardened the byte math: #6011 aggregates loss against decoded byte totals instead of token-weighted batch BPB and preserves loss from zero-byte token IDs, failing fast when perplexity-gap offsets are not source-aligned, and #6055 adds source-document BPB so TokenMonster control tokens group with the bytes they modify rather than silently dropping their loss. Pranshu Chaturvedi added native Grug checkpoint support to the model-score runner in #6056, handling the alternate max_seq_len and MoE-layout conventions so the d512/d768/d1024 MoE ladder being tracked in #5821 can flow through the same gap workflow.

Two cleanup items also merged: #5946 stops uploading model perplexity score bundles to W&B as tracker artifacts, writing them only to the GCS output path while keeping scalar logging and gap reports, and #5803 documents the end-to-end raw and supervised-target gap workflow from provider registration through Iris scoring, report retry, heatmap generation, and website publication. The FEVER factuality slice #5863 drew a useful exchange with Yiyuan Li on whether evidence-conditioned entailment is the right proxy; David Hall framed it as "throwing crap at the wall" to find ICL proxies for a model's awareness of evidentiary reasoning, with the real downstream goal being hallucination robustness left to post-training. On the adjacent pass@k front, Rohith Kuditipudi's #4549 got a full 36-cell GSM8K mask_00 readout across the Delphi midtrain matrix.

11 autocategorized

#5360 Epic closes on a blind intruder test of v0 quality and domain taggers

Epic title: Data pipeline: quality scores + dedup param selection


Progress: 100% Risk: ● On track
The epic #5360 closed on 2026-05-29 with all four V0 components — dedup, decontamination, quality, and domain — having both a defined interface and an independent metric, capped by the blind LLM intruder study in #5822 that scores each tagger and shows v0 beating the Dolma3 baselines. Quality intruder accuracy (0.292) is only modestly above chance and the 20T-token target in #3101 retired stale at ~12T, but those are explicitly carried into the June issues #6037 and #6036 rather than being unmet milestone goals.

Summary: The May milestone exercises the Marin pretraining pipeline end-to-end. Each component \- datakit, data mixing, training, evaluation \- needs a well-defined interface and an independent metric to climb.

7/13 sub-issues closed

The epic closed this week. Rafal Wojdyla wrapped the last three deliverables — Dolma3 quality scores #5812, Luxical-One domain buckets #5808, and Dolma3 WebOrganizer topic buckets #5811 — into #5822, then closed #5360 on 2026-05-29. The independent metric the milestone was meant to climb finally exists: a blind LLM intruder study, where a judge picks the odd-one-out from four same-bucket docs plus one contrast doc, gives each tagger a single accuracy number. On domain, the v0 Luxical K=40 clustering beat the off-the-shelf Dolma3 WebOrganizer classifier cleanly, 0.710 vs 0.490 with separated Wilson CIs; on quality, the oracle-LLM-distilled v0 classifier edged Dolma3 fasttext 0.292 vs 0.214 (z=2.99, p≈0.003). Both quality numbers sit only modestly above the 0.20 five-way-chance floor, and Rafal Wojdyla framed this explicitly as a starting point — enough to begin, with June's job being to push these up.

The dedup side of "dedup param selection" got its first full-corpus run, reported in #deduplication: 15,089,658,952 docs in, 68.33% singletons, 31.67% landing in some duplicate cluster, of which 22.25% are removable non-canonicals — so the kept set is 77.75% of documents. Without FineTranslations (~3T), that takes a ~15T-token input down to roughly 12T deduped tokens. Doc-level decontamination removes only ~0.04% / 6M elements, which Rafal Wojdyla suspects carries a fair number of false positives, prompting David Hall to ask for examples of what gets thrown out. Notably, the long-standing #3101 ("ensure we have 20T deduped tokens") was auto-closed as stale rather than satisfied — the actual yield is ~12T, so the headline token target was retired rather than hit.

The data-mixing channel surfaced a result that bears directly on how these quality scores get used downstream: starting from proportional and shifting 50% of each low-quality split's budget to its high-quality counterpart did not yield uniform HellaSwag gains, which convinced the team that quality "should not be modeled as a scalar discounting factor" — a caution about treating the new per-doc quality scores as a simple reweighting knob. Forward work is now split out into June follow-on issues filed against this epic: #6036 (get the tokenized, mixing-compatible pretraining data into the CoreWeave cluster for the June run) and #6037 (a time-boxed process for climbing local dedup/decon/quality/domain metrics), with Will Moss's Nemotron-ferry tuning #6065 still open on the throughput front.

0 PRs this week, 3 new comments, and 0 new issues (13 total)
Sort:
4 autocategorized

#5358 isoFLOP results land: May Recipe ~2.12x faster than v16, June compute set

Epic title: Land the scaling recipe for June model


Progress: 100% Risk: ● On track
The definition of done—isoFLOP results enabling a June forecast—is met: the four-budget isoFLOP sweep #6074 produced fitted bowls and a scaling law (loss(C)=1.6+88.90·C^-0.0941) showing a uniform ~2.12× equal-TPS gain over v16, the refit LR law #5951 landed in heuristic.py, and a per-dim compute-optimal table now forecasts the June run. The remaining optimizer probes (#6066, #5728, #6082) only refine, not gate, the locked recipe; preregistration of the June loss guess sits on #6046 and per the epic body may wait on #5359.

Summary: Def of done: Scaling recipe with isoFLOP results, enable forecast of June run [may not include pre-reg until #5359]

43/50 sub-issues closed

The epic's definition of done—a scaling recipe with isoFLOP results good enough to forecast the June run—effectively landed this week. Larry Dial closed out the four-budget isoFLOP sweep in #6074 (1e18 / 3e18 / 1e19 / 3e19 across hidden dims 512–1280 on v5p-32), fit a parabola in log(tokens) at each budget, and tied the optima together into a scaling law: loss(C) = 1.6 + 88.90·C^-0.0941 on the drop-1e18 fit, which carries the same exponent as the v16 README curve and therefore implies a uniform ~2.12× equal-TPS speedup at every budget for the combined MuonH May Recipe. Projecting forward, the May Recipe reaches the v16 baseline's 1e23 loss at roughly half the compute, and Larry published a compute-optimal recommendation table—compute, tokens, batch, steps, and LR per dim—mapping straight onto the June shape (e.g. d=1280 → 3.46e19 / 1.5e10 tokens / projected loss 2.889). He explicitly down-weighted the 1e18 bowl, where Muon does disproportionately well at tiny scale with 1pct warmup, and flagged it as a mild extrapolation he'll revisit if compute opens up above 3e19.

Feeding that table is the refit LR scaling law in #5951, now landed in heuristic.py as muonh_lr = 18.31 · tokens^-0.395 · dim^-0.150 · batch_size^0.5 (R²=0.996 over 17 cells). Larry caught and corrected a real bug along the way: the first fit fed tokens = R·p_active with intermediate_dim = d×4 while the actual heuristic uses d/2, inflating the coefficient ~1.74×—worth noting because the surprising ~60% LR jump it initially implied turned out to be partly that units bug, not a real shift. The corrected law still raises LR ~15–20% over v16 at the optima and steepens the token exponent (−0.395 vs −0.281), which in turn pushes the June compute-optimal points to ~1.7–1.8× larger budgets per dim with ~20–30% lower LR. The notable absence is preregistration: we opened #6046 to determine the path to register a loss guess for the June '26 run, and in #news Percy Liang pressed to actually name and version the dataset (marin-data-0.1) the June model will train on.

The optimizer search that has been running under agent.md autonomy mostly served to lock the recipe by ruling alternatives out. MuonEq #6066 failed gate 1—its pre-orthogonalization equilibration costs 5–6% throughput for no loss gain (effective speedup 0.94 at d512); the AMUSE schedule-free + hyperball tuning #6064 was closed not-planned after failing to beat the MuonH baseline; Kaiyue Wen's KLSOAPH port #5728 got an upstream-parity audit that found five silently-simplified primitives, and a full-matrix SinkSOAP reparam variant #6082 is now training stably at precond_freq=10 (~292k tok/s, ~13% under MuonH's 333k). On the architecture side the sandwich gated post-norm #5938, GLU-vs-SwiGLU #5934, and parallel attn/MLP layouts #5932 all failed gate 1 (parallel layouts bought 7–10% throughput but the loss regression swamped it), and the muon_epsilon sweep #5933 confirmed the recipe is insensitive there—so the default 1e-8 stays and the knob is kept out of the scaling heuristic. None of these displaced the canonical baseline, which is exactly what "land the recipe" needs.

Tooling and the GPU hand-off filled in the rest. David Hall merged final-only checkpoint retention for the Grug launchers #5945 and factored the whole MoE experiment workflow—gates, effective-speedup math, Iris submission, W&B monitoring—into a reusable agent skill #6068, which is what has let the gate-1 experiment fan-out run largely hands-off. The open #6069 reshards the MoE block back to batch/expert layout before the experts run, removing the attention-to-MLP layout cliff under FA4; the motivation showed up starkly in #gpu, where dlwh's B200 profiling found Grug crushing Megatron at d1024 (373k vs 166k tok/s) but collapsing at d5120 (14–24k tok/s)—a routing-layout problem he believes is addressable, and one that matters because the June hero run is a GPU run. Larry also wrote up the recipe's scaling philosophy in #news: keep expert count at 4-active-of-256 (×sparsity always wins on loss; granularity past 4 active hurts MFU), ~1 layer per 100 dims, and init_std = 0.5/hidden_dim to hold activation magnitude constant as width scales. Externally, Poolside's Laguna report—flagged by Larry in #news—independently uses the identical lr = a·tokens^b·size^c·bs^0.5 transfer formula and a proxy-swarm data-mixing approach, a useful convergent signal that the recipe is on a defensible path.

0 PRs this week, 32 new comments, and 4 new issues (50 total)
Sort:
12 autocategorized

#5357 FA4 THD attention beats CuTe 2x on B200; d5120 MoE still 4-6x behind Megatron

Epic title: H100 kernel perf


Progress: 25% Risk: ● Off track
Real kernel progress landed this week — the FA4 THD probe roughly halves attention-core time versus the current CuTe path (#6039, #6059) — but the first fair B200 vs Megatron profile shows Grug at d5120/L8 running 4-6x behind Megatron (about 14-24k vs 91.6k tok/s), with MoE routing and ragged_all_to_all as the dominant gap #5815. The def-of-done targets Nemotron-class H100 MFU, work is still on B200 and far from parity, and the parent #6049 June 15 deadline makes the remaining gap a schedule risk.

Summary: Def of done: Get to Nemotron +/- ε MFU on H100s

Last week's FA4/CuTe attention backend crossed Transformer Engine on a single B200 shape; this week David Hall pushed on the THD/varlen direction and produced the first full-node B200 profile against Megatron. The headline attention result came out of the probe issue #6039 and the draft backend #6059: reshaping packed BSHD to compact THD with cu_seqlens built from contiguous segment IDs and calling upstream FlashAttention-4 CuTe varlen kernels roughly halves attention-core time on the exact d5120 Grug shape (B=1, S=4096, Hq=40, Hkv=8, D=128, four packed segments), measuring 0.000665s fwd+bwd versus 0.001308s for the current segmented FA4/CuTe path. Getting there took a long sequence of false starts documented in the probe thread: a private JAX cuDNN packed-offset path looked fastest but failed numerics on the GQA shape (Hq=40, Hkv=8) and was discarded, and the win only held once timing used FA4's internal fwd/bwd APIs and cu_seqlens was built ahead of the timed core. The integration in #6059 evolved from a Torch/DLPack bridge to a jax.pure_callback path and finally to a direct cutlass.jax.cutlass_call into the CuTe SM100 kernels, leaving no Torch or DLPack on the JAX-facing path; it remains narrow by design (causal self-attention, fixed max-segment metadata, SM100/SM110, head_dim 128, GQA).

The week's most consequential finding came from the reproducible BF16 B200 baseline #5815, which moved off the old single-GH200 synthetic profile to fair-fit d5120/L8 full-node runs. With Megatron given its native stack (Transformer Engine, flash attention, native DeepEP), the gap is stark: Megatron reaches about 91.6k post-profile tokens/s at d5120/L8 while Grug reaches only about 14.2k with ragged_all_to_all and about 24.4k with a ring/all-gather comparator. David Hall summarized the scale crossover in #gpu: at d1024 Grug is far ahead (373k vs 166k tok/s), but at d5120 Megatron scales much better, and the regression traces mostly to MoE routing. The conclusion in #5815 is that ragged_all_to_all is a real bottleneck but not the whole gap, motivating per-block forward/backward timing across attention, routing, dispatch+MLP, and optimizer.

Two new design issues were carved out of the THD work to push the layout story end to end. #6071 proposes making the Grug block's internal residual stream token-major x_td [T, D] with THD metadata as the attention ABI so attention and MoE share an execution layout, keeping external APIs BSD and measuring pack/unpack and reshard costs separately; its TPU subissue #6072 tracks a packed-THD TPU attention kernel that skips inactive blocks rather than treating the batch as dense padded BSD. The B200 MoE GMM tuning issue #5894 and the segmented-attention issue #5896 were both consolidated into managed experiment TL;DRs this week (the GMM result is a ~1.19-1.20x Blackwell BF16 forward win from an m128_n256 tile, still narrow to one shape class). The longer-horizon FP8 #6048 and pallas xent dtype (#2988, now closed/stale) threads saw no movement, and the parent #6049 sets a June 15 deadline for roughly out-of-the-box Megatron MFU.

10 autocategorized

#5356 June MoE run scoped into 10T-token plan; DeepEP backend lands in draft

Epic title: Run MoEs on multinode GPUs (H100s) on CoreWeave


Progress: 25% Risk: ● At risk
No multi-host H100 run ran this week, but the goal was concretely scoped via the 10T 120B-A2B run target #6044 and the 6/15 cluster-readiness ticket #6052, and the clean DeepEP backend #5982 is in draft. Risk is yellow because @dlwh's B200 numbers show Grug MoE at d5120 falling far behind Megatron on throughput, an unresolved gap on the path to a usable June run on the H100 cluster.

Summary: Def of done: we can train a June (16B-A2B) MoE for X (~1k) steps on 2+ GPU hosts.

No multi-host H100 run was attempted this week, so the definition-of-done — training a June 16B-A2B MoE for ~1k steps on 2+ GPU hosts — is still unmet. The week's substantive move was planning: we scoped the downstream run target #6044, a 10T-token pre-training-plus-mid-training run with a 120B-A2B MoE whose def-of-done is the run going by roughly June 19th conditional on CoreWeave working, and the matching infra-readiness ticket #6052 to make sure the 6/15 H100 cluster works and can carry the June run. Note the scale gap: the epic body still names a 16B-A2B target, while the registered run in #6044 is 120B-A2B, so the bar for "multinode on CoreWeave" has effectively been raised by the run plan it now feeds.

On the kernel side, David Hall reworked the DeepEP MoE backend into a clean, reviewable form. The earlier draft #5957 carried research logs, debug notes, and benchmark scratch files and was closed in favor of #5982, which adds only the production pieces: an implementation="deepep" selector in the Grug MoE backend, the intranode DeepEP path under lib/levanter/src/levanter/grug/_moe/ep_deepep.py, a JAX FFI package for DeepEP dispatch/combine, and a python -m levanter.kernels.deepep.preflight check with install docs. Both PRs are still drafts — #5982 tripped the review bot's draft stop condition — so neither has merged, and DeepEP is intranode rather than the cross-host transport the multinode goal ultimately needs.

The clearest risk signal came from David Hall's GPU MoE benchmarking on B200s, where MoE routing perf was "not great" at scale. At d1024 Grug beat Megatron (373k tok/s vs 166k), but at d5120/L8 the picture inverted hard: Megatron TE+DeepEP ran 91.6k tok/s while Grug's ragged path managed only 14.2k and its ring path 24.4k, a gap a missing flash comparison does not explain. Since the June run is a wide model, closing this large-d MoE throughput gap is now on the critical path to the cluster being usable, not just a perf nicety.

Two adjacent issues were also filed this week. Romain Yon opened #6041 and #6042 for GrugMoE inference support in vLLM on TPU and GPU respectively, with the GPU side not needed before #6044 lands. Separately, Ahmed Ahmed filed the Iris bug #6087, where a slow TPU create-status response causes Iris to delete a live, already-registered slice — orchestration robustness that matters most for paired train/rollout jobs but bears on any large multi-host launch.

7 autocategorized

Other Changes


A large batch of post-training data work landed outside the milestone epics, all from Tai Vu: SFT dataset additions including SYNTHETIC-2 verified SFT #5944, WebInstruct-verified #5955, OpenR1 Math 220K #5948 and filtered Open R1 code controls #5950, plus a run of Nemotron SFT lanes covering math, science, safety, finance, proofs and agentic tool-use (#5968, #5969, #5970, #5971, #5972, #5973, #5978, #5979). On the RL side, Tai Vu added GRPO and Dr.GRPO losses #5949, a k1 KL loss mode #5943, and an on-policy distillation MVP #5939, and refactored rollout decoding across vLLM and Levanter #5035. Ahmed Ahmed opened LoRA-DPO support for Levanter #4637.

The Zephyr-to-Arrow/Polars migration continued under Will Moss, with scatter internals moved to Polars and Parquet #5963 and a burndown epic #5992 tracking remaining work on RecordBatch exposure, stats reporting, and speculative execution (#5993, #5994, #5995, #5998). On tooling and CI, Russell Power set up grouped weekly Dependabot config for uv and npm #6027 (driving a wave of dependency bumps), drove code quality from the advanced CodeQL workflow #6035, added an ml-slop-test lint detector for low-value agent-written tests #6022, lifted function-local imports to module top to break import cycles #6007, and stood up PyPI trusted publishing for the marin-* libs #5870 — the latter prompted by marin-finelog being missing from PyPI and breaking installs #5867. The recurring nightshift agent landed several multi-cleanup passes and raised the slow-test threshold from 8s to 60s #5980.

Iris control-plane hardening produced a cluster of follow-up issues from Russell Power and Rafal Wojdylabot: a wedged BUILDING task that never times out and leaks capacity #6077, making reconcile batch apply order-independent #6081, push-and-forget task-status updates lacking ACK or retry #5921, autoscaler scaledown not communicated to the controller #3042, and minimum-lease handling for batch-priority tasks under preemption (#5942, #5975). Rafal Wojdylabot also designed an Artifact.from_id + ArtifactRegistry for execution #6016. Finally, we filed a set of planning issues on post-training strategy and eval direction (#6047, #6051, #6053, #6054) plus an RFC on where design docs should live #5962.

73 PRs this week, 104 new comments, and 38 issues closed (38 total)
Sort:

Community Pulse


Two contributors from outside the regular team touched the repo this week. @suraj-ranganath opened #3037, a 16-run 50k-step GPT2-small-equivalent sweep reproducing the NAMO weight-decay ablation in Marin, finding the reported endpoint reversed: NAMO/NAMO-D with AdamW-style matrix decay finished ahead of their adaptive-decay counterparts. @Rahul007007 filed #6075 to align the nightshift agent prompts with the author-pr skill, dropping the haiku ritual in favor of plain-text commit-style PR bodies and fixing a pre-existing KeyError in nightshift_doc_drift.py.

The week's Discord activity drew external researchers directly into the active optimizer and MoE work. New arrival Tim Tsz-Kit Lau shared his symmetry-compatible optimizer paper and offered to see how its per-matrix update stack performs on Marin, landing as the optimizer sweep under #5358 was ruling out MuonEq #6066 and AMUSE #6064 against the MuonH baseline. Leena Vankadara (Gatsby, UCL) shared her group's Maximally Scale-Stable Parameterization paper on scaling expert count and active experts, offering to help map its rules onto Marin's setup — directly relevant to the recipe's 4-active-of-256 choice carried into the June run #6044. On the eval side, Percy Liang connected a prospective contributor interested in adding evals, who then asked in #infra for the happy path to serve an HTTP endpoint over a multi-host JAX mesh without Ray — exactly the eval-serving shape #5368 is converging on.

Eighteen members introduced themselves, a mix of incoming PhD students, infrastructure and systems engineers, and researchers leaving frontier labs for open work — several arriving via Percy Liang's ICLR 2026 talk and citing Marin's scaling-law approach. Their stated interests cluster tightly around this milestone's open threads: optimization and pretraining data (Minhak, a Stanford PhD starting this fall), an architecture experiment Percy suggested running on Marin (shahir, also Stanford), and posttraining alignment, evals, and environment scaling (allie, recently of xAI's RL team; Vincent, formerly Anthropic pretraining/RL). By background, Tim Tsz-Kit Lau (DRW, ex-Penn) intersects directly with the optimizer-stability work driving the MuonH recipe, and Leena Vankadara brings MoE scaling theory bearing on the June expert-count decisions. Beyond the self-introductions, several recognizable researchers joined silently through the server-join stream — Martin Jaggi (EPFL), whose distributed-optimization work intersects the optimizer and MoE-throughput threads; Fares Obeid, known for Muon-family optimizer work tying into the same sweep; and Kaifeng Lyu (vfleaking), whose training-dynamics and scaling theory bears on the scaling-law direction.

The research the community circulated was dominated by scaling-law, optimizer, and MoE-parameterization work — from Poolside's Laguna report, which independently uses the same LR-transfer formula, to MoE scaling-rule and symmetry-compatible-optimizer papers offered straight into the recipe discussion.

News & research shared

Active collaborators this week

Stanford · CRFM 7 people · 3 PRs · 10 issues filed · 17 comments · 53 Discord msgs

Collaborator activity this week

Lab / Org People PRs Issues filed Comments Discord msgs Total
Stanford · CRFM 7 3 10 17 53 83
Common Crawl Foundation
Princeton · Dao Lab
GitHub activity from 107 other contributors

Will Moss · Industry (other) 6 PRs, 23 comments

  • #6038 Add only, skip and agent-output options to pre-commit 💬3 +43 −6
  • #6065 [datakit] Optimize the nemotron ferry +30 −13
  • #5963 [zephyr] Move scatter internals to Polars & Parquet 💬15 +777 −1119
  • #5442 [zephyr] Design document for moving Zephyr to Arrow / Polars 💬3 +1044 −0
  • #5481 [zephyr] Move scatter internals to Polars 💬1 +710 −738
  • #6001 [tokenize] Reduce memory pressure in `_exemplar_for` 💬7 +3 −7
23 comments on 13 threads
  • #5963 [zephyr] Move scatter internals to Polars & Parquet ×6
  • #6001 [tokenize] Reduce memory pressure in `_exemplar_for` ×4
  • #6038 Add only, skip and agent-output options to pre-commit ×3
  • #6007 Lift function-local imports to module top; break small import cycles
  • #5910 [zephyr] split execution.py into a package
  • #5861 [zephyr] Experiment: Dataset can read SQL queries via DataFusion
  • #5442 [zephyr] Design document for moving Zephyr to Arrow / Polars
  • #5481 [zephyr] Move scatter internals to Polars
  • #5241 [Design] stats_service
  • #5352 Zephyr Performance Improvements
  • #5353 Zephyr: Better Resource Utilization per Stage
  • #5909 [iris] Move SetTaskStatusText off the controller onto the finelog stats service
  • #5992 [zephyr][epic] Zephyr Burndown

Benjamin Feuer · NYU · (other) 0 PRs, 24 Discord msgs

Tai Vu · Stanford · (other) 19 PRs, 2 comments

  • #5944 [SFT] Add SYNTHETIC-2 verified SFT dataset +73 −9
  • #5035 [rl] Refactor rollout decoding across vLLM and Levanter 💬4 +1147 −342
  • #5979 [datakit] Add Nemotron Agentic tool-use data +562 −0
  • #5978 [posttrain] Add Nemotron SFT instruction-following chat v2 +181 −2
  • #5973 [Posttrain] Add AceReason 1.1 SFT views +205 −0
  • #5972 [posttrain] Add Nemotron finance SFT dataset +113 −2
  • #5971 [posttrain] Add NVIDIA math reasoning SFT dataset views +478 −7
  • #5970 [posttrain] Add Nemotron Math Proofs SFT dataset +160 −6
  • #5969 [posttrain] Add Nemotron Science SFT lanes +140 −5
  • #5968 [posttrain] Add Nemotron safety SFT dataset +65 −0
  • #5955 [SFT] Add WebInstruct-verified dataset +77 −12
  • #5950 [SFT] Add filtered Open R1 code dataset controls +345 −1
  • #5949 [RL] Add GRPO and Dr.GRPO losses 💬1 +592 −84
  • #5948 [SFT] Add OpenR1 Math 220K dataset +111 −13
  • #5947 [posttrain] Add Dolci Think Python SFT dataset +74 −0
  • #5943 [RL] Add k1 KL loss mode +50 −11
  • #5939 [RL] Add sampled-token on-policy distillation MVP 💬1 +1740 −10
  • #4858 [datasets] Add Hermes trace support to the SFT pipeline 💬3 +563 −43
  • #4524 [RL] Add GPU smoke probe and resource-aware config 💬1 +1103 −213
2 comments on 2 threads
  • #5949 [RL] Add GRPO and Dr.GRPO losses
  • #5939 [RL] Add sampled-token on-policy distillation MVP

Al (@alxrms) · Unclassified 0 PRs, 9 Discord msgs

Yiyuan Li · UNC · (other) 0 PRs, 5 comments

5 comments on 3 threads
  • #5863 [evals] Add FEVER factuality PPL slice ×3
  • #5934 Agent MoE Experiment: GLU vs SwiGLU activation (gate 1)
  • #5821 Experiment: Measure MoE sensitivity to tokenizer choice

Rabrg · Unclassified 0 PRs, 5 Discord msgs

Declan Michaels · Unclassified 0 PRs, 5 Discord msgs

MoZayed · Unclassified 0 PRs, 4 Discord msgs

Bilibird · Unclassified 0 PRs, 4 Discord msgs

leenacvankadara · Unclassified 0 PRs, 3 Discord msgs

Rahul007007 · Unclassified 1 PR, 1 comment

  • #6075 [nightshift] Align prompts with author-pr skill (plain-text PRs) 💬1 +113 −37
1 comment on 1 thread
  • #3782 nightshift: align agent prompts with PR workflow

w2sgarnav · Unclassified 0 PRs, 2 Discord msgs

HypnoPump17 · Unclassified 0 PRs, 2 Discord msgs

Leo Li · Unclassified 0 PRs, 2 Discord msgs

Minhak · Unclassified 0 PRs, 2 Discord msgs

Gerardo Salazar · Unclassified 0 PRs, 2 Discord msgs

whynot · Unclassified 0 PRs, 2 Discord msgs

Tim Tsz-Kit Lau · Unclassified 0 PRs, 2 Discord msgs

Pushpa Kumar Balan · Unclassified 0 PRs, 2 Discord msgs

shahir · Unclassified 0 PRs, 2 Discord msgs

fxiao369 · Unclassified 0 PRs, 2 Discord msgs

false · Unclassified 0 PRs, 2 Discord msgs

avinash · Unclassified 0 PRs, 2 Discord msgs

michaelf · Unclassified 0 PRs, 2 Discord msgs

dreamer.eternal · Unclassified 0 PRs, 2 Discord msgs

allie · Unclassified 0 PRs, 2 Discord msgs

lliu606 · Unclassified 0 PRs, 1 comment

1 comment on 1 thread
  • #6082 Agent MoE Experiment: SinkSOAPH (Gram-Sinkhorn preconditioner + hyperball)

jder · Unclassified 0 PRs, 1 comment

1 comment on 1 thread
  • #5990 [iris] stamp BUILD_DATE on every wheel build

valyala · Unclassified 0 PRs, 1 comment

1 comment on 1 thread
  • #5241 [Design] stats_service

suraj-ranganath · Unclassified 0 PRs

Rohith Kuditipudi · Stanford · (other) 0 PRs

Trishanu · Unclassified 0 PRs, 1 Discord msg

Swagatam · Unclassified 0 PRs, 1 Discord msg

venti · Unclassified 0 PRs, 1 Discord msg

Hitman · Unclassified 0 PRs, 1 Discord msg

Zhenfei · Unclassified 0 PRs, 1 Discord msg

Ryan Peters · Unclassified 0 PRs, 1 Discord msg

Maverick · Unclassified 0 PRs, 1 Discord msg

srikanth · Unclassified 0 PRs, 1 Discord msg

boltzmann · Unclassified 0 PRs, 1 Discord msg

rladeira · Unclassified 0 PRs, 1 Discord msg

Chang-Han Chen · Unclassified 0 PRs, 1 Discord msg

tombornal · Unclassified 0 PRs, 1 Discord msg

dhruv · Unclassified 0 PRs, 1 Discord msg

secemp · Unclassified 0 PRs, 1 Discord msg

David · Unclassified 0 PRs, 1 Discord msg

lentils. · Unclassified 0 PRs, 1 Discord msg

awwkl · Unclassified 0 PRs, 1 Discord msg

Valentin · Unclassified 0 PRs, 1 Discord msg

Yee · Unclassified 0 PRs, 1 Discord msg

Alvin Zhang · Unclassified 0 PRs, 1 Discord msg

mohdelgaar · Unclassified 0 PRs, 1 Discord msg

Jack Wind · Unclassified 0 PRs, 1 Discord msg

Aman · Unclassified 0 PRs, 1 Discord msg

blink1205 · Unclassified 0 PRs, 1 Discord msg

u1501307 · Unclassified 0 PRs, 1 Discord msg

sigh.j · Unclassified 0 PRs, 1 Discord msg

Amir Hagai · Unclassified 0 PRs, 1 Discord msg

faresobeid · Unclassified 0 PRs, 1 Discord msg

Alex · Unclassified 0 PRs, 1 Discord msg

martinjaggi · Unclassified 0 PRs, 1 Discord msg

Nan · Unclassified 0 PRs, 1 Discord msg

BioGeek · Unclassified 0 PRs, 1 Discord msg

og the gifted · Unclassified 0 PRs, 1 Discord msg

Justin Jung · Unclassified 0 PRs, 1 Discord msg

sam_acqua · Unclassified 0 PRs, 1 Discord msg

Steffi · Unclassified 0 PRs, 1 Discord msg

T-Bone · Unclassified 0 PRs, 1 Discord msg

vincent · Unclassified 0 PRs, 1 Discord msg

varun · Unclassified 0 PRs, 1 Discord msg

sword865 · Unclassified 0 PRs, 1 Discord msg

Vijai · Unclassified 0 PRs, 1 Discord msg

Jenna · Unclassified 0 PRs, 1 Discord msg

Badri · Unclassified 0 PRs, 1 Discord msg

Austin · Unclassified 0 PRs, 1 Discord msg

killabeeze · Unclassified 0 PRs, 1 Discord msg

abishop · Unclassified 0 PRs, 1 Discord msg

posaka · Unclassified 0 PRs, 1 Discord msg

bebc · Unclassified 0 PRs, 1 Discord msg

pulkit · Unclassified 0 PRs, 1 Discord msg

riley · Unclassified 0 PRs, 1 Discord msg

lesturge · Unclassified 0 PRs, 1 Discord msg

AyayaFan · Unclassified 0 PRs, 1 Discord msg

Xuchen Gong · Unclassified 0 PRs, 1 Discord msg

Nandan · Unclassified 0 PRs, 1 Discord msg

JarrodBarnes · Unclassified 0 PRs, 1 Discord msg

Miguel Ramos · Unclassified 0 PRs, 1 Discord msg

Dr.N0 · Unclassified 0 PRs, 1 Discord msg

Woody · Unclassified 0 PRs, 1 Discord msg

roundrobin · Unclassified 0 PRs, 1 Discord msg

Luna · Unclassified 0 PRs, 1 Discord msg

vertimento · Unclassified 0 PRs, 1 Discord msg

vfleaking · Unclassified 0 PRs, 1 Discord msg

H · Unclassified 0 PRs, 1 Discord msg

mnaeraxr · Unclassified 0 PRs, 1 Discord msg

Wenbo Chen · Unclassified 0 PRs, 1 Discord msg

stephwang · Unclassified 0 PRs, 1 Discord msg

PengfeiLi · Unclassified 0 PRs, 1 Discord msg

Ahhyun Lucy Lee · Unclassified 0 PRs, 1 Discord msg

Roy1 · Unclassified 0 PRs, 1 Discord msg

rgvb_ · Unclassified 0 PRs, 1 Discord msg

sippycoder · Unclassified 0 PRs, 1 Discord msg

노란토끼 · Unclassified 0 PRs, 1 Discord msg

V. · Unclassified 0 PRs, 1 Discord msg

elie · Unclassified 0 PRs, 1 Discord msg

dokhyon · Unclassified 0 PRs, 1 Discord msg

Sankalp Jajee · Unclassified 0 PRs, 1 Discord msg

Agent MoE speedup


Completed marin-community/marin_moe runs, grouped by Agent MoE budget. Speedup is relative to the original baseline run for each budget and charges each variant by its actual reported FLOPs. Best observed point is 4.20× from muonh-may-recipe-lr-v1-d1280-R60-lr1p0.

baseline (1×) this week's runs older runs running best higher is better
d512 / 2.19e17 FLOPs
100 completed runs; 64 this period
baseline loss 3.8104
1x amuse-d512-2.19e17-sweep-C-r3-r-1.0-b0.9-c1200-lr0.5-w0.01-r1-ZY: 0.25x, loss 3.8649, May 29 amuse-d512-2.19e17-sweep-C-r3-r-2.0-b0.9-c1200-lr0.5-w0.01-r2-ZY: 0.23x, loss 3.8868, May 29 amuse-d512-2.19e17-sweep-C-r3-lr-1.0x-b0.9-c1200-lr1-w0.01-r0-ZY: 0.15x, loss 3.9698, May 29 amuse-d512-2.19e17-sweep-C-r3-lr-0.7x-b0.9-c1200-lr0.7-w0.01-r0-ZY: 0.17x, loss 3.9465, May 29 amuse-d512-2.19e17-sweep-C-r3-warmup-0.02-b0.9-c1200-lr0.5-w0.02-r0-ZY: 0.32x, loss 3.8144, May 29 amuse-d512-2.19e17-sweep-C-r3-lr-0.3x-b0.9-c1200-lr0.3-w0.01-r0-ZY: 0.22x, loss 3.8877, May 29 amuse-d512-2.19e17-sweep-B-r1-anchor-b0.9-c1600-lr1-w0.01-r0-zY: 0.23x, loss 3.8870, May 29 amuse-d512-2.19e17-sweep-B-r1-beta-0.5-b0.5-c1600-lr1-w0.01-r0-zY: 0.04x, loss 4.3195, May 29 amuse-d512-2.19e17-sweep-B-r1-r-2.0-b0.9-c1600-lr1-w0.01-r2-zY: 0.17x, loss 3.9552, May 29 amuse-d512-2.19e17-sweep-B-r1-r-1.0-b0.9-c1600-lr1-w0.01-r1-zY: 0.19x, loss 3.9201, May 29 amuse-d512-2.19e17-sweep-B-r1-beta-0.99-b0.99-c1600-lr1-w0.01-r0-zY: 0.18x, loss 3.9315, May 29 amuse-d512-2.19e17-sweep-B-r1-beta-0.7-b0.7-c1600-lr1-w0.01-r0-zY: 0.09x, loss 4.1015, May 29 amuse-d512-2.19e17-sweep-B-r1-cwarm-1200-b0.9-c1200-lr1-w0.01-r0-zY: 0.21x, loss 3.9024, May 29 amuse-d512-2.19e17-sweep-B-r1-beta-0.95-b0.95-c1600-lr1-w0.01-r0-zY: 0.25x, loss 3.8665, May 29 amuse-d512-2.19e17-sweep-B-r1-lr-0.5x-b0.9-c1600-lr0.5-w0.01-r0-zY: 0.37x, loss 3.7848, May 29 amuse-d512-2.19e17-sweep-B-r1-lr-0.7x-b0.9-c1600-lr0.7-w0.01-r0-zY: 0.27x, loss 3.8537, May 29 amuse-d512-2.19e17-sweep-B-r1-r-0.5-b0.9-c1600-lr1-w0.01-r0.5-zY: 0.20x, loss 3.9117, May 29 amuse-d512-2.19e17-sweep-B-r1-cwarm-2400-b0.9-c2400-lr1-w0.01-r0-zY: 0.20x, loss 3.9130, May 29 amuse-d512-2.19e17-sweep-B-r1-warmup-0.05-b0.9-c1600-lr1-w0.05-r0-zY: 0.23x, loss 3.8838, May 29 amuse-d512-2.19e17-sweep-B-r1-lr-1.5x-b0.9-c1600-lr1.5-w0.01-r0-zY: 0.16x, loss 3.9670, May 29 amuse-d512-2.19e17-sweep-B-r1-warmup-0.02-b0.9-c1600-lr1-w0.02-r0-zY: 0.23x, loss 3.8782, May 29 amuse-d512-2.19e17-sweep-B-r1-cwarm-800-b0.9-c800-lr1-w0.01-r0-zY: 0.23x, loss 3.8853, May 29 amuse-d512-2.19e17-sweep-B-r2-beta-0.95-b0.95-c1600-lr1-w0.01-r0-zY: 0.23x, loss 3.8878, May 29 amuse-d512-2.19e17-sweep-B-r2-beta-0.97-b0.97-c1600-lr1-w0.01-r0-zY: 0.24x, loss 3.8724, May 29 amuse-d512-2.19e17-sweep-B-r2-cwarm-1200-b0.99-c1200-lr1-w0.01-r0-zY: 0.18x, loss 3.9390, May 29 amuse-d512-2.19e17-sweep-B-r2-r-0.5-b0.99-c1600-lr1-w0.01-r0.5-zY: 0.17x, loss 3.9459, May 29 amuse-d512-2.19e17-sweep-B-r2-beta-0.995-b0.995-c1600-lr1-w0.01-r0-zY: 0.12x, loss 4.0314, May 29 amuse-d512-2.19e17-sweep-B-r2-beta-0.9-b0.9-c1600-lr1-w0.01-r0-zY: 0.20x, loss 3.9096, May 29 amuse-d512-2.19e17-sweep-B-r2-warmup-0.05-b0.99-c1600-lr1-w0.05-r0-zY: 0.19x, loss 3.9300, May 29 amuse-d512-2.19e17-sweep-B-r2-cwarm-800-b0.99-c800-lr1-w0.01-r0-zY: 0.20x, loss 3.9092, May 29 amuse-d512-2.19e17-sweep-B-r2-beta-0.999-b0.999-c1600-lr1-w0.01-r0-zY: 0.04x, loss 4.3208, May 29 amuse-d512-2.19e17-sweep-B-r2-lr-0.3x-b0.99-c1600-lr0.3-w0.01-r0-zY: 0.17x, loss 3.9517, May 29 amuse-d512-2.19e17-sweep-B-r2-lr-1.5x-b0.99-c1600-lr1.5-w0.01-r0-zY: 0.16x, loss 3.9695, May 29 amuse-d512-2.19e17-sweep-B-r2-cwarm-2400-b0.99-c2400-lr1-w0.01-r0-zY: 0.15x, loss 3.9790, May 29 amuse-d512-2.19e17-sweep-B-r2-lr-0.5x-b0.99-c1600-lr0.5-w0.01-r0-zY: 0.23x, loss 3.8864, May 29 amuse-d512-2.19e17-sweep-B-r2-lr-0.7x-b0.99-c1600-lr0.7-w0.01-r0-zY: 0.19x, loss 3.9263, May 29 amuse-d512-2.19e17-sweep-B-r2-warmup-0.02-b0.99-c1600-lr1-w0.02-r0-zY: 0.16x, loss 3.9627, May 29 amuse-d512-2.19e17-sweep-B-r2-r-2.0-b0.99-c1600-lr1-w0.01-r2-zY: 0.10x, loss 4.0802, May 29 amuse-d512-2.19e17-sweep-B-r2-r-1.0-b0.99-c1600-lr1-w0.01-r1-zY: 0.13x, loss 4.0087, May 30 amuse-d512-2.19e17-sweep-B-r3-cwarm-1200-b0.99-c1200-lr0.5-w0.01-r0-zY: 0.23x, loss 3.8866, May 30 amuse-d512-2.19e17-sweep-B-r3-lr-0.4x-b0.99-c1600-lr0.4-w0.01-r0-zY: 0.21x, loss 3.9073, May 30 amuse-d512-2.19e17-sweep-B-r3-warmup-0.05-b0.99-c1600-lr0.5-w0.05-r0-zY: 0.20x, loss 3.9166, May 30 amuse-d512-2.19e17-sweep-B-r3-cwarm-800-b0.99-c800-lr0.5-w0.01-r0-zY: 0.21x, loss 3.9073, May 30 amuse-d512-2.19e17-sweep-B-r3-beta-0.97-b0.97-c1600-lr0.5-w0.01-r0-zY: 0.28x, loss 3.8420, May 30 amuse-d512-2.19e17-sweep-B-r3-lr-0.3x-b0.99-c1600-lr0.3-w0.01-r0-zY: 0.17x, loss 3.9521, May 30 amuse-d512-2.19e17-sweep-B-r3-r-2.0-b0.99-c1600-lr0.5-w0.01-r2-zY: 0.13x, loss 4.0052, May 30 amuse-d512-2.19e17-sweep-B-r3-beta-0.999-b0.999-c1600-lr0.5-w0.01-r0-zY: 0.05x, loss 4.2173, May 30 amuse-d512-2.19e17-sweep-B-r3-beta-0.95-b0.95-c1600-lr0.5-w0.01-r0-zY: 0.31x, loss 3.8234, May 30 amuse-d512-2.19e17-sweep-B-r3-cwarm-2400-b0.99-c2400-lr0.5-w0.01-r0-zY: 0.22x, loss 3.8948, May 30 muoneqh-d512-2.19e17-muoneqh-combined-e-0.25: 0.66x, loss 3.6613, May 30 muoneqh-d512-2.19e17-muoneqh-combined-e-0.5: 0.68x, loss 3.6537, May 30 sinksoaph-d512-2.19e17-sinksoaph-combined: 0.14x, loss 3.9621, May 31 sinksoaph-d512-2.19e17-sinksoaph-combined-lr-3.0x-lr3-w0.01: 0.07x, loss 4.1087, May 31 sinksoaph-d512-2.19e17-sinksoaph-combined-lr-0.25x-lr0.25-w0.01: 0.15x, loss 3.9472, May 31 sinksoaph-d512-2.19e17-sinksoaph-combined-lr-0.5x-lr0.5-w0.01: 0.17x, loss 3.9259, May 31 sinksoaph-d512-2.19e17-sinksoaph-combined-lr-2.0x-lr2-w0.01: 0.09x, loss 4.0576, May 31 sinksoaph-d512-2.19e17-sinksoaph-combined-warmup-0.05-lr1-w0.05: 0.16x, loss 3.9371, May 31 sinksoaph-d512-2.19e17-sinksoaph-combined-warmup-0.1-lr1-w0.1: 0.19x, loss 3.8953, May 31 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-T10-lr1-w0.1-T10: 0.20x, loss 3.8859, May 31 moe_may_compute_opt_d512: 0.80x, loss 3.5416, May 31 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-warmup-0.15-lr1-w0.15-T10: 0.20x, loss 3.8906, May 31 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-lr-0.25x-lr0.25-w0.1-T10: 0.11x, loss 4.0184, May 31 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-warmup-0.2-lr1-w0.2-T10: 0.21x, loss 3.8812, May 31 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-warmup-0.05-lr1-w0.05-T10: 0.18x, loss 3.9052, May 31 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-lr-0.7x-lr0.7-w0.1-T10: 0.22x, loss 3.8661, Jun 1 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-lr-1.5x-lr1.5-w0.1-T10: 0.15x, loss 3.9513, Jun 1 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-warmup-0.02-lr1-w0.02-T10: 0.17x, loss 3.9243, Jun 1 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-gb-0.9-lr1-w0.1-gb0.9-T10: 0.18x, loss 3.9064, Jun 1 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-lr-2.0x-lr2-w0.1-T10: 0.13x, loss 3.9875, Jun 1 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-lr-3.0x-lr3-w0.1-gb0.95-T10: 0.09x, loss 4.0801, Jun 1 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-warmup-0.3-lr1-w0.3-gb0.95-T10: 0.22x, loss 3.8705, Jun 1 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-warmup-0.005-lr1-w0.005-gb0.95-T10: 0.14x, loss 4.0098, Jun 1 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-warmup-0.4-lr1-w0.4-gb0.95-T10: 0.20x, loss 3.8877, Jun 1 sinksoap-reparam-muon-d512-2.19e17-sinksoap-reparam-muon-anchor-hp-lr0.7-w0.1-gb0.95-T10: 0.41x, loss 3.7203, Jun 1 sinksoap-reparam-d512-2.19e17-sinksoap-reparam-lr0.7-w0.15-lr0.7-w0.15-gb0.95-T10: 0.22x, loss 3.8636, Jun 1 sinksoap-reparam-muon-d512-2.19e17-sinksoap-reparam-muon-ns-lr0.5-lr0.5-w0.1-gb0.95-T10: 0.34x, loss 3.7629, Jun 1 sinksoap-reparam-muon-d512-2.19e17-sinksoap-reparam-muon-ns-w0.3-lr0.7-w0.3-gb0.95-T10: 0.37x, loss 3.7430, Jun 1 sinksoap-reparam-muon-d512-2.19e17-sinksoap-reparam-muon-ns-lr1.0-lr1-w0.1-gb0.95-T10: 0.46x, loss 3.6983, Jun 1 sinksoap-reparam-muon-d512-2.19e17-sinksoap-reparam-muon-ns-w0.2-lr0.7-w0.2-gb0.95-T10: 0.41x, loss 3.7230, Jun 1 sinksoap-reparam-muon-d512-2.19e17-sinksoap-reparam-muon-muonh-hp-lr1-w0.01-gb0.95-T10: 0.47x, loss 3.6953, Jun 1 sinksoap-reparam-muon-d512-2.19e17-sinksoap-reparam-muon-ns-w0.05-lr0.7-w0.05-gb0.95-T10: 0.45x, loss 3.7058, Jun 1 sinksoap-reparam-muon-d512-2.19e17-sinksoap-reparam-muon-ns-w0.02-lr0.7-w0.02-gb0.95-T10: 0.45x, loss 3.7023, Jun 1 sinksoap-reparam-muon-d512-2.19e17-sinksoap-reparam-muon-ns-lr1.0-w0.005-lr1-w0.005-gb0.95-T10: 0.48x, loss 3.6917, Jun 1 sinksoap-reparam-muon-d512-2.19e17-sinksoap-reparam-muon-ns-lr1.5-lr1.5-w0.1-gb0.95-T10: 0.40x, loss 3.7262, Jun 1 soap-reparam-muon-d512-2.19e17-soap-reparam-muon-soap-anchor-lr1-w0.005-gb0.95-b20.95-T10: 0.58x, loss 3.6595, Jun 2 soap-reparam-muon-d512-2.19e17-soap-reparam-muon-b2-0.99-lr1-w0.005-gb0.95-b20.99-T10: 0.59x, loss 3.6574, Jun 2 soap-reparam-muon-d512-2.19e17-soap-reparam-muon-lr1.5-lr1.5-w0.005-gb0.95-b20.95-T10: 0.53x, loss 3.6765, Jun 2 soap-reparam-muon-d512-2.19e17-soap-reparam-muon-nest-off-lr1-w0.005-gb0.95-b20.95-T10: 0.58x, loss 3.6606, Jun 2 soap-reparam-muon-d512-2.19e17-soap-reparam-muon-w0.002-lr1-w0.002-gb0.95-b20.95-T10: 0.58x, loss 3.6587, Jun 2 soap-reparam-muon-d512-2.19e17-soap-reparam-muon-b2-0.999-lr1-w0.005-gb0.95-b20.999-T10: 0.60x, loss 3.6557, Jun 2 soap-reparam-muon-d512-2.19e17-soap-reparam-muon-w0.01-lr1-w0.01-gb0.95-b20.95-T10: 0.59x, loss 3.6555, Jun 2 soap-reparam-muon-d512-2.19e17-soap-reparam-muon-lr0.7-lr0.7-w0.005-gb0.95-b20.95-T10: 0.53x, loss 3.6792, Jun 2 marin-big-run-moe_may_compute_opt_d512: 1.18x, loss 3.5438, Jun 2 muonh-eman-d512-2.19e17-muonh-eman-onball-g0.99-b0.5-r0.2: 0.72x, loss 3.6507, Jun 2 muonh-tokenhyperball-d512-2.19e17-muonh-thb-e0.02-h0.05-hm0.95: 0.35x, loss 3.7960, Jun 3 muonh-tokenhyperball-d512-2.19e17-muonh-thb-e0.02-h0.02-hm0.95: 0.45x, loss 3.7484, Jun 3 muonh-tokenhyperball-d512-2.19e17-muonh-thb-e0.05-h0.02-hm0.95: 0.37x, loss 3.7871, Jun 3 muonh-tokenhyperball-d512-2.19e17-muonh-thb-e0.02-h0.01-hm0.95: 0.49x, loss 3.7300, Jun 3 muonh-tokenhyperball-d512-2.19e17-muonh-thb-e0.01-h0.02-hm0.95: 0.47x, loss 3.7396, Jun 3 muonh-eman-d512-2.19e17-muonh-eman-betaprobe-g0.99-b0.5-r0.2: 0.73x, loss 3.6507, Jun 3 May 29 Jun 3
Best
1.18× marin-big-run-moe_may_compute_opt_d512 loss 3.5438
This week
0.80× moe_may_compute_opt_d512 loss 3.5416
Baseline
moe-v16-compute-opt-d512-2.19e+17
d768 / 1.70e18 FLOPs
100 completed runs; 13 this period
baseline loss 3.4339
1x muonh-may-arch-1pct-lr-lmhead-0p3-v1-d768-1.70e18: 0.90x, loss 3.3135, May 16 muonh-may-arch-1pct-lr-lmhead-0p7-v1-d768-1.70e18: 0.97x, loss 3.3003, May 16 muonh-may-arch-1pct-lr-lmhead-0p5-v1-d768-1.70e18: 0.97x, loss 3.3019, May 16 muonh-may-arch-1pct-lr-lmhead-0p8-v1-d768-1.70e18: 0.97x, loss 3.3016, May 16 muonh-may-arch-1pct-lr-embed-0p6-v1-d768-1.70e18: 0.94x, loss 3.3074, May 16 muonh-may-arch-1pct-lr-embed-0p8-v1-d768-1.70e18: 0.95x, loss 3.3045, May 16 muonh-may-arch-1pct-lr-embed-1p0-v1-d768-1.70e18: 0.96x, loss 3.3028, May 16 muonh-may-arch-1pct-lr-embed-1p2-v1-d768-1.70e18: 0.99x, loss 3.2981, May 16 muonh-may-arch-1pct-lr-embed-1p4-v1-d768-1.70e18: 0.98x, loss 3.3001, May 16 muonh-may-arch-1pct-lr-adam-1p1-v1-d768-1.70e18: 0.97x, loss 3.3020, May 16 muonh-may-arch-1pct-lr-adam-1p5-v1-d768-1.70e18: 0.70x, loss 3.3066, May 16 muonh-may-arch-1pct-lr-adam-1p3-v1-d768-1.70e18: 0.97x, loss 3.3028, May 16 muonh-may-arch-1pct-lr-muonh-0p7-v1-d768-1.70e18: 0.90x, loss 3.3127, May 16 muonh-may-arch-1pct-lr-muonh-1p3-v1-d768-1.70e18: 0.94x, loss 3.3076, May 16 muonh-may-arch-1pct-aurorah-gn-v1-d768-1.70e18: 1.04x, loss 3.3054, May 16 muonh-may-arch-1pct-aurorah-kv-v1-d768-1.70e18: 1.06x, loss 3.3034, May 16 muonh-may-arch-1pct-aurorah-expout-v1-d768-1.70e18: 1.00x, loss 3.3013, May 16 muonh-may-arch-1pct-routedscale-2p0-v1-d768-1.70e18: 0.97x, loss 3.3010, May 16 muonh-may-arch-1pct-routedscale-3p0-v1-d768-1.70e18: 0.97x, loss 3.3021, May 16 muonh-may-arch-1pct-routedscale-2p5-v1-d768-1.70e18: 0.97x, loss 3.3010, May 16 muonh-may-arch-1pct-lr-lmhead-1p2-v1-d768-1.70e18: 0.96x, loss 3.3030, May 16 muonh-may-arch-gn-muonh-1pct-clip-v1-d768-1.70e18: 0.98x, loss 3.3008, May 16 muonh-may-arch-1pct-aurorah-expert-io-v1-d768-1.70e18: 0.89x, loss 3.2982, May 16 muonh-may-arch-1pct-routed-2p5x-half-lr-v1-d768-1.70e18: 0.88x, loss 3.3186, May 16 muonh-may-arch-1pct-finer-lr-mlpout-1p3-mlpin-0p7-v1-d768-1.70e18: 0.91x, loss 3.3121, May 17 muonh-may-arch-1pct-finer-lr-qk-0p7-vo-1p3-v1-d768-1.70e18: 0.95x, loss 3.3058, May 17 muonh-may-arch-1pct-finer-lr-mlpout-0p7-mlpin-1p3-v1-d768-1.70e18: 0.94x, loss 3.3070, May 17 muonh-may-arch-1pct-finer-lr-qk-1p3-vo-0p7-v1-d768-1.70e18: 0.98x, loss 3.3002, May 17 muonh-may-arch-gn-muonh-0pct-noclip-v1-d768-1.70e18: 0.96x, loss 3.3036, May 17 muonh-may-arch-1pct-gn-wup0p8-wdown1p2-v1-d768-1.70e18: 0.97x, loss 3.3015, May 17 muonh-may-arch-1pct-normuon-expert-io-v1-d768-1.70e18: 1.07x, loss 3.2998, May 17 muonh-may-arch-1pct-pko-segment-safe-v1-d768-1.70e18: 0.93x, loss 3.3081, May 17 muonh-may-arch-1pct-split-wupgate-v1-d768-1.70e18: 1.08x, loss 3.2990, May 17 muonh-may-arch-1pct-bos-key-embed-prerope-v1-d768-1.70e18: 0.95x, loss 3.3046, May 17 muonh-may-arch-1pct-bos-key-embed-v1-d768-1.70e18: 0.93x, loss 3.3072, May 17 muonh-may-arch-1pct-routing-renorm-x1p0-v1-d768-1.70e18: 0.97x, loss 3.3017, May 17 muonh-may-arch-1pct-shared-gate-sig-v1-d768-1.70e18: 0.97x, loss 3.3006, May 18 muonh-may-arch-1pct-pko-bos-zero-v1-d768-1.70e18: 0.93x, loss 3.3082, May 18 muonh-may-arch-1pct-routing-renorm-x4p0-v1-d768-1.70e18: 1.00x, loss 3.2969, May 18 muonh-may-arch-1pct-no-pko-v1-d768-1.70e18: 0.85x, loss 3.3224, May 18 muonh-may-arch-per-expert-lr-gate1-shrink-expert-retry2-d768-1.70e18: 0.43x, loss 3.4379, May 18 muonh-may-arch-per-expert-lr-gate1-boost-nonexpert-retry2-d768-1.70e18: 0.46x, loss 3.4276, May 18 muonh-may-arch-per-expert-lr-gate1-mid-ratio-retry2-d768-1.70e18: 0.52x, loss 3.4065, May 18 muonh-may-arch-1pct-routing-renorm-x2p0-v1-d768-1.70e18: 0.99x, loss 3.2988, May 18 muonh-may-arch-1pct-routing-renorm-x2p5-v1-d768-1.70e18: 1.01x, loss 3.2959, May 18 muonh-may-arch-1pct-routing-renorm-x3p0-v1-d768-1.70e18: 1.00x, loss 3.2975, May 18 muonh-may-arch-1pct-embed-adam-lr-1p0-v1-d768-1.70e18: 0.97x, loss 3.3015, May 18 muonh-may-arch-1pct-embed-adam-lr-0p7-v1-d768-1.70e18: 0.96x, loss 3.3031, May 18 muonh-may-arch-1pct-embed-adam-lr-1p3-v1-d768-1.70e18: 0.97x, loss 3.3009, May 18 muonh-may-arch-1pct-pko-first-bos-zero-v1-d768-1.70e18: 0.94x, loss 3.3061, May 18 muonh-may-arch-per-expert-lr-gate1-shrink-expert-retry3-d768-1.70e18: 0.43x, loss 3.4381, May 18 muonh-may-arch-1pct-pko-first-bos-zero-no-q-norm-v1-d768-1.70e18: 0.93x, loss 3.3090, May 18 muonh-may-arch-1pct-combined-combined-v1-d768-1.70e18: 0.91x, loss 3.2998, May 18 muonh-may-arch-1pct-combined-baseline-v1-d768-1.70e18: 0.96x, loss 3.3028, May 18 muonh-may-arch-1pct-combined-no-arch-v1-d768-1.70e18: 0.95x, loss 3.3040, May 18 muonh-may-recipe-lr-v1-d768-R10-lr1p3: 0.65x, loss 3.6411, May 20 muonh-may-recipe-lr-v1-d768-R10-lr0p4: 0.24x, loss 3.8372, May 20 context-norm-gate1-v1-d768-1.70e18: 0.10x, loss 3.7711, May 20 muonh-may-recipe-lr-v1-d768-R10-lr1p0: 0.66x, loss 3.6371, May 20 muonh-may-recipe-lr-v1-d768-R10-lr1p6: 0.60x, loss 3.6571, May 20 context-norm-no-xsa-gate1-v1-d768-1.70e18: 0.54x, loss 3.4478, May 20 muonh-may-recipe-lr-v1-d768-R10-lr0p7: 0.56x, loss 3.6702, May 20 muonh-may-recipe-lr-v1-d768-R20-lr0p4: 0.47x, loss 3.5684, May 20 muonh-may-recipe-lr-v1-d768-R20-lr0p7: 0.78x, loss 3.4788, May 21 muonh-may-recipe-lr-v1-d768-R20-lr1p3: 0.84x, loss 3.4659, May 21 muonh-may-recipe-lr-v1-d768-R20-lr1p0: 0.85x, loss 3.4634, May 21 muonh-may-recipe-lr-v1-d768-R20-lr1p6: 0.75x, loss 3.4844, May 21 muonh-may-arch-1pct-pko-shift32-v1-d768-1.70e18: 1.00x, loss 3.3123, May 21 grug_moe_mix_v4_path_r1_t050_d768-1.70e+18: 0.67x, loss 3.4114, May 21 grug_moe_mix_v4_path_r1_t025_d768-1.70e+18: 0.71x, loss 3.4020, May 21 muonh-may-arch-1pct-pko-split-distance-v1-d768-1.70e18: 1.00x, loss 3.3131, May 21 grug_moe_mix_v4_path_r1_t075_d768-1.70e+18: 0.63x, loss 3.4215, May 21 muonh-may-recipe-lr-v1-d768-R60-lr0p4: 0.70x, loss 3.3120, May 21 muonh-may-recipe-lr-v1-d768-R60-lr1p3: 0.90x, loss 3.2708, May 21 muonh-may-recipe-lr-v1-d768-R60-lr0p7: 0.92x, loss 3.2678, May 21 muonh-may-recipe-lr-v1-d768-R60-lr1p0: 0.94x, loss 3.2647, May 21 muonh-may-recipe-lr-v1-d768-R60-lr1p6: 0.83x, loss 3.2846, May 21 muonh-may-arch-1pct-flag-subset-routing-embed-split-v1-d768-1.70e18: 0.98x, loss 3.2987, May 21 muonh-may-arch-1pct-flag-subset-routing-embed-pko-v1-d768-1.70e18: 0.96x, loss 3.3034, May 22 muonh-may-arch-1pct-flag-subset-routing-embed-v1-d768-1.70e18: 0.99x, loss 3.2978, May 22 grug-moe-parallel-all-d768-1.70e18-v1: 0.78x, loss 3.3538, May 24 grug-moe-parallel-half-d768-1.70e18-v1: 0.95x, loss 3.3212, May 24 grug-moe-direct-d768-1.70e18-v1: 1.04x, loss 3.3052, May 24 grug-moe-glu-sigmoid-d768-1.70e18-v1: 0.63x, loss 3.3510, May 24 hrm-repro-d768-4.00e10-v7-shards200: 0.00x, loss 4.9800, May 24 tokenizer-sensitivity-moe-d768-tokenmonster-englishcode-32k: 0.26x, loss 3.7992, May 26 tokenizer-sensitivity-moe-d768-llama3-128k: 0.60x, loss 3.5041, May 26 tokenizer-sensitivity-moe-d768-marin-128k: 0.60x, loss 3.5032, May 26 tokenizer-sensitivity-moe-d768-qwen3-152k: 0.91x, loss 3.4078, May 26 tokenizer-sensitivity-moe-d768-gpt-oss-200k: 0.33x, loss 3.5410, May 26 tokenizer-sensitivity-moe-d768-gemma3-262k: 0.53x, loss 3.3117, May 26 grug-moe-may-recipe-newlr-d768-newlr-v2: 1.05x, loss 3.3033, May 27 grug-moe-isoflop-v18-d768-v1: 0.86x, loss 3.4021, May 27 grug-moe-isoflop-v3e18-d768-v1: 1.12x, loss 3.2209, May 28 grug-moe-plain-muon-d768-3e18-v1: 2.58x, loss 3.2584, May 29 grug-moe-lmhead-adam-d768-3e18-v1: 3.35x, loss 3.2174, May 29 muoneqh-d768-1.70e18-muoneqh-combined-e-0.5: 0.90x, loss 3.3055, May 30 muoneqh-d768-1.70e18-muoneqh-combined-e-0.25: 0.85x, loss 3.3140, May 30 moe_may_compute_opt_d768: 1.05x, loss 3.2261, Jun 2 marin-big-run-moe_may_compute_opt_d768: 1.41x, loss 3.2330, Jun 2 May 16 Jun 2
Best
3.35× grug-moe-lmhead-adam-d768-3e18-v1 loss 3.2174
This week
3.35× grug-moe-lmhead-adam-d768-3e18-v1 loss 3.2174
Baseline
moe-v16-compute-opt-d768-1.70e+18
d1024 / 9.00e18 FLOPs
97 completed runs; 12 this period
baseline loss 3.1605
1x muonh-matrix-baseline-adam-mask-d1024-9.00e18: 0.77x, loss 3.1357, May 10 muonh-nowarmup-d1024-9.00e18: 0.84x, loss 3.1230, May 13 muonh-gn-adamh-v1-d1024-9.00e18: 0.75x, loss 3.1390, May 14 muonh-gn-adam-v1-d1024-9.00e18: 0.75x, loss 3.1392, May 14 muonh-drop-gn-attngate-v1-d1024-9.00e18: 0.69x, loss 3.1532, May 14 muonh-gn-muonh-v1-d1024-9.00e18: 0.77x, loss 3.1352, May 14 muonh-gn-adamh-2pct-warmup-v1-d1024-9.00e18: 0.82x, loss 3.1265, May 15 muonh-wdown-paired-v1-d1024-9.00e18: 0.75x, loss 3.1393, May 15 muonh-kv-adamh-v1-d1024-9.00e18: 0.76x, loss 3.1380, May 15 muonh-gn-adamh-5pct-warmup-v1-d1024-9.00e18: 0.80x, loss 3.1309, May 15 muonh-drop-gn-attngate-lr-muonh-0p7-v1-d1024-9.00e18: 0.68x, loss 3.1563, May 15 muonh-drop-gn-attngate-lr-adamh-1p3-v1-d1024-9.00e18: 0.69x, loss 3.1538, May 15 normuon-wider-axis-split-gn-adamh-no-warmup-v2-d1024-9.00e18: 0.82x, loss 3.1252, May 15 muonh-drop-gn-attngate-lr-adamh-0p7-v1-d1024-9.00e18: 0.72x, loss 3.1483, May 15 muonh-drop-gn-attngate-lr-muonh-1p3-v1-d1024-9.00e18: 0.67x, loss 3.1575, May 15 muonh-drop-gn-attngate-lr-adam-1p3-v1-d1024-9.00e18: 0.73x, loss 3.1462, May 15 muonh-may-arch-default-v1-d1024-9.00e18: 1.15x, loss 3.0574, May 15 muonh-may-arch-gn-1p3-v1-d1024-9.00e18: 1.15x, loss 3.0574, May 15 muonh-gn-muon-no-hyperball-v1-d1024-9.00e18: 0.77x, loss 3.1366, May 16 muonh-may-arch-gnwup-1p3-v1-d1024-9.00e18: 1.14x, loss 3.0587, May 16 muonh-may-arch-gnwup-0p7-v1-d1024-9.00e18: 1.18x, loss 3.0550, May 16 muonh-may-arch-gn-0p7-v1-d1024-9.00e18: 1.15x, loss 3.0576, May 16 muonh-may-arch-gn-muonh-1pct-noclip-v1-d1024-9.00e18: 1.11x, loss 3.0610, May 16 muonh-may-arch-1pct-lr-lmhead-0p9-v1-d1024-9.00e18: 1.11x, loss 3.0618, May 16 muonh-may-arch-1pct-kv-colnorm-v1-d1024-9.00e18: 1.13x, loss 3.0596, May 16 muonh-may-arch-1pct-lr-lmhead-0p8-v1-d1024-9.00e18: 1.12x, loss 3.0599, May 16 muonh-may-arch-1pct-lr-lmhead-0p7-v1-d1024-9.00e18: 1.11x, loss 3.0612, May 16 muonh-may-arch-1pct-lr-lmhead-0p5-v1-d1024-9.00e18: 1.13x, loss 3.0584, May 16 muonh-may-arch-1pct-lr-embed-0p6-v1-d1024-9.00e18: 1.08x, loss 3.0651, May 16 muonh-may-arch-1pct-lr-embed-0p8-v1-d1024-9.00e18: 1.08x, loss 3.0650, May 16 muonh-may-arch-1pct-lr-embed-1p0-v1-d1024-9.00e18: 1.11x, loss 3.0617, May 16 muonh-may-arch-1pct-lr-lmhead-0p3-v1-d1024-9.00e18: 1.09x, loss 3.0644, May 16 muonh-may-arch-1pct-lr-lmhead-1p2-v1-d1024-9.00e18: 1.12x, loss 3.0608, May 16 muonh-may-arch-1pct-routedscale-2p0-v1-d1024-9.00e18: 1.11x, loss 3.0618, May 16 muonh-may-arch-1pct-routedscale-2p5-v1-d1024-9.00e18: 1.11x, loss 3.0613, May 16 muonh-may-arch-1pct-routedscale-3p0-v1-d1024-9.00e18: 1.13x, loss 3.0593, May 16 muonh-may-arch-gn-muonh-2pct-noclip-v1-d1024-9.00e18: 1.08x, loss 3.0654, May 17 muonh-may-arch-gn-muonh-1pct-clip-v1-d1024-9.00e18: 1.12x, loss 3.0618, May 17 muonh-may-arch-1pct-aurorah-expert-io-v1-d1024-9.00e18: 1.07x, loss 3.0562, May 17 muonh-may-arch-1pct-lr-adam-1p1-v1-d1024-9.00e18: 1.10x, loss 3.0626, May 17 muonh-may-arch-1pct-lr-embed-1p2-v1-d1024-9.00e18: 1.15x, loss 3.0566, May 17 muonh-may-arch-1pct-lr-adam-1p3-v1-d1024-9.00e18: 1.09x, loss 3.0642, May 17 muonh-may-arch-1pct-lr-embed-1p4-v1-d1024-9.00e18: 1.13x, loss 3.0588, May 17 muonh-may-arch-1pct-lr-muonh-1p3-v1-d1024-9.00e18: 1.05x, loss 3.0688, May 17 muonh-may-arch-1pct-lr-adam-1p5-v1-d1024-9.00e18: 1.08x, loss 3.0651, May 17 muonh-may-arch-1pct-lr-muonh-0p7-v1-d1024-9.00e18: 1.07x, loss 3.0674, May 17 muonh-may-arch-gn-muonh-0pct-noclip-v1-d1024-9.00e18: 1.15x, loss 3.0573, May 17 muonh-may-arch-1pct-gn-wup0p8-wdown1p2-v1-d1024-9.00e18: 1.14x, loss 3.0581, May 17 muonh-may-arch-1pct-finer-lr-qk-1p2-vo-0p8-v1-d1024-9.00e18: 1.12x, loss 3.0605, May 17 muonh-may-arch-1pct-normuon-expert-io-v1-d1024-9.00e18: 1.18x, loss 3.0592, May 17 muonh-may-arch-1pct-split-wupgate-v1-d1024-9.00e18: 1.20x, loss 3.0574, May 17 muonh-may-arch-1pct-embed-adam-lr-1p0-v1-d1024-9.00e18: 1.13x, loss 3.0589, May 18 muonh-may-arch-1pct-embed-adam-lr-0p7-v1-d1024-9.00e18: 1.12x, loss 3.0599, May 18 muonh-may-arch-1pct-embed-adam-lr-1p3-v1-d1024-9.00e18: 1.12x, loss 3.0601, May 18 muonh-may-arch-per-expert-lr-gate2-mid-ratio-v1-ram256-d1024-9.00e18: 0.57x, loss 3.1569, May 19 muonh-may-arch-1pct-routing-renorm-x2p0-v1-d1024-9.00e18: 1.17x, loss 3.0544, May 19 muonh-may-arch-1pct-routing-renorm-x2p5-v1-d1024-9.00e18: 1.15x, loss 3.0562, May 19 muonh-may-arch-1pct-routing-renorm-x3p0-v1-d1024-9.00e18: 1.16x, loss 3.0561, May 19 muonh-may-arch-1pct-combined-baseline-v1-d1024-9.00e18: 1.11x, loss 3.0617, May 19 muonh-may-arch-1pct-combined-combined-v1-d1024-9.00e18: 1.04x, loss 3.0588, May 19 muonh-may-arch-1pct-combined-no-arch-v1-d1024-9.00e18: 1.10x, loss 3.0620, May 19 muonh-may-arch-1pct-aurorah-gn-v1-d1024-9.00e18: 1.13x, loss 3.0652, May 20 muonh-may-arch-1pct-aurorah-kv-v1-d1024-9.00e18: 1.18x, loss 3.0607, May 20 muonh-may-arch-1pct-aurorah-expout-v1-d1024-9.00e18: 1.12x, loss 3.0606, May 20 muonh-may-arch-1pct-pko-one-kv-head-v1-d1024-9.00e18: 1.14x, loss 3.0647, May 20 muonh-may-recipe-lr-v1-d1024-R10-lr0p4: 0.41x, loss 3.4571, May 21 grug_moe_mix_v4_path_r1_t050_d1024-9.00e+18: 0.82x, loss 3.1372, May 21 muonh-may-recipe-lr-v1-d1024-R10-lr0p7: 0.75x, loss 3.3527, May 21 muonh-may-recipe-lr-v1-d1024-R10-lr1p0: 0.82x, loss 3.3392, May 21 muonh-may-recipe-lr-v1-d1024-R10-lr1p3: 0.79x, loss 3.3452, May 21 grug_moe_mix_v4_path_r1_t025_d1024-9.00e+18: 0.82x, loss 3.1374, May 21 muonh-may-recipe-lr-v1-d1024-R10-lr1p6: 0.72x, loss 3.3606, May 21 grug_moe_mix_v4_path_r1_t075_d1024-9.00e+18: 0.75x, loss 3.1487, May 21 muonh-may-arch-1pct-gqa-q1p5-pko-only-v1-d1024-9.00e18: 1.20x, loss 3.0546, May 22 muonh-may-arch-1pct-gqa-q1p5-every-v1-d1024-9.00e18: 1.18x, loss 3.0494, May 22 muonh-may-arch-1pct-gqa-q2p0-nonpko-only-v1-d1024-9.00e18: 1.19x, loss 3.0459, May 22 muonh-may-arch-1pct-gqa-q1p5-nonpko-only-v1-d1024-9.00e18: 1.20x, loss 3.0504, May 22 muonh-may-arch-1pct-gqa-q2p0-every-v1-d1024-9.00e18: 1.16x, loss 3.0437, May 22 muonh-may-arch-1pct-gqa-q2p0-pko-only-v1-d1024-9.00e18: 1.18x, loss 3.0540, May 22 muonh-may-recipe-lr-v1-d1024-R20-lr0p4: 0.64x, loss 3.2671, May 22 muonh-may-recipe-lr-v1-d1024-R20-lr1p3: 0.95x, loss 3.2059, May 22 muonh-may-recipe-lr-v1-d1024-R20-lr1p0: 1.00x, loss 3.1984, May 22 muonh-may-recipe-lr-v1-d1024-R20-lr1p6: 0.86x, loss 3.2207, May 22 muonh-may-recipe-lr-v1-d1024-R20-lr0p7: 0.98x, loss 3.2017, May 23 tokenizer-sensitivity-moe-d1024-tokenmonster-englishcode-32k: 0.24x, loss 3.4930, May 26 tokenizer-sensitivity-moe-d1024-llama3-128k: 0.74x, loss 3.2224, May 26 tokenizer-sensitivity-moe-d1024-qwen3-152k: 1.21x, loss 3.1357, May 26 tokenizer-sensitivity-moe-d1024-gpt-oss-200k: 0.43x, loss 3.2607, May 26 hrm-repro-d1024-4.00e10-v8-eastpin: 0.00x, loss 4.5913, May 26 muonh-may-arch-1pct-flag-subset-routing-embed-v1-d1024-9.00e18: 1.14x, loss 3.0583, May 27 muonh-may-arch-1pct-flag-subset-routing-embed-pko-v1-d1024-9.00e18: 1.11x, loss 3.0608, May 27 muonh-may-arch-1pct-flag-subset-routing-embed-split-v1-d1024-9.00e18: 0.94x, loss 3.0584, May 27 grug-moe-isoflop-v1e19-d1024-v1: 3.38x, loss 3.0422, May 27 grug-moe-may-recipe-newlr-d1024-newlr-v2: 1.17x, loss 3.0613, May 28 grug-moe-isoflop-v3e18-d1024-v1: 0.92x, loss 3.2409, May 28 grug-moe-nopko-d1024-1e19-v1: 1.75x, loss 3.0559, May 29 marin-big-run-moe_may_compute_opt_d1024: 1.83x, loss 3.0297, Jun 3 May 10 Jun 3
Best
3.38× grug-moe-isoflop-v1e19-d1024-v1 loss 3.0422
This week
3.38× grug-moe-isoflop-v1e19-d1024-v1 loss 3.0422
Baseline
moe-v16-compute-opt-d1024-9.00e+18
d1280 / 2.83e19 FLOPs
30 completed runs; 7 this period
baseline loss 3.0065
1x muonh-matrix-baseline-adam-mask-d1280-2.83e19: 0.83x, loss 2.9888, May 11 muonh-nowarmup-d1280-2.83e19: 0.95x, loss 2.9706, May 13 muonh-gn-adamh-v1-d1280-2.83e19: 0.85x, loss 2.9855, May 15 muonh-may-recipe-lr-v1-d1280-R4-lr1p6: 1.51x, loss 3.4269, May 21 muonh-may-recipe-lr-v1-d1280-R4-lr0p4: 0.45x, loss 3.6450, May 21 muonh-may-recipe-lr-v1-d1280-R4-lr1p3: 1.59x, loss 3.4167, May 21 muonh-may-recipe-lr-v1-d1280-R4-lr0p7: 1.20x, loss 3.4664, May 22 muonh-may-recipe-lr-v1-d1280-R4-lr1p0: 1.59x, loss 3.4172, May 22 muonh-may-recipe-lr-v1-d1280-R20-lr0p4: 2.33x, loss 3.1066, May 22 muonh-may-recipe-lr-v1-d1280-R20-lr1p3: 3.44x, loss 3.0522, May 22 muonh-may-recipe-lr-v1-d1280-R20-lr1p0: 3.63x, loss 3.0448, May 22 muonh-may-recipe-lr-v1-d1280-R20-lr0p7: 3.42x, loss 3.0532, May 22 context-norm-no-xsa-gate2-v1-d1280-2.83e19: 0.76x, loss 3.0107, May 22 muonh-may-recipe-lr-v1-d1280-R20-lr1p6: 0.03x, loss 3.0692, May 22 grug_moe_mix_v4_path_r1_t050_d1280-2.83e+19: 0.90x, loss 2.9884, May 22 grug_moe_mix_v4_path_r1_t075_d1280-2.83e+19: 0.84x, loss 2.9962, May 22 muonh-may-recipe-lr-v1-d1280-R60-lr1p0: 4.20x, loss 2.8851, May 22 muonh-may-recipe-lr-v1-d1280-R60-lr1p6: 3.47x, loss 2.9081, May 22 muonh-may-recipe-lr-v1-d1280-R60-lr0p7: 4.18x, loss 2.8856, May 22 muonh-may-recipe-lr-v1-d1280-R60-lr0p4: 3.13x, loss 2.9211, May 22 muonh-may-recipe-lr-v1-d1280-R60-lr1p3: 3.87x, loss 2.8948, May 22 grug_moe_mix_v4_path_r1_t025_d1280-2.83e+19: 0.92x, loss 2.9851, May 24 muonh-may-recipe-lr-v1-d1280-R120-lr0p4: 3.24x, loss 2.8338, May 25 muonh-may-recipe-lr-v1-d1280-R120-lr0p7: 4.12x, loss 2.8063, May 25 muonh-may-recipe-lr-v1-d1280-R120-lr1p6: 3.32x, loss 2.8307, May 25 muonh-may-recipe-lr-v1-d1280-R120-lr1p3: 3.76x, loss 2.8162, May 25 muonh-may-recipe-lr-v1-d1280-R120-lr1p0: 3.99x, loss 2.8097, May 25 grug-moe-isoflop-v3e18-d1280-v1: 1.34x, loss 3.2983, May 27 grug-moe-isoflop-v3e19-d1280-v1: 3.27x, loss 2.9045, May 29 marin-big-run-moe_may_compute_opt_d1280: 2.04x, loss 2.8963, Jun 3 May 11 Jun 3
Best
4.20× muonh-may-recipe-lr-v1-d1280-R60-lr1p0 loss 2.8851
This week
4.12× muonh-may-recipe-lr-v1-d1280-R120-lr0p7 loss 2.8063
Baseline
moe-v16-compute-opt-d1280-2.83e+19

Top 15 runs (by FLOPs) this week (completed, running, crashed)


This week's compute was dominated by Michael Ryan's data-curation sweep, a battery of fixed-FLOP runs that pit DCLM against Nemotron and against WARC quality buckets to settle the data mixture for the June model (epics #5359 and #5360). The headline comparison is clean: at the 9e20 FLOP / 2.9B-param point, DCLM (curation-dclm_10k-expFM_natural-9e+20-d2432-L24-B256, MFU 0.47) finished at eval bpb 0.881, decisively ahead of the matched Nemotron arm (curation-nemotron_10k-expFM_natural-9e+20-d2432-L24-B256) at 1.031, still running as of this writing. The gap holds across Paloma and uncheatable-eval slices, and is widest on long-form English (ao3, bbc_news) where DCLM's cleaner web text pays off.

The sweep was rough on hardware: of the fifteen top-FLOP runs, five crashed mid-training, including the largest Nemotron arm (curation-nemotron_10k-expFM_natural-9e+21-d2432-L24-B512, the week's single biggest run at ~1.85e21 HW FLOPs across 64 v5 chips) and both 8B-param d3584 arms (curation-dclm_10k-expFM_natural-2e+21-d3584-L35-B128 and curation-dclm_10k-expFM_natural-9e+20-d3584-L35-B128, the latter dying after only 7.4 hours). The crashes left several arms without a clean endpoint, so the head-to-head verdict at 8B scale is still partial. At the small end the WARC quality-bucket triplet (high/med/low, all 69M params, low finishing best at 1.468 bpb) behaved counterintuitively, with the nominal low-quality bucket edging out high- and med-quality on held-out bpb, a result worth chasing down before the buckets feed the production swarm.

On the post-training side, Moo Jin Kim ran a matched pair of Qwen3-1.7B science-distillation jobs from Qwen3-30B-A3B-Thinking traces. The soft-label run (e3956np_soft_qwen3_1pt7b_...science_50k_pt2_4e5) and the hard-SFT run (e3956np_sft_qwen3_1pt7b_...science_50k_pt2_4e5) trained on identical data and chips, and the contrast is stark: soft-label distillation held eval bpb at 0.204 while hard SFT drove training loss to ~0.002 yet ballooned eval bpb to 0.873, a textbook case of label memorization against the teacher's logprobs. The same logprob-distillation idea is being formalized in #5939.

Two preregistration items are worth flagging. Will Held filed #6063, a pre-registered "Will Vibe Mix" that, before any training, records which 300m-swarm benchmarks the hand-tuned mixture is expected to win, lose, or move uncertainly on, derived from the DSP regression fit to the 241-run swarm. Separately, the MoE compute-optimal isoflop sweeps in #6074 (rolling up to #5358) began establishing compute-optimal points for the MuonH May recipe across 1e18-3e19 budgets, with LR set from the heuristic in #5951, to project a target shape for the June run against the v16 scaling law.

Run User Hardware(?) Hours(?) FLOP Budget(?) Loss BPB(?)
#5359 curation-dclm_10k-expFM_natural-9e+20-d2432-L24-B256 Michael Ryan TPU v5
(64 chips)
20.2h 9.00e20 model
1.93e21 HW (47%)
BPB: 0.881
#5359 curation-nemotron_10k-expFM_natural-9e+21-d2432-L24-B512 Michael Ryan TPU v5
(64 chips)
18.5h 9.11e20 model
1.85e21 HW (49%)
BPB: 1.028
#5359 curation-nemotron_10k-expFM_natural-9e+20-d2432-L24-B256 Michael Ryan TPU v5
(64 chips)
16.2h 7.42e20 model
1.57e21 HW (47%)
BPB: 1.031
curation-nemotron_10k-expFM_natural-2e+21-d2432-L24-B256 Michael Ryan TPU v5
(32 chips)
1.0d 4.50e20 model
1.23e21 HW (37%)
BPB: 1.040
#5939 e3956np_soft_qwen3_1pt7b_qwen3_30ba3b_thinking_science_50k_pt2_4e5 Moo Jin Kim TPU v5
(16 chips)
2.0d 5.41e20 model
1.22e21 HW (44%)
BPB: 0.204
#5939 e3956np_sft_qwen3_1pt7b_qwen3_30ba3b_thinking_science_50k_pt2_4e5 Moo Jin Kim TPU v5
(16 chips)
2.0d 5.41e20 model
1.20e21 HW (45%)
BPB: 0.873
#5359 curation-dclm_10k-expFM_natural-2e+21-d3584-L35-B128 Michael Ryan TPU v5
(32 chips)
1.0d 5.65e20 model
1.20e21 HW (47%)
BPB: 0.997
curation-dclm_10k-expFM_natural-2e+20-d1024-L11-B512 Michael Ryan TPU v5
(32 chips)
13.9h 1.80e20 model
9.10e20 HW (20%)
BPB: 0.994
curation-nemotron_10k-expFM_natural-3e+20-d1024-L11-B512 Michael Ryan TPU v5
(32 chips)
17.6h 2.30e20 model
8.70e20 HW (26%)
BPB: 1.128
#5359 curation-dclm_10k-expFM_natural-9e+20-d3584-L35-B128 Michael Ryan TPU v5
(64 chips)
7.4h 3.42e20 model
8.12e20 HW (42%)
BPB: 1.047
curation-nemotron_10k-expFM_natural-3e+20-d3584-L35-B32 Michael Ryan TPU v5
(32 chips)
17.3h 3.00e20 model
7.90e20 HW (38%)
BPB: 1.015
curation-dclm_10k-expFM_natural-3e+20-d3584-L35-B32 Michael Ryan TPU v5
(32 chips)
17.2h 3.00e20 model
7.89e20 HW (38%)
BPB: 0.959
#5360 curation-high_quality_100-expWARC_natural-1e+20-d256-L3-B2048 Michael Ryan TPU v5
(16 chips)
1.2d 10.00e19 model
7.03e20 HW (14%)
BPB: 1.736
#5360 curation-low_quality_100-expWARC_natural-1e+20-d256-L3-B2048 Michael Ryan TPU v5
(16 chips)
1.1d 9.93e19 model
7.02e20 HW (14%)
BPB: 1.468
#5360 curation-med_quality_100-expWARC_natural-1e+20-d256-L3-B2048 Michael Ryan TPU v5
(16 chips)
1.5d 10.00e19 model
7.01e20 HW (14%)
BPB: 1.523
Merged PR Open PR Draft PR Closed PR Open issue Closed issue

Keyboard shortcuts

?
Toggle this help
j / k
Next / previous section
t
Toggle details in current section
s
Cycle sort order in current section
o
Open current epic on GitHub
m
Open current milestone on GitHub
M
Open milestones list on GitHub
Data: weekly-data-2026-05-25_2026-05-31.json · sections-2026-05-25_2026-05-31.json · wandb-flops-2026-05-25_2026-05-31.json · tpu-usage-2026-05-25_2026-05-31.json · token-counts-2026-05-25_2026-05-31.json · cluster-status-2026-05-25_2026-05-31.json · discord-2026-05-25_2026-05-31.json · agent-moe-2026-05-25_2026-05-31.json