{"id":"https://openalex.org/W7224115104","doi":"https://doi.org/10.48550/arxiv.2610.37333","title":"diff-soup: GPU-native Differentiable Simulation and Inference for Large-scale Stiff Dynamical Systems","display_name":"diff-soup: GPU-native Differentiable Simulation and Inference for Large-scale Stiff Dynamical Systems","publication_year":2026,"publication_date":"2026-10-27","ids":{"openalex":"https://openalex.org/W7224115104","doi":"https://doi.org/10.48550/arxiv.2610.37333"},"language":"en","primary_location":{"id":"pmh:oai:arXiv.org:2610.37333","doi":null,"is_oa":true,"landing_page_url":"https://arxiv.org/abs/2610.37333","pdf_url":"https://arxiv.org/pdf/2610.37333","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,"listed_in":[],"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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},"volume":null,"type":"preprint","indexed_in":["arxiv","datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://arxiv.org/pdf/2610.37333","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5115550141","display_name":"Samuel Degnan-Morgenstern","orcid":"https://orcid.org/0009-0001-5023-1160","observed_orcids":["https://orcid.org/0009-0001-5023-1160"]},"institutions":[{"id":"https://openalex.org/I63966007","display_name":"Massachusetts Institute of Technology","ror":"https://ror.org/042nb2s44","country_code":"US","type":"education","lineage":["https://openalex.org/I63966007"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Degnan-Morgenstern, Samuel","raw_affiliation_strings":["Department of Chemical Engineering , Massachusetts Institute of Technology , 25 Ames St. , Cambridge , MA 02139 , USA"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Department of Chemical Engineering , Massachusetts Institute of Technology , 25 Ames St. , Cambridge , MA 02139 , USA","institution_ids":["https://openalex.org/I63966007"]}]},{"author_position":"last","author":{"id":"https://openalex.org/A5070486479","display_name":"Martin Z. Bazant","orcid":"https://orcid.org/0000-0002-8200-4501","observed_orcids":["https://orcid.org/0000-0002-8200-4501"]},"institutions":[{"id":"https://openalex.org/I63966007","display_name":"Massachusetts Institute of Technology","ror":"https://ror.org/042nb2s44","country_code":"US","type":"education","lineage":["https://openalex.org/I63966007"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Bazant, Martin Z.","raw_affiliation_strings":["Department of Chemical Engineering , Massachusetts Institute of Technology , 25 Ames St. , Cambridge , MA 02139 , USA","Department of Mathematics , Massachusetts Institute of Technology , 182 Memorial Dr. , Cambridge , MA 02139 , USA"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Department of Chemical Engineering , Massachusetts Institute of Technology , 25 Ames St. , Cambridge , MA 02139 , USA","institution_ids":["https://openalex.org/I63966007"]},{"raw_affiliation_string":"Department of Mathematics , Massachusetts Institute of Technology , 182 Memorial Dr. , Cambridge , MA 02139 , USA","institution_ids":["https://openalex.org/I63966007"]}]}],"institutions":[],"countries_distinct_count":1,"institutions_distinct_count":1,"corresponding_author_ids":[],"corresponding_institution_ids":["https://openalex.org/I63966007"],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":true,"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/T11206","display_name":"Model Reduction and Neural Networks","score":0.9491999745368958,"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"}},"topics":[{"id":"https://openalex.org/T11206","display_name":"Model Reduction and Neural Networks","score":0.9491999745368958,"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"}},{"id":"https://openalex.org/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.021900000050663948,"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/T11416","display_name":"Numerical methods for differential equations","score":0.009499999694526196,"subfield":{"id":"https://openalex.org/subfields/2612","display_name":"Numerical Analysis"},"field":{"id":"https://openalex.org/fields/26","display_name":"Mathematics"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/differentiable-simulation","display_name":"differentiable simulation","score":0.7850000262260437},{"id":"https://openalex.org/keywords/partial-differential-equations","display_name":"partial differential equations","score":0.9089999794960022},{"id":"https://openalex.org/keywords/sparse-linear-solvers","display_name":"sparse linear solvers","score":0.7490000128746033},{"id":"https://openalex.org/keywords/gpu-acceleration","display_name":"GPU acceleration","score":0.6629999876022339},{"id":"https://openalex.org/keywords/jax","display_name":"JAX","score":0.718999981880188},{"id":"https://openalex.org/keywords/pde-constrained-optimization","display_name":"PDE-constrained optimization","score":0.8980000019073486},{"id":"https://openalex.org/keywords/bayesian-inference","display_name":"Bayesian inference","score":0.8659999966621399},{"id":"https://openalex.org/keywords/implicit-neural-representations","display_name":"implicit neural representations","score":0.699999988079071},{"id":"https://openalex.org/keywords/fixed-bed-reactor","display_name":"fixed-bed reactor","score":0.8629999756813049},{"id":"https://openalex.org/keywords/catalytic-activity","display_name":"catalytic activity","score":0.8899999856948853},{"id":"https://openalex.org/keywords/multi-gpu-parallelism","display_name":"multi-GPU parallelism","score":0.7770000100135803}],"concepts":[{"id":"https://openalex.org/C2778770139","wikidata":"https://www.wikidata.org/wiki/Q1966904","display_name":"Solver","level":2,"score":0.6855999827384949},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.597000002861023},{"id":"https://openalex.org/C79379906","wikidata":"https://www.wikidata.org/wiki/Q3174497","display_name":"Dynamical systems theory","level":2,"score":0.589900016784668},{"id":"https://openalex.org/C202615002","wikidata":"https://www.wikidata.org/wiki/Q783507","display_name":"Differentiable function","level":2,"score":0.5695000290870667},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.5529000163078308},{"id":"https://openalex.org/C108583219","wikidata":"https://www.wikidata.org/wiki/Q197536","display_name":"Deep learning","level":2,"score":0.4722000062465668},{"id":"https://openalex.org/C114275822","wikidata":"https://www.wikidata.org/wiki/Q621512","display_name":"Linear dynamical system","level":3,"score":0.43630000948905945},{"id":"https://openalex.org/C33962884","wikidata":"https://www.wikidata.org/wiki/Q378637","display_name":"Dynamical system (definition)","level":3,"score":0.43459999561309814},{"id":"https://openalex.org/C160234255","wikidata":"https://www.wikidata.org/wiki/Q812535","display_name":"Bayesian inference","level":3,"score":0.4018000066280365},{"id":"https://openalex.org/C93779851","wikidata":"https://www.wikidata.org/wiki/Q271977","display_name":"Partial differential equation","level":2,"score":0.39980000257492065},{"id":"https://openalex.org/C9652623","wikidata":"https://www.wikidata.org/wiki/Q190109","display_name":"Field (mathematics)","level":2,"score":0.399399995803833},{"id":"https://openalex.org/C28826006","wikidata":"https://www.wikidata.org/wiki/Q33521","display_name":"Applied mathematics","level":1,"score":0.39430001378059387},{"id":"https://openalex.org/C116672817","wikidata":"https://www.wikidata.org/wiki/Q1454986","display_name":"Physical system","level":2,"score":0.3889999985694885},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.38359999656677246},{"id":"https://openalex.org/C50644808","wikidata":"https://www.wikidata.org/wiki/Q192776","display_name":"Artificial neural network","level":2,"score":0.37209999561309814},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.367900013923645},{"id":"https://openalex.org/C2777472644","wikidata":"https://www.wikidata.org/wiki/Q16968992","display_name":"Approximate inference","level":3,"score":0.35120001435279846},{"id":"https://openalex.org/C134261354","wikidata":"https://www.wikidata.org/wiki/Q938438","display_name":"Statistical inference","level":2,"score":0.3402000069618225},{"id":"https://openalex.org/C99844830","wikidata":"https://www.wikidata.org/wiki/Q102441924","display_name":"Scaling","level":2,"score":0.32589998841285706},{"id":"https://openalex.org/C2776359362","wikidata":"https://www.wikidata.org/wiki/Q2145286","display_name":"Representation (politics)","level":3,"score":0.31450000405311584},{"id":"https://openalex.org/C78045399","wikidata":"https://www.wikidata.org/wiki/Q11214","display_name":"Differential equation","level":2,"score":0.30649998784065247},{"id":"https://openalex.org/C137836250","wikidata":"https://www.wikidata.org/wiki/Q984063","display_name":"Optimization problem","level":2,"score":0.2962000072002411},{"id":"https://openalex.org/C126255220","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Mathematical optimization","level":1,"score":0.2921999990940094},{"id":"https://openalex.org/C32230216","wikidata":"https://www.wikidata.org/wiki/Q7882499","display_name":"Uncertainty quantification","level":2,"score":0.2791000008583069},{"id":"https://openalex.org/C158622935","wikidata":"https://www.wikidata.org/wiki/Q660848","display_name":"Nonlinear system","level":2,"score":0.27399998903274536},{"id":"https://openalex.org/C67186912","wikidata":"https://www.wikidata.org/wiki/Q367664","display_name":"Data modeling","level":2,"score":0.2728999853134155},{"id":"https://openalex.org/C6802819","wikidata":"https://www.wikidata.org/wiki/Q1072174","display_name":"Linear system","level":2,"score":0.2660999894142151},{"id":"https://openalex.org/C107673813","wikidata":"https://www.wikidata.org/wiki/Q812534","display_name":"Bayesian probability","level":2,"score":0.26589998602867126},{"id":"https://openalex.org/C81587630","wikidata":"https://www.wikidata.org/wiki/Q5319044","display_name":"Dynamical simulation","level":2,"score":0.2644999921321869},{"id":"https://openalex.org/C79581498","wikidata":"https://www.wikidata.org/wiki/Q1367530","display_name":"Suite","level":2,"score":0.2558000087738037},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.25459998846054077}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:oai:arXiv.org:2610.37333","doi":null,"is_oa":true,"landing_page_url":"https://arxiv.org/abs/2610.37333","pdf_url":"https://arxiv.org/pdf/2610.37333","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,"listed_in":[],"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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},{"id":"doi:10.48550/arxiv.2610.37333","doi":"https://doi.org/10.48550/arxiv.2610.37333","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2610.37333","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,"listed_in":[],"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-nc-sa","license_id":"https://openalex.org/licenses/cc-by-nc-sa","version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"}],"best_oa_location":{"id":"pmh:oai:arXiv.org:2610.37333","doi":null,"is_oa":true,"landing_page_url":"https://arxiv.org/abs/2610.37333","pdf_url":"https://arxiv.org/pdf/2610.37333","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,"listed_in":[],"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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},"sustainable_development_goals":[],"study_designs":[],"awards":[{"id":"https://openalex.org/G539115378","display_name":"Graduate Research Fellowship Program (GRFP)","funder_award_id":"2141064","funder_id":"https://openalex.org/F4320306076","funder_display_name":"National Science Foundation"}],"funders":[{"id":"https://openalex.org/F4320306076","display_name":"National Science Foundation","ror":"https://ror.org/021nxhr62"}],"has_content":{"pdf":true,"grobid_xml":true},"content_urls":{"pdf":"https://content.openalex.org/works/W7224115104.pdf","grobid_xml":"https://content.openalex.org/works/W7224115104.grobid-xml"},"referenced_works_count":0,"referenced_works":[],"related_works":["https://openalex.org/W7164235321","https://openalex.org/W3140585314","https://openalex.org/W3044664263","https://openalex.org/W4403592915","https://openalex.org/W2888717642","https://openalex.org/W4388741168","https://openalex.org/W4309955173","https://openalex.org/W7169867101","https://openalex.org/W7160955485","https://openalex.org/W7156045231"],"abstract_inverted_index":{"Learning":[0],"physical":[1],"models":[2,23],"from":[3,114],"spatiotemporal":[4],"data":[5,106],"requires":[6],"simulating":[7],"systems":[8,29],"of":[9,71,138,171],"partial":[10],"differential":[11],"equations":[12],"(PDEs)":[13],"and":[14,45,52,69,105,109,130,144,157,190],"differentiating":[15],"through":[16,96],"their":[17],"solution.":[18],"Upon":[19],"spatial":[20],"discretization,":[21],"these":[22],"give":[24],"rise":[25],"to":[26,49,93,118,135,184],"stiff":[27,73,123],"dynamical":[28,74],"that":[30],"require":[31,46],"repeatedly":[32],"solving":[33],"large,":[34,72],"sparse":[35,89,100],"linear":[36],"systems.":[37,75],"Traditional":[38],"scientific":[39],"computing":[40],"frameworks":[41,132],"are":[42],"largely":[43],"CPU-based":[44],"problem-specific":[47],"tuning":[48],"accelerate,":[50],"scale,":[51],"differentiate.":[53],"In":[54],"this":[55],"work,":[56],"we":[57],"introduce":[58],"diff-soup,":[59],"a":[60,82,149,158,164,172],"differentiable,":[61],"sparse-oriented,":[62],"unified":[63],"PDE":[64,124],"suite":[65],"for":[66,86,167],"GPU-native":[67],"simulation":[68],"inference":[70,117,152],"Built":[76],"in":[77,140,187],"JAX,":[78],"diff-soup":[79,126,179],"provides":[80],"1)":[81],"fully":[83],"differentiable":[84],"design":[85],"problems":[87,95],"with":[88,103],"structure,":[90],"2)":[91],"scaling":[92],"large":[94],"the":[97,168],"NVIDIA":[98],"cuDSS":[99],"direct":[101],"solver":[102],"model":[104],"multi-GPU":[107],"parallelism,":[108],"3)":[110],"end-to-end":[111],"PDE-constrained":[112],"optimization,":[113],"classical":[115,150],"statistical":[116],"embedded":[119],"deep":[120,160],"learning.":[121],"On":[122],"benchmarks,":[125],"outperforms":[127],"established":[128],"CPU-":[129],"GPU-based":[131],"by":[133],"one":[134],"two":[136],"orders":[137],"magnitude":[139],"both":[141,148],"forward":[142],"solves":[143],"gradients.":[145],"We":[146,177],"demonstrate":[147],"Bayesian":[151],"workflow,":[153,162],"identifying":[154],"constitutive":[155],"relationships,":[156],"modern":[159],"learning":[161],"training":[163],"neural":[165],"field":[166],"catalyst":[169],"activity":[170],"three-dimensional":[173],"packed":[174],"bed":[175],"reactor.":[176],"provide":[178],"as":[180],"an":[181],"open-source":[182],"repository":[183],"facilitate":[185],"research":[186],"computational":[188],"science":[189],"engineering.":[191]},"counts_by_year":[],"updated_date":"2026-11-01T11:40:01.658920","created_date":"2026-10-29T00:00:00"}