It’s done! Marin’s Mochi 535B-A23B finished its main pretraining phase this morning. 18T tokens! Long context this weekend, SFT over the next week or so. Hoping to release checkpoints next week. Some EU paperwork first. mumtracker.exe.xyz/hero-run-535b
David Hall
@dlwh
Creator of Levanter and @MarinCommunity. MTS @ Open Athena. Previously Research Engineering @StanfordCRFM, co-founder at Semantic Machines ⟶ MSFT.
1.4K Following 4.3K Followers 2K Posts
Latest posts 5
So, about that Uncheatable forecast. I used an estimate that wasn’t properly flop-controlled, and hadn’t checked its asymptote carefully enough.
I should have checked it before posting. Easy fix. Details below.
I should have checked it before posting. Easy fix. Details below.
Marin's 535b is now >80% through pretraining! Here’s loss vs. our scaling-ladder forecasts. The sharp drop comes with the 4K→16K context switch, so we're kind of cheating the forecast. (This was planned.) We'll re-eval at 4k too. forge.coreweave.com/wandb/marin-co…View quoted post on X
As usual Larry has the actually useful details.
background- our data has longer docs so intent was to launch at 16k like MAI. However our MoE kernel only supported 4k (high dropping), so we planned to extend after kernel was improved. Since that eta was unknown, scaling ladder was kept at 4k, with knowledge that scaling ladder would cease to apply once new kernel landed and context grew. Quite remarkably that and many other kernel improvements landed, raising MFU from 20 to 30 mid run.View quoted post on X
The forecast on Uncheatable was added mid run where its fcst starts, but we didn't calibrate its asymptote or fit it to the data from the run, so its quite off. The main value we got from it was to directionally estimate impact of the earlier datamix change. Paloma was the eval used to evaluate if we were 'on track'.
Memory caps tokens per batch, so 4x longer context is paired with 4x fewer sequences.
Paloma forecast is much better/more correct btw, up to 4k->16k changeover
Marin's 535b is now >80% through pretraining! Here’s loss vs. our scaling-ladder forecasts. The sharp drop comes with the 4K→16K context switch, so we're kind of cheating the forecast. (This was planned.) We'll re-eval at 4k too. forge.coreweave.com/wandb/marin-co…View quoted post on X
Marin's 535b is now >80% through pretraining! Here’s loss vs. our scaling-ladder forecasts. The sharp drop comes with the 4K→16K context switch, so we're kind of cheating the forecast. (This was planned.) We'll re-eval at 4k too. forge.coreweave.com/wandb/marin-co…