The Well: a Large-Scale Collection of Diverse Physics Simulations for Machine Learning
Not on the map. Why?
Abstract (arXiv)
Machine learning based surrogate models offer researchers powerful tools for accelerating simulation-based workflows. However, as standard datasets in this space often cover small classes of physical behavior, it can be difficult to evaluate the efficacy of new approaches. To address this gap, we introduce the Well: a large-scale collection of datasets containing numerical simulations of a wide variety of spatiotemporal physical systems. The Well draws from domain experts and numerical software developers to provide 15TB of data across 16 datasets covering diverse domains such as biological systems, fluid dynamics, acoustic scattering, as well as magneto-hydrodynamic simulations of extra-galactic fluids or supernova explosions. These datasets can be used individually or as part of a broader benchmark suite. To facilitate usage of the Well, we provide a unified PyTorch interface for training and evaluating models. We demonstrate the function of this library by introducing example baselines that highlight the new challenges posed by the complex dynamics of the Well. The code and data is available at https://github.com/PolymathicAI/the_well.
Key facts. Cited by 134 papers in the map (data through 7 Oct 2026). Builds on 42 of its 216 references that are in the map. First posted on arXiv 30 Nov 2024.
Builds on
42 of 216 references.
References · 42
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2023GPT-4 Technical Reportcited by 18,132
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,131
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 2021Scaling Language Models: Methods, Analysis & Insights from Training Gophercited by 1,199
- 2023The RefinedWeb Dataset for Falcon LLM: Outperforming Curated Corpora with Web Data, and Web Data Onlycited by 755
- 2023Textbooks Are All You Need II: phi-1.5 technical reportcited by 559
- 2023The Falcon Series of Open Language Modelscited by 526
- 2019The frontier of simulation-based inferencecited by 490
30 more
- 2022PDEBENCH: An Extensive Benchmark for Scientific Machine Learningcited by 396
- 2023OBELICS: An Open Web-Scale Filtered Dataset of Interleaved Image-Text Documentscited by 316
- 2022GraphCast: Learning skillful medium-range global weather forecastingcited by 261
- 2023Convolutional Neural Operators for robust and accurate learning of PDEscited by 207
- 2022Towards Multi-spatiotemporal-scale Generalized PDE Modelingcited by 190
- 2021Exact imposition of boundary conditions with distance functions in physics-informed deep neural networkscited by 135
- 2023Towards Foundation Models for Scientific Machine Learning: Characterizing Scaling and Transfer Behaviorcited by 128
- 2023In-Context Operator Learning with Data Prompts for Differential Equation Problemscited by 98
- 2023Multiple Physics Pretraining for Physical Surrogate Modelscited by 90
- 2023PINNacle: A Comprehensive Benchmark of Physics-Informed Neural Networks for Solving PDEscited by 84
- 2024Croissant: A Metadata Format for ML-Ready Datasetscited by 79
- 2022AirfRANS: High Fidelity Computational Fluid Dynamics Dataset for Approximating Reynolds-Averaged Navier-Stokes Solutionscited by 77
- 2023Multi-Grid Tensorized Fourier Neural Operator for High-Resolution PDEscited by 66
- 2023Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulationcited by 65
- 2023Learning Physical Models that Can Respect Conservation Lawscited by 58
- 2024Pretraining Codomain Attention Neural Operators for Solving Multiphysics PDEscited by 58
- 2023Eagle: Large-Scale Learning of Turbulent Fluid Dynamics with Mesh Transformerscited by 54
- 2023Towards Stability of Autoregressive Neural Operatorscited by 54
- 2024Towards a Foundation Model for Partial Differential Equations: Multi-Operator Learning and Extrapolationcited by 45
- 2023LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suitecited by 26
- 2022Guiding continuous operator learning through Physics-based boundary constraintscited by 21
- 2021Effects of boundary conditions in fully convolutional networks for learning spatio-temporal dynamicscited by 14
- 2023BubbleML: A Multi-Physics Dataset and Benchmarks for Machine Learningcited by 14
- 20233D-Spatiotemporal Forecasting the Expansion of Supernova Shells Using Deep Learning toward High-Resolution Galaxy Simulationscited by 12
- 2023Turbulence in Focus: Benchmarking Scaling Behavior of 3D Volumetric Super-Resolution with BLASTNet 2.0 Datacited by 11
- 2023Learning fast, accurate, and stable closures of a kinetic theory of an active fluidcited by 9
- 2023Invariant preservation in machine learned PDE solvers via error correctioncited by 7
- 2023Surrogate Modeling for Computationally Expensive Simulations of Supernovae in High-Resolution Galaxy Simulationscited by 6
- 2023Training Deep Surrogate Models with Large Scale Online Learningcited by 5
- 2023Multi-scale clustering and source separation of InSight mission seismic datacited by 3
Built on by
Citing papers per year: 2024: 1; 2025: 60; 2026: 73.
134 papers.
Sort:
By area:
- 2025EXP-Bench: Can AI Conduct AI Research Experiments?cited by 36
- 2025PDE-Transformer: Efficient and Versatile Transformers for Physics Simulationscited by 29
- 2025Generative Latent Neural PDE Solver using Flow Matchingcited by 28
- 2025Walrus: A Cross-Domain Foundation Model for Continuum Dynamicscited by 26
- 2025Reduced Order Modeling with Shallow Recurrent Decoder Networkscited by 24
- 2025DISCO: learning to DISCover an evolution Operator for multi-physics-agnostic predictioncited by 21
- 2025PhysiX: A Foundation Model for Physics Simulationscited by 21
- 2025FunDiff: Diffusion Models over Function Spaces for Physics-Informed Generative Modelingcited by 19
- 2025Lost in Latent Space: An Empirical Study of Latent Diffusion Models for Physics Emulationcited by 19
- 2025Defining Foundation Models for Computational Science: A Call for Clarity and Rigorcited by 16
- 2026RealPDEBench: A Benchmark for Complex Physical Systems with Real-World Datacited by 13
- 2025MORPH: PDE Foundation Models with Arbitrary Data Modalitycited by 12
38 more
- 2025Principled Approaches for Extending Neural Architectures to Function Spaces for Operator Learningcited by 11
- 2024Predicting Change, Not States: An Alternate Framework for Neural PDE Surrogatescited by 8
- 2025FourierFlow: Frequency-aware Flow Matching for Generative Turbulence Modelingcited by 8
- 2026Test-time Generalization for Physics through Neural Operator Splittingcited by 8
- 2025TANTE: Time-Adaptive Operator Learning via Neural Taylor Expansioncited by 7
- 2025Towards autonomous quantum physics research using LLM agents with access to intelligent toolscited by 7
- 2025SIMSHIFT: A Benchmark for Adapting Neural Surrogates to Distribution Shiftscited by 6
- 2025EngiBench: A Framework for Data-Driven Engineering Design Researchcited by 6
- 2025EddyFormer: Accelerated Neural Simulations of Three-Dimensional Turbulence at Scalecited by 6
- 2025When is a System Discoverable from Data? Discovery Requires Chaoscited by 6
- 2025Towards a Foundation Model for Partial Differential Equations Across Physics Domainscited by 6
- 2025Towards a Physics Foundation Modelcited by 5
- 2026Structure-Preserving Learning Improves Geometry Generalization in Neural PDEscited by 5
- 2025FD-Bench: A Modular and Fair Benchmark for Data-driven Fluid Simulationcited by 4
- 2025T-SHRED: Symbolic Regression for Regularization and Model Discovery with Transformer Shallow Recurrent Decoderscited by 4
- 2025Physics-Grounded Motion Forecasting via Equation Discovery for Trajectory-Guided Image-to-Video Generationcited by 4
- 2025TC-LoRA: Temporally Modulated Conditional LoRA for Adaptive Diffusion Controlcited by 4
- 2026DiffusionRollout: Uncertainty-Aware Rollout Planning in Long-Horizon PDE Solvingcited by 4
- 2026CTF4Nuclear: Common Task Framework for Nuclear Fission and Fusion Modelscited by 4
- 2025Surrogate Modeling of 3D Rayleigh-Benard Convection with Equivariant Autoencoderscited by 3
- 2025AI for Scientific Discovery is a Social Problemcited by 3
- 2025AutoHood3D: A Multi-Modal Benchmark for Automotive Hood Design and Fluid-Structure Interactioncited by 3
- 2025Physics Steering: Causal Control of Cross-Domain Concepts in a Physics Foundation Modelcited by 3
- 2025AI Benchmark Democratization and Carpentrycited by 3
- 2026LEVDA: Latent Ensemble Variational Data Assimilation via Differentiable Dynamicscited by 3
- 2026MENO: MeanFlow-Enhanced Neural Operators for Dynamical Systemscited by 3
- 2026Discovering Physical Directions in Weight Space: Composing Neural PDE Expertscited by 3
- 2026Emergent Transfer of a Physics Foundation Model from Simulation to Laboratory Turbulencecited by 3
- 2025Beyond Accuracy: EcoL2 Metric for Sustainable Neural PDE Solverscited by 2
- 2025Autoregressive regularized score-based diffusion models for multi-scenarios fluid flow predictioncited by 2
- 2025An MLCommons Scientific Benchmarks Ontologycited by 2
- 2025MIMIC-MJX: Neuromechanical Emulation of Animal Behaviorcited by 2
- 2025Efficient Generative Transformer Operators For Million-Point PDEscited by 2
- 2025SynthPix: A lightspeed PIV image generatorcited by 2
- 2026Streaming Operator Inference for Model Reduction of Large-Scale Dynamical Systemscited by 2
- 2026Latent Generative Solvers for Generalizable Long-Term Physics Simulationcited by 2
- 2026Semigroup Consistency as a Diagnostic for Learned Physics Simulatorscited by 2
- 2025Data-Driven Forecasting of High-Dimensional Transient and Stationary Processes via Space-Time Projectioncited by 1
50 most cited of 134 All 134 (JSON)
- 2026Transferability of Learned States in Neural PDE Solvers7 Oct 2026
- 2026Cross-Domain Pretraining for Steady-State Neural CFD Surrogates7 Oct 2026
- 2026Scale-Split Neural Operator for Memory- and Data-Efficient 3D Turbulence Prediction30 Sep 2026
- 2026Does Text Steer Neural PDE Surrogates? A Controlled Diagnostic with OperatorCLIP29 Sep 2026
- 2026PhysFieldBench: Can Multimodal Models Understand Physical Fields?28 Sep 2026
- 2026Energy-efficient operation of neural operators for virtual sensing24 Sep 2026
- 2026Beyond Compression: Training Latent Representations for Stable Long-Horizon Rollout in Neural Surrogate Solvers24 Sep 2026
- 2026RD-JEPA: Predictive latent pretraining for few-trajectory transfer across reaction--diffusion equations24 Sep 2026
- 2026Large Distant Gradients Need Not Be Reliable: reliability-weighted credit assignment for long-horizon autoregressive forecasting11 Sep 2026
- 2026Geometry-aware Latent Autoregressive Generative Model for PDEs in Complex Domains31 Aug 2026
- 2026Beyond Pairwise Graphs in Science: Hypergraph Adaptive Wavelet Operators for Parametric PDEs28 Aug 2026
- 2026Dandelion: A Spherical Flower for Neural Simulation of Planetary Dynamics27 Aug 2026
38 more
- 2026PhysicsBench: A Unified Leaderboard for Generative and Predictive Models in Engineering Design and Simulation25 Aug 2026
- 2026Response Renormalization for Critical Deep Equilibrium Models24 Aug 2026
- 2026When Does AI for PDEs Yield Scientific Evidence?23 Aug 2026
- 2026From Fixed Grids to Moving Particles:A Transferable Latent Operator for Fluid Dynamics14 Aug 2026
- 2026Distillation of Foundation Models for Time-dependent PDEs12 Aug 2026
- 2026Unsupervised Adaptation of PDE Foundation Models7 Aug 2026
- 2026Kastor: An efficient fine-tuning strategy for generative emulation of PDE simulations6 Aug 2026
- 2026TIDE: A Physically Diverse 3D Turbulence Benchmark Dataset for Advancing Scientific Machine Learning4 Aug 2026
- 2026Round-Trip Consistency: Bidirectional Diffusion Models Can Predict Their Own Rollout Errors1 Aug 2026
- 2026Physics Transformer: Tailoring Transformer for General PDE Prediction27 Jul 2026
- 2026Generalized Neural Operator for Parametric and Boundary-Value Problems24 Jul 2026
- 2026Multilevel Graph Wavelet Compressed Sensing with Scale-Aware Neural Recovery23 Jul 2026
- 2026EVOLVE: Efficient Learned Volume Compression with Variable-Rate Encoding on a Cross-Domain Database20 Jul 2026
- 2026LiNO: Lifting based multiresolution neural operator2 Jul 2026
- 2026Native Multi-Dimensional Subquadratic Operators via Input Dependent Long Convolutions1 Jul 2026
- 2026Higher-Order Fourier Neural Operator: Explicit Mode Mixer for Nonlinear PDEs26 Jun 2026
- 2026Advances in Scientific Machine Learning for Coupled Fluid Flow and Transport17 Jun 2026
- 2026ThousandWorlds: A benchmark for climate emulation of potentially habitable exoplanets16 Jun 2026
- 2026Reliability of Probabilistic Emulation of Physical Systems11 Jun 2026
- 2026MeshTok: Efficient Multi-Scale Tokenization for Scalable PDE Transformers3 Jun 2026
- 2026Emergent Transfer of a Physics Foundation Model from Simulation to Laboratory Turbulence31 May 2026
- 2026Cellular Sheaf Neural Operators for Structure-Preserving Surrogate Modeling of Constrained PDEs31 May 2026
- 2026Recursive Flow Matching26 May 2026
- 2026Semigroup Consistency as a Diagnostic for Learned Physics Simulators25 May 2026
- 2026Small Models, Strong Priors: Architectural Inductive Bias for Parameter-Efficient Neural PDE Solvers25 May 2026
- 2026NPSolver: Neural Poisson Solver with Iterative Physics Supervision25 May 2026
- 2026FLOATBench: A Dataset and Benchmark for Floating Offshore Wind Turbine Tower Fatigue25 May 2026
- 2026Iterative Refinement Neural Operators are Learned Fixed-Point Solvers: A Principled Approach to Spectral Bias Mitigation21 May 2026
- 2026Harnessing AI for Inverse Partial Differential Equation Problems: Past, Present, and Prospects16 May 2026
- 2026Wavelet Flow Matching for Multi-Scale Physics Emulation15 May 2026
- 2026Multi-Fidelity Flow Matching: Cascaded Refinement of PDE Solutions15 May 2026
- 2026CTF4Nuclear: Common Task Framework for Nuclear Fission and Fusion Models15 May 2026
- 2026Breakeven complexity: A new perspective on neural partial differential equation solvers14 May 2026
- 2026Eradicating Negative Transfer in Multi-Physics Foundation Models via Sparse Mixture-of-Experts Routing14 May 2026
- 2026Discovering Physical Directions in Weight Space: Composing Neural PDE Experts14 May 2026
- 2026Bridging Spectral Operator Learning and U-Net Hierarchies: SpectraNet for Stable Autoregressive PDE Surrogates9 May 2026
- 2026CarCrashNet: A Large-Scale Dataset and Hierarchical Neural Solver for Data-Driven Structural Crash Simulation8 May 2026
- 2026A Robust In-Context Model for Conservation Laws: Injecting Context into Flux Neural Operators via Recurrent Vision Transformers6 May 2026
50 newest of 134 All 134 (JSON)
- 2025Generative Latent Neural PDE Solver using Flow Matchingcited by 28
- 2025FunDiff: Diffusion Models over Function Spaces for Physics-Informed Generative Modelingcited by 19
- 2025Lost in Latent Space: An Empirical Study of Latent Diffusion Models for Physics Emulationcited by 19
- 2025FourierFlow: Frequency-aware Flow Matching for Generative Turbulence Modelingcited by 8
- 2025TC-LoRA: Temporally Modulated Conditional LoRA for Adaptive Diffusion Controlcited by 4
- 2026DiffusionRollout: Uncertainty-Aware Rollout Planning in Long-Horizon PDE Solvingcited by 4
- 2026MENO: MeanFlow-Enhanced Neural Operators for Dynamical Systemscited by 3
- 2025Autoregressive regularized score-based diffusion models for multi-scenarios fluid flow predictioncited by 2
- 2025Efficient probabilistic surrogate modeling techniques for partially-observed large-scale dynamical systemscited by 1
- 2026Cosmo3DFlow: Wavelet Flow Matching for Spatial-to-Spectral Compression in Reconstructing the Early Universecited by 1
- 2025GeoFunFlow: Geometric Function Flow Matching for Inverse Operator Learning over Complex Geometriescited by 0
- 2026Multi-Fidelity Flow Matching: Cascaded Refinement of PDE Solutionscited by 0
12 most cited of 17 All 17 (JSON)
- 2026Geometry-aware Latent Autoregressive Generative Model for PDEs in Complex Domains31 Aug 2026
- 2026Round-Trip Consistency: Bidirectional Diffusion Models Can Predict Their Own Rollout Errors1 Aug 2026
- 2026Reliability of Probabilistic Emulation of Physical Systems11 Jun 2026
- 2026Recursive Flow Matching26 May 2026
- 2026Wavelet Flow Matching for Multi-Scale Physics Emulation15 May 2026
- 2026Multi-Fidelity Flow Matching: Cascaded Refinement of PDE Solutions15 May 2026
- 2026MENO: MeanFlow-Enhanced Neural Operators for Dynamical Systems8 Apr 2026
- 2026DiffusionRollout: Uncertainty-Aware Rollout Planning in Long-Horizon PDE Solving14 Feb 2026
- 2026Cosmo3DFlow: Wavelet Flow Matching for Spatial-to-Spectral Compression in Reconstructing the Early Universe10 Feb 2026
- 2025Efficient probabilistic surrogate modeling techniques for partially-observed large-scale dynamical systems6 Nov 2025
- 2025TC-LoRA: Temporally Modulated Conditional LoRA for Adaptive Diffusion Control10 Oct 2025
- 2025GeoFunFlow: Geometric Function Flow Matching for Inverse Operator Learning over Complex Geometries28 Sep 2025
12 newest of 17 All 17 (JSON)
- 2025EXP-Bench: Can AI Conduct AI Research Experiments?cited by 36
- 2025PDE-Transformer: Efficient and Versatile Transformers for Physics Simulationscited by 29
- 2025Walrus: A Cross-Domain Foundation Model for Continuum Dynamicscited by 26
- 2025Reduced Order Modeling with Shallow Recurrent Decoder Networkscited by 24
- 2025DISCO: learning to DISCover an evolution Operator for multi-physics-agnostic predictioncited by 21
- 2025PhysiX: A Foundation Model for Physics Simulationscited by 21
- 2025Defining Foundation Models for Computational Science: A Call for Clarity and Rigorcited by 16
- 2026RealPDEBench: A Benchmark for Complex Physical Systems with Real-World Datacited by 13
- 2025MORPH: PDE Foundation Models with Arbitrary Data Modalitycited by 12
- 2025Principled Approaches for Extending Neural Architectures to Function Spaces for Operator Learningcited by 11
- 2024Predicting Change, Not States: An Alternate Framework for Neural PDE Surrogatescited by 8
- 2026Test-time Generalization for Physics through Neural Operator Splittingcited by 8
12 most cited of 112 All 112 (JSON)
- 2026Transferability of Learned States in Neural PDE Solvers7 Oct 2026
- 2026Cross-Domain Pretraining for Steady-State Neural CFD Surrogates7 Oct 2026
- 2026Scale-Split Neural Operator for Memory- and Data-Efficient 3D Turbulence Prediction30 Sep 2026
- 2026Does Text Steer Neural PDE Surrogates? A Controlled Diagnostic with OperatorCLIP29 Sep 2026
- 2026Energy-efficient operation of neural operators for virtual sensing24 Sep 2026
- 2026Beyond Compression: Training Latent Representations for Stable Long-Horizon Rollout in Neural Surrogate Solvers24 Sep 2026
- 2026RD-JEPA: Predictive latent pretraining for few-trajectory transfer across reaction--diffusion equations24 Sep 2026
- 2026Large Distant Gradients Need Not Be Reliable: reliability-weighted credit assignment for long-horizon autoregressive forecasting11 Sep 2026
- 2026Beyond Pairwise Graphs in Science: Hypergraph Adaptive Wavelet Operators for Parametric PDEs28 Aug 2026
- 2026Dandelion: A Spherical Flower for Neural Simulation of Planetary Dynamics27 Aug 2026
- 2026PhysicsBench: A Unified Leaderboard for Generative and Predictive Models in Engineering Design and Simulation25 Aug 2026
- 2026Response Renormalization for Critical Deep Equilibrium Models24 Aug 2026
12 newest of 112 All 112 (JSON)