{"id":"https://openalex.org/W7166122089","doi":"https://doi.org/10.1145/3815572.3815765","title":"Combining Domain and Tensor Parallelism to Train Multi-billion-parameter AI Weather Models","display_name":"Combining Domain and Tensor Parallelism to Train Multi-billion-parameter AI Weather Models","publication_year":2026,"publication_date":"2026-06-26","ids":{"openalex":"https://openalex.org/W7166122089","doi":"https://doi.org/10.1145/3815572.3815765"},"language":null,"primary_location":{"id":"doi:10.1145/3815572.3815765","is_oa":true,"landing_page_url":"https://doi.org/10.1145/3815572.3815765","pdf_url":null,"source":null,"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the Platform for Advanced Scientific Computing Conference","raw_type":"proceedings-article"},"type":"conference-paper","indexed_in":["crossref"],"open_access":{"is_oa":true,"oa_status":"gold","oa_url":"https://doi.org/10.1145/3815572.3815765","any_repository_has_fulltext":null},"authorships":[{"author_position":"first","author":{"id":null,"display_name":"Deifilia Kieckhefen","orcid":"https://orcid.org/0009-0000-1197-5493"},"institutions":[{"id":"https://openalex.org/I102335020","display_name":"Karlsruhe Institute of Technology","ror":"https://ror.org/04t3en479","country_code":"DE","type":"education","lineage":["https://openalex.org/I102335020","https://openalex.org/I1305996414"]}],"countries":["DE"],"is_corresponding":false,"raw_author_name":"Deifilia Kieckhefen","raw_affiliation_strings":["Karlsruhe Institute of Technology (KIT), Karlsruhe, Germany"],"raw_orcid":"https://orcid.org/0009-0000-1197-5493","affiliations":[{"raw_affiliation_string":"Karlsruhe Institute of Technology (KIT), Karlsruhe, Germany","institution_ids":["https://openalex.org/I102335020"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5067996051","display_name":"Markus G\u00f6tz","orcid":"https://orcid.org/0000-0002-2233-1041"},"institutions":[{"id":"https://openalex.org/I102335020","display_name":"Karlsruhe Institute of Technology","ror":"https://ror.org/04t3en479","country_code":"DE","type":"education","lineage":["https://openalex.org/I102335020","https://openalex.org/I1305996414"]}],"countries":["DE"],"is_corresponding":false,"raw_author_name":"Markus G\u00f6tz","raw_affiliation_strings":["Karlsruhe Institute of Technology (KIT), Karlsruhe, Germany and Helmholtz AI, Karlsruhe, Germany"],"raw_orcid":"https://orcid.org/0000-0002-2233-1041","affiliations":[{"raw_affiliation_string":"Karlsruhe Institute of Technology (KIT), Karlsruhe, Germany and Helmholtz AI, Karlsruhe, Germany","institution_ids":["https://openalex.org/I102335020"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5012750908","display_name":"Lars H. Heyen","orcid":"https://orcid.org/0000-0001-7949-1858"},"institutions":[{"id":"https://openalex.org/I102335020","display_name":"Karlsruhe Institute of Technology","ror":"https://ror.org/04t3en479","country_code":"DE","type":"education","lineage":["https://openalex.org/I102335020","https://openalex.org/I1305996414"]}],"countries":["DE"],"is_corresponding":false,"raw_author_name":"Lars Helge Heyen","raw_affiliation_strings":["Karlsruhe Institute of Technology (KIT), Karlsruhe, Germany"],"raw_orcid":"https://orcid.org/0000-0001-7949-1858","affiliations":[{"raw_affiliation_string":"Karlsruhe Institute of Technology (KIT), Karlsruhe, Germany","institution_ids":["https://openalex.org/I102335020"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5030034618","display_name":"Achim Streit","orcid":"https://orcid.org/0000-0002-5065-469X"},"institutions":[{"id":"https://openalex.org/I102335020","display_name":"Karlsruhe Institute of Technology","ror":"https://ror.org/04t3en479","country_code":"DE","type":"education","lineage":["https://openalex.org/I102335020","https://openalex.org/I1305996414"]}],"countries":["DE"],"is_corresponding":false,"raw_author_name":"Achim Streit","raw_affiliation_strings":["Karlsruhe Institute of Technology (KIT), Karlsruhe, Germany"],"raw_orcid":"https://orcid.org/0000-0002-5065-469X","affiliations":[{"raw_affiliation_string":"Karlsruhe Institute of Technology (KIT), Karlsruhe, Germany","institution_ids":["https://openalex.org/I102335020"]}]},{"author_position":"last","author":{"id":"https://openalex.org/A5024295845","display_name":"Charlotte Debus","orcid":"https://orcid.org/0000-0002-7156-2022"},"institutions":[{"id":"https://openalex.org/I102335020","display_name":"Karlsruhe Institute of Technology","ror":"https://ror.org/04t3en479","country_code":"DE","type":"education","lineage":["https://openalex.org/I102335020","https://openalex.org/I1305996414"]}],"countries":["DE"],"is_corresponding":false,"raw_author_name":"Charlotte Debus","raw_affiliation_strings":["Karlsruhe Institute of Technology (KIT), Karlsruhe, Germany"],"raw_orcid":"https://orcid.org/0000-0002-7156-2022","affiliations":[{"raw_affiliation_string":"Karlsruhe Institute of Technology (KIT), Karlsruhe, Germany","institution_ids":["https://openalex.org/I102335020"]}]}],"institutions":[],"countries_distinct_count":1,"institutions_distinct_count":1,"corresponding_author_ids":[],"corresponding_institution_ids":["https://openalex.org/I102335020"],"apc_list":null,"apc_paid":null,"fwci":0.0,"has_fulltext":false,"cited_by_count":0,"citation_normalized_percentile":{"value":0.76703841,"is_in_top_1_percent":false,"is_in_top_10_percent":false},"cited_by_percentile_year":null,"biblio":{"volume":null,"issue":null,"first_page":"1","last_page":"13"},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T10466","display_name":"Meteorological Phenomena and Simulations","score":0.6373000144958496,"subfield":{"id":"https://openalex.org/subfields/1902","display_name":"Atmospheric Science"},"field":{"id":"https://openalex.org/fields/19","display_name":"Earth and Planetary Sciences"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T10466","display_name":"Meteorological Phenomena and Simulations","score":0.6373000144958496,"subfield":{"id":"https://openalex.org/subfields/1902","display_name":"Atmospheric Science"},"field":{"id":"https://openalex.org/fields/19","display_name":"Earth and Planetary Sciences"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T12303","display_name":"Tensor decomposition and applications","score":0.09220000356435776,"subfield":{"id":"https://openalex.org/subfields/2605","display_name":"Computational Mathematics"},"field":{"id":"https://openalex.org/fields/26","display_name":"Mathematics"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T11206","display_name":"Model Reduction and Neural Networks","score":0.061400000005960464,"subfield":{"id":"https://openalex.org/subfields/3109","display_name":"Statistical and Nonlinear Physics"},"field":{"id":"https://openalex.org/fields/31","display_name":"Physics and Astronomy"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/domain","display_name":"Domain (mathematical analysis)","score":0.536899983882904},{"id":"https://openalex.org/keywords/parallelism","display_name":"Parallelism (grammar)","score":0.43950000405311584},{"id":"https://openalex.org/keywords/tensor","display_name":"Tensor (intrinsic definition)","score":0.37229999899864197},{"id":"https://openalex.org/keywords/numerical-weather-prediction","display_name":"Numerical weather prediction","score":0.33169999718666077},{"id":"https://openalex.org/keywords/weather-forecasting","display_name":"Weather forecasting","score":0.2930999994277954}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6122000217437744},{"id":"https://openalex.org/C36503486","wikidata":"https://www.wikidata.org/wiki/Q11235244","display_name":"Domain (mathematical analysis)","level":2,"score":0.536899983882904},{"id":"https://openalex.org/C2781172179","wikidata":"https://www.wikidata.org/wiki/Q853109","display_name":"Parallelism (grammar)","level":2,"score":0.43950000405311584},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.4366999864578247},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.3894999921321869},{"id":"https://openalex.org/C155281189","wikidata":"https://www.wikidata.org/wiki/Q3518150","display_name":"Tensor (intrinsic definition)","level":2,"score":0.37229999899864197},{"id":"https://openalex.org/C147947694","wikidata":"https://www.wikidata.org/wiki/Q837552","display_name":"Numerical weather prediction","level":2,"score":0.33169999718666077},{"id":"https://openalex.org/C21001229","wikidata":"https://www.wikidata.org/wiki/Q182868","display_name":"Weather forecasting","level":2,"score":0.2930999994277954},{"id":"https://openalex.org/C77618280","wikidata":"https://www.wikidata.org/wiki/Q1155772","display_name":"Scheme (mathematics)","level":2,"score":0.28529998660087585},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.2526000142097473},{"id":"https://openalex.org/C45374587","wikidata":"https://www.wikidata.org/wiki/Q12525525","display_name":"Computation","level":2,"score":0.25220000743865967}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.1145/3815572.3815765","is_oa":true,"landing_page_url":"https://doi.org/10.1145/3815572.3815765","pdf_url":null,"source":null,"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the Platform for Advanced Scientific Computing Conference","raw_type":"proceedings-article"}],"best_oa_location":{"id":"doi:10.1145/3815572.3815765","is_oa":true,"landing_page_url":"https://doi.org/10.1145/3815572.3815765","pdf_url":null,"source":null,"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the Platform for Advanced Scientific Computing Conference","raw_type":"proceedings-article"},"sustainable_development_goals":[{"score":0.8603821396827698,"id":"https://metadata.un.org/sdg/13","display_name":"Climate action"}],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":28,"referenced_works":["https://openalex.org/W2024966118","https://openalex.org/W2039747889","https://openalex.org/W2809789958","https://openalex.org/W2900933286","https://openalex.org/W2951981190","https://openalex.org/W2952046647","https://openalex.org/W2972087877","https://openalex.org/W3025949386","https://openalex.org/W3081844928","https://openalex.org/W3099145603","https://openalex.org/W3129831491","https://openalex.org/W3138516171","https://openalex.org/W4225595610","https://openalex.org/W4251946001","https://openalex.org/W4318620401","https://openalex.org/W4381245607","https://openalex.org/W4382048606","https://openalex.org/W4383218913","https://openalex.org/W4386768656","https://openalex.org/W4388654737","https://openalex.org/W4388728292","https://openalex.org/W4390788999","https://openalex.org/W4405376708","https://openalex.org/W4405387414","https://openalex.org/W4415797118","https://openalex.org/W6888777245","https://openalex.org/W7131212557","https://openalex.org/W7138983925"],"related_works":[],"abstract_inverted_index":{"AI-based":[0,17],"methods":[1],"have":[2,11],"recently":[3],"revolutionized":[4],"atmospheric":[5,24],"modeling.":[6],"Successes":[7],"in":[8],"medium-range":[9],"forecasting":[10],"led":[12],"to":[13,69,111],"rapid":[14],"developments":[15],"towards":[16],"models.":[18,124],"However,":[19],"accurate":[20],"modeling":[21],"of":[22,32],"complex":[23],"dynamics":[25],"at":[26],"high":[27],"spatial":[28],"resolutions":[29],"requires":[30],"billions":[31],"neural":[33],"network":[34],"parameters":[35],"and":[36,43,60,66,76,121],"gigabyte-sized":[37],"data":[38,142,183],"samples,":[39],"making":[40],"accelerator":[41],"memory":[42,71],"I/O-bandwidth":[44],"the":[45,82,127,160,176],"bottlenecks":[46,178],"for":[47,100,159],"model":[48,93],"training.":[49],"To":[50,125],"overcome":[51],"these":[52],"limitations,":[53],"we":[54,106,130,157],"introduce":[55],"Jigsaw,":[56,105],"a":[57,91],"distributed":[58],"training":[59,163],"inference":[61],"scheme":[62,85],"that":[63,97,148,175],"leverages":[64],"domain":[65,188],"tensor":[67],"parallelism":[68],"eliminate":[70],"redundancy":[72],"across":[73,143],"model-parallel":[74],"processes":[75],"reduce":[77],"I/O":[78],"demands.":[79],"We":[80],"apply":[81],"Jigsaw":[83],"parallelization":[84],"into":[86],"an":[87],"MLP-Mixer":[88],"architecture,":[89],"WeatherMixer,":[90],"multi-layer-perceptron-based":[92],"with":[94,109,117],"global":[95,135],"vision":[96],"is":[98],"well-suited":[99],"learning":[101],"weather":[102,119],"phenomena.":[103],"Using":[104],"train":[107],"WeatherMixer":[108],"up":[110],"3.2B-parameters,":[112],"achieving":[113],"predictive":[114],"performance":[115],"competitive":[116],"numerical":[118],"prediction":[120],"state-of-the-art":[122],"AI":[123],"highlight":[126],"computational":[128,194],"performance,":[129],"perform":[131],"scaling":[132,172],"experiments":[133,173],"on":[134,164],"0.25\u00b0":[136],"(\u2248":[137,166],"30":[138,167],"km":[139,168],"resolution)":[140,169],"ERA5":[141],"two":[144],"HPC":[145],"systems.":[146],"Anticipating":[147],"future":[149],"reanalysis":[150],"datasets":[151],"will":[152],"include":[153],"even":[154],"higher":[155],"resolutions,":[156],"demonstrate,":[158],"first":[161],"time,":[162],"0.125\u00b0":[165],"data.":[170],"The":[171],"demonstrate":[174],"dataloading":[177],"arising":[179],"from":[180],"high-resolution":[181],"input":[182],"samples":[184],"are":[185],"reduced":[186],"through":[187],"parallelism;":[189],"this":[190],"subsequently":[191],"improves":[192],"per-GPU":[193],"throughput.":[195]},"counts_by_year":[],"updated_date":"2026-07-29T14:22:42.915294","created_date":"2026-06-27T00:00:00"}