{"id":"https://openalex.org/W7152032112","doi":"https://doi.org/10.48550/arxiv.2604.05091","title":"MegaTrain: Full Precision Training of 100B+ Parameter Large Language Models on a Single GPU","display_name":"MegaTrain: Full Precision Training of 100B+ Parameter Large Language Models on a Single GPU","publication_year":2026,"publication_date":"2026-04-06","ids":{"openalex":"https://openalex.org/W7152032112","doi":"https://doi.org/10.48550/arxiv.2604.05091"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2604.05091","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.05091","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":null,"license_id":null,"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.2604.05091","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5133206568","display_name":"Zhengqing Yuan","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yuan, Zhengqing","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133199798","display_name":"Hanchi Sun","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Sun, Hanchi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133187192","display_name":"Lichao Sun","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Sun, Lichao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5133213968","display_name":"Yanfang Ye","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Ye, Yanfang","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/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.27649998664855957,"subfield":{"id":"https://openalex.org/subfields/1708","display_name":"Hardware and Architecture"},"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/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.27649998664855957,"subfield":{"id":"https://openalex.org/subfields/1708","display_name":"Hardware and Architecture"},"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/T10036","display_name":"Advanced Neural Network Applications","score":0.1695999950170517,"subfield":{"id":"https://openalex.org/subfields/1707","display_name":"Computer Vision and Pattern Recognition"},"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/T12292","display_name":"Graph Theory and Algorithms","score":0.1307000070810318,"subfield":{"id":"https://openalex.org/subfields/1707","display_name":"Computer Vision and Pattern Recognition"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/cuda","display_name":"CUDA","score":0.48980000615119934},{"id":"https://openalex.org/keywords/throughput","display_name":"Throughput","score":0.4375},{"id":"https://openalex.org/keywords/graph","display_name":"Graph","score":0.41670000553131104},{"id":"https://openalex.org/keywords/stream-processing","display_name":"Stream processing","score":0.4090999960899353},{"id":"https://openalex.org/keywords/language-model","display_name":"Language model","score":0.4090000092983246},{"id":"https://openalex.org/keywords/context","display_name":"Context (archaeology)","score":0.40880000591278076},{"id":"https://openalex.org/keywords/flexibility","display_name":"Flexibility (engineering)","score":0.3986999988555908},{"id":"https://openalex.org/keywords/host","display_name":"Host (biology)","score":0.3815999925136566},{"id":"https://openalex.org/keywords/key","display_name":"Key (lock)","score":0.3783000111579895}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.8827000260353088},{"id":"https://openalex.org/C173608175","wikidata":"https://www.wikidata.org/wiki/Q232661","display_name":"Parallel computing","level":1,"score":0.5321000218391418},{"id":"https://openalex.org/C2778119891","wikidata":"https://www.wikidata.org/wiki/Q477690","display_name":"CUDA","level":2,"score":0.48980000615119934},{"id":"https://openalex.org/C157764524","wikidata":"https://www.wikidata.org/wiki/Q1383412","display_name":"Throughput","level":3,"score":0.4375},{"id":"https://openalex.org/C132525143","wikidata":"https://www.wikidata.org/wiki/Q141488","display_name":"Graph","level":2,"score":0.41670000553131104},{"id":"https://openalex.org/C107027933","wikidata":"https://www.wikidata.org/wiki/Q2006448","display_name":"Stream processing","level":2,"score":0.4090999960899353},{"id":"https://openalex.org/C137293760","wikidata":"https://www.wikidata.org/wiki/Q3621696","display_name":"Language model","level":2,"score":0.4090000092983246},{"id":"https://openalex.org/C2779343474","wikidata":"https://www.wikidata.org/wiki/Q3109175","display_name":"Context (archaeology)","level":2,"score":0.40880000591278076},{"id":"https://openalex.org/C2780598303","wikidata":"https://www.wikidata.org/wiki/Q65921492","display_name":"Flexibility (engineering)","level":2,"score":0.3986999988555908},{"id":"https://openalex.org/C126831891","wikidata":"https://www.wikidata.org/wiki/Q221673","display_name":"Host (biology)","level":2,"score":0.3815999925136566},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.3783000111579895},{"id":"https://openalex.org/C82876162","wikidata":"https://www.wikidata.org/wiki/Q17096504","display_name":"Latency (audio)","level":2,"score":0.3736000061035156},{"id":"https://openalex.org/C67388219","wikidata":"https://www.wikidata.org/wiki/Q207440","display_name":"Hash table","level":3,"score":0.36399999260902405},{"id":"https://openalex.org/C99138194","wikidata":"https://www.wikidata.org/wiki/Q183427","display_name":"Hash function","level":2,"score":0.35010001063346863},{"id":"https://openalex.org/C175291020","wikidata":"https://www.wikidata.org/wiki/Q1156822","display_name":"Offset (computer science)","level":2,"score":0.3483999967575073},{"id":"https://openalex.org/C48145219","wikidata":"https://www.wikidata.org/wiki/Q1335365","display_name":"Security token","level":2,"score":0.34709998965263367},{"id":"https://openalex.org/C93518851","wikidata":"https://www.wikidata.org/wiki/Q180160","display_name":"Metadata","level":2,"score":0.3264000117778778},{"id":"https://openalex.org/C86111242","wikidata":"https://www.wikidata.org/wiki/Q859595","display_name":"Coprocessor","level":2,"score":0.3262999951839447},{"id":"https://openalex.org/C206729178","wikidata":"https://www.wikidata.org/wiki/Q2271896","display_name":"Scheduling (production processes)","level":2,"score":0.3255999982357025},{"id":"https://openalex.org/C190839683","wikidata":"https://www.wikidata.org/wiki/Q2448197","display_name":"Train","level":2,"score":0.32109999656677246},{"id":"https://openalex.org/C79403827","wikidata":"https://www.wikidata.org/wiki/Q3988","display_name":"Real-time computing","level":1,"score":0.3188999891281128},{"id":"https://openalex.org/C2776257435","wikidata":"https://www.wikidata.org/wiki/Q1576430","display_name":"Bandwidth (computing)","level":2,"score":0.29910001158714294},{"id":"https://openalex.org/C103613024","wikidata":"https://www.wikidata.org/wiki/Q230924","display_name":"Stateless protocol","level":3,"score":0.27619999647140503},{"id":"https://openalex.org/C120314980","wikidata":"https://www.wikidata.org/wiki/Q180634","display_name":"Distributed computing","level":1,"score":0.2759999930858612},{"id":"https://openalex.org/C2779227376","wikidata":"https://www.wikidata.org/wiki/Q6505497","display_name":"Layer (electronics)","level":2,"score":0.272599995136261},{"id":"https://openalex.org/C168781493","wikidata":"https://www.wikidata.org/wiki/Q80585","display_name":"Associative array","level":2,"score":0.26989999413490295},{"id":"https://openalex.org/C199360897","wikidata":"https://www.wikidata.org/wiki/Q9143","display_name":"Programming language","level":1,"score":0.2639999985694885},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.2542000114917755},{"id":"https://openalex.org/C133095886","wikidata":"https://www.wikidata.org/wiki/Q1307173","display_name":"Single-precision floating-point format","level":3,"score":0.2526000142097473},{"id":"https://openalex.org/C113775141","wikidata":"https://www.wikidata.org/wiki/Q428691","display_name":"Computer engineering","level":1,"score":0.2522999942302704}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2604.05091","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.05091","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":null,"license_id":null,"version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2604.05091","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.05091","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":null,"license_id":null,"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":{"We":[0,70,94],"present":[1],"MegaTrain,":[2],"a":[3,18,72,120,164],"memory-centric":[4],"system":[5],"that":[6,77],"efficiently":[7],"trains":[8,130],"100B+":[9],"parameter":[10,79],"large":[11],"language":[12],"models":[13,131],"at":[14],"full":[15],"precision":[16],"on":[17,163],"single":[19,121,165],"GPU.":[20],"Unlike":[21],"traditional":[22],"GPU-centric":[23],"systems,":[24],"MegaTrain":[25,128,153],"stores":[26],"parameters":[27,48],"and":[28,36,50,82],"optimizer":[29],"states":[30],"in":[31,49,117],"host":[32,126],"memory":[33],"(CPU":[34],"memory)":[35],"treats":[37],"GPUs":[38],"as":[39,106],"transient":[40],"compute":[41,51],"engines.":[42],"For":[43],"each":[44],"layer,":[45],"we":[46,64],"stream":[47,108],"gradients":[52],"out,":[53],"minimizing":[54],"persistent":[55,96,111],"device":[56],"state.":[57],"To":[58],"battle":[59],"the":[60,140],"CPU-GPU":[61],"bandwidth":[62],"bottleneck,":[63],"adopt":[65],"two":[66],"key":[67],"optimizations.":[68],"1)":[69],"introduce":[71],"pipelined":[73],"double-buffered":[74],"execution":[75],"engine":[76],"overlaps":[78],"prefetching,":[80],"computation,":[81],"gradient":[83],"offloading":[84,148],"across":[85],"multiple":[86],"CUDA":[87],"streams,":[88],"enabling":[89],"continuous":[90],"GPU":[91,123],"execution.":[92],"2)":[93],"replace":[95],"autograd":[97],"graphs":[98],"with":[99,124,146,159],"stateless":[100],"layer":[101],"templates,":[102],"binding":[103],"weights":[104],"dynamically":[105],"they":[107],"in,":[109],"eliminating":[110],"graph":[112],"metadata":[113],"while":[114],"providing":[115],"flexibility":[116],"scheduling.":[118],"On":[119],"H200":[122],"1.5TB":[125],"memory,":[127],"reliably":[129],"up":[132],"to":[133],"120B":[134],"parameters.":[135],"It":[136],"also":[137,154],"achieves":[138],"1.84$\\times$":[139],"training":[141,150,158],"throughput":[142],"of":[143],"DeepSpeed":[144],"ZeRO-3":[145],"CPU":[147],"when":[149],"14B":[151],"models.":[152],"enables":[155],"7B":[156],"model":[157],"512k":[160],"token":[161],"context":[162],"GH200.":[166]},"counts_by_year":[],"updated_date":"2026-07-28T07:46:37.118299","created_date":"2026-04-09T00:00:00"}
