{"id":"https://openalex.org/W7170474404","doi":"https://doi.org/10.48550/arxiv.2607.20548","title":"SOAP, Muon, and Beyond: Pushing LLM Pretraining Scales","display_name":"SOAP, Muon, and Beyond: Pushing LLM Pretraining Scales","publication_year":2026,"publication_date":"2026-07-13","ids":{"openalex":"https://openalex.org/W7170474404","doi":"https://doi.org/10.48550/arxiv.2607.20548"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2607.20548","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2607.20548","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2607.20548","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5025224995","display_name":"Mikail Khona","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Khona, Mikail","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5143559638","display_name":"Aditya Vavre","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Vavre, Aditya","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5143583891","display_name":"Boxiang Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Boxiang","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5102651661","display_name":"Deyu Fu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Fu, Deyu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5143585357","display_name":"Hao Wu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wu, Hao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5143542294","display_name":"Mike Chrzanowski","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chrzanowski, Mike","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5143556120","display_name":"Bryan Catanzaro","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Catanzaro, Bryan","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5066366329","display_name":"Dheevatsa Mudigere","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Mudigere, Dheevatsa","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5063722719","display_name":"Jeff Pool","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Pool, Jeff","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5143552836","display_name":"Michael Lightstone","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Lightstone, Michael","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5128760297","display_name":"Mohammad Shoeybi","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Shoeybi, Mohammad","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5143567100","display_name":"Mostofa Patwary","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Patwary, Mostofa","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5011019569","display_name":"Nima Tajbakhsh","orcid":"https://orcid.org/0000-0001-8614-4811"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Tajbakhsh, Nima","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5143540576","display_name":"Tijmen Blankevoort","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Blankevoort, Tijmen","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":0,"corresponding_author_ids":[],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":false,"cited_by_count":0,"citation_normalized_percentile":null,"cited_by_percentile_year":null,"biblio":{"volume":null,"issue":null,"first_page":null,"last_page":null},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T11612","display_name":"Stochastic Gradient Optimization Techniques","score":0.07169999927282333,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T11612","display_name":"Stochastic Gradient Optimization Techniques","score":0.07169999927282333,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10764","display_name":"Privacy-Preserving Technologies in Data","score":0.053199999034404755,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T12579","display_name":"Muon and positron interactions and applications","score":0.043699998408555984,"subfield":{"id":"https://openalex.org/subfields/2211","display_name":"Mechanics of Materials"},"field":{"id":"https://openalex.org/fields/22","display_name":"Engineering"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/orthogonalization","display_name":"Orthogonalization","score":0.6966000199317932},{"id":"https://openalex.org/keywords/stability","display_name":"Stability (learning theory)","score":0.6690999865531921},{"id":"https://openalex.org/keywords/convergence","display_name":"Convergence (economics)","score":0.5723999738693237},{"id":"https://openalex.org/keywords/quality","display_name":"Quality (philosophy)","score":0.5052000284194946},{"id":"https://openalex.org/keywords/matching","display_name":"Matching (statistics)","score":0.46239998936653137},{"id":"https://openalex.org/keywords/benchmark","display_name":"Benchmark (surveying)","score":0.44290000200271606},{"id":"https://openalex.org/keywords/codebase","display_name":"Codebase","score":0.4284000098705292},{"id":"https://openalex.org/keywords/rate-of-convergence","display_name":"Rate of convergence","score":0.37070000171661377}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7358999848365784},{"id":"https://openalex.org/C47559304","wikidata":"https://www.wikidata.org/wiki/Q1702189","display_name":"Orthogonalization","level":2,"score":0.6966000199317932},{"id":"https://openalex.org/C112972136","wikidata":"https://www.wikidata.org/wiki/Q7595718","display_name":"Stability (learning theory)","level":2,"score":0.6690999865531921},{"id":"https://openalex.org/C2777303404","wikidata":"https://www.wikidata.org/wiki/Q759757","display_name":"Convergence (economics)","level":2,"score":0.5723999738693237},{"id":"https://openalex.org/C2779530757","wikidata":"https://www.wikidata.org/wiki/Q1207505","display_name":"Quality (philosophy)","level":2,"score":0.5052000284194946},{"id":"https://openalex.org/C165064840","wikidata":"https://www.wikidata.org/wiki/Q1321061","display_name":"Matching (statistics)","level":2,"score":0.46239998936653137},{"id":"https://openalex.org/C185798385","wikidata":"https://www.wikidata.org/wiki/Q1161707","display_name":"Benchmark (surveying)","level":2,"score":0.44290000200271606},{"id":"https://openalex.org/C51929080","wikidata":"https://www.wikidata.org/wiki/Q2425187","display_name":"Codebase","level":3,"score":0.4284000098705292},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.39660000801086426},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.375},{"id":"https://openalex.org/C57869625","wikidata":"https://www.wikidata.org/wiki/Q1783502","display_name":"Rate of convergence","level":3,"score":0.37070000171661377},{"id":"https://openalex.org/C2776359362","wikidata":"https://www.wikidata.org/wiki/Q2145286","display_name":"Representation (politics)","level":3,"score":0.3601999878883362},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.34700000286102295},{"id":"https://openalex.org/C125411270","wikidata":"https://www.wikidata.org/wiki/Q18653","display_name":"Encoding (memory)","level":2,"score":0.3402999937534332},{"id":"https://openalex.org/C57273362","wikidata":"https://www.wikidata.org/wiki/Q576722","display_name":"Decoding methods","level":2,"score":0.3321000039577484},{"id":"https://openalex.org/C137800194","wikidata":"https://www.wikidata.org/wiki/Q11713455","display_name":"Interpolation (computer graphics)","level":3,"score":0.33169999718666077},{"id":"https://openalex.org/C2776760102","wikidata":"https://www.wikidata.org/wiki/Q5139990","display_name":"Code (set theory)","level":3,"score":0.3314000070095062},{"id":"https://openalex.org/C113775141","wikidata":"https://www.wikidata.org/wiki/Q428691","display_name":"Computer engineering","level":1,"score":0.30730000138282776},{"id":"https://openalex.org/C126255220","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Mathematical optimization","level":1,"score":0.30320000648498535},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.3003999888896942},{"id":"https://openalex.org/C28719098","wikidata":"https://www.wikidata.org/wiki/Q44946","display_name":"Point (geometry)","level":2,"score":0.289000004529953},{"id":"https://openalex.org/C2779960059","wikidata":"https://www.wikidata.org/wiki/Q7113681","display_name":"Overhead (engineering)","level":2,"score":0.2825999855995178},{"id":"https://openalex.org/C174839445","wikidata":"https://www.wikidata.org/wiki/Q1134386","display_name":"Lock (firearm)","level":2,"score":0.2815000116825104},{"id":"https://openalex.org/C2777211547","wikidata":"https://www.wikidata.org/wiki/Q17141490","display_name":"Training (meteorology)","level":2,"score":0.27480000257492065},{"id":"https://openalex.org/C120314980","wikidata":"https://www.wikidata.org/wiki/Q180634","display_name":"Distributed computing","level":1,"score":0.2709999978542328},{"id":"https://openalex.org/C2778755073","wikidata":"https://www.wikidata.org/wiki/Q10858537","display_name":"Scale (ratio)","level":2,"score":0.2705000042915344}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2607.20548","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2607.20548","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2607.20548","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2607.20548","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Higher-order":[0],"optimizers":[1,149],"such":[2],"as":[3],"Muon":[4,127],"and":[5,16,30,54,62,70,87,126,153,178,196],"SOAP":[6,49,125],"offer":[7],"faster":[8],"convergence":[9,191],"than":[10],"AdamW,":[11],"but":[12],"their":[13,190],"computational":[14],"cost":[15],"numerical":[17],"stability":[18,152],"challenges":[19,39],"have":[20],"limited":[21],"adoption":[22],"at":[23,50,131,137,162],"scale.":[24],"In":[25],"this":[26,103],"work,":[27],"we":[28,105,134,165,194,212],"adapt":[29],"enhance":[31],"preconditioned":[32],"gradient":[33],"methods":[34],"to":[35,92,142,184,201],"overcome":[36],"the":[37,108,132,185,209],"practical":[38],"of":[40,84,102,111,121,140],"large-scale":[41],"LLM":[42],"pretraining.":[43],"We":[44,77],"first":[45],"identify":[46,195],"instabilities":[47],"in":[48,74],"large":[51,163],"batch":[52,138],"sizes":[53,139],"propose":[55],"algorithmic":[56],"modifications":[57],"including":[58],"per-step":[59],"QR":[60],"orthogonalization":[61,109],"improved":[63],"preconditioning":[64],"strategies":[65],"that":[66,124,216],"eliminate":[67],"loss":[68],"spikes":[69],"enable":[71,159],"stable":[72],"training":[73,151,161],"these":[75,148],"regimes.":[76],"then":[78],"present":[79],"a":[80,167,214],"unified":[81],"empirical":[82],"study":[83],"SOAP,":[85],"Muon,":[86],"AdamW":[88,130,156],"using":[89],"update-RMS":[90],"matching":[91],"ensure":[93],"fair":[94],"learning":[95],"rate":[96],"transfer":[97],"across":[98],"optimizers.":[99],"As":[100],"part":[101],"analysis,":[104],"empirically":[106],"evaluate":[107],"quality":[110,154],"Muon.":[112],"Our":[113,174],"experiments":[114],"on":[115,119],"multi-billion-parameter":[116],"models":[117],"trained":[118],"trillions":[120],"tokens":[122,144],"reveal":[123],"consistently":[128],"outperform":[129],"scales":[133],"tested.":[135],"Notably,":[136],"up":[141],"100M":[143],"for":[145,220],"next-token":[146],"prediction,":[147],"maintain":[150],"while":[155,181],"degrades.":[157],"To":[158,207],"efficient":[160],"scale,":[164],"introduce":[166],"layer-wise":[168,205],"distributed":[169],"optimizer":[170,186],"compatible":[171],"with":[172],"Megatron-LM.":[173],"implementation":[175],"balances":[176],"memory":[177],"hides":[179],"communication":[180],"avoiding":[182],"approximations":[183],"computations,":[187],"thus":[188],"retaining":[189],"benefits.":[192],"Additionally,":[193],"build":[197],"specific":[198],"system-level":[199],"improvements":[200],"further":[202],"accelerate":[203],"our":[204],"implementation.":[206],"support":[208],"research":[210],"community,":[211],"release":[213],"codebase":[215],"contains":[217],"emerging":[218],"algorithms":[219],"optimization:":[221],"https://github.com/NVIDIA-NeMo/Emerging-Optimizers":[222]},"counts_by_year":[],"updated_date":"2026-07-28T07:46:37.118299","created_date":"2026-07-25T00:00:00"}