Anima Anandkumar on One Model for All of Physics
Anima Anandkumar and co-founder Benedikt Jenik are building Accelerated Understanding, a single foundation model for the physical world: many domains of physics in one model, trained on data from numerical simulators and improved by the laws of physics themselves, using neural operators to reach context lengths of trillions where transformers cannot.
Everything Into One Model
Language models won by pushing every task into a single general model instead of many narrow ones, and Accelerated Understanding is making the same bet for physical simulation.
let's just put everything into one model and that turned out to work phenomenally well and exceed pretty much everything else those specialized models. And so now we were asking ourselves, can we do the same thing for physical simulation and for physical understanding?
You Cannot Download New Physics
Scientific discovery is by definition not in any dataset, so a purely data-driven model tops out at the average of its data, and the laws of physics become both the extra teacher and the way past that ceiling.
if you think about scientific discovery, it's about doing something new. So, by definition, it cannot be there in the training data. And so, this reliance on just purely datadriven AI is not going to be enough. And that's where you know adding the laws of physics is really critical.
Same Equations, Different Regimes
Fluid in a catheter and fluid in a rocket obey the same equations at different settings, and deeper still, physics shares features like conservation, causality, and time, so one model can carry learning across very different domains.
there is something that is shared among them and that's what we've seen this play out that across different domains if you think energy semiconductor aerospace you know it looks very different but there are common underlying principles
Multi-Physics Beats Specialization
A single model trained on several kinds of physics outperforms separate models of the same size each trained on one, which points to shared, emergent learning rather than just more parameters.
it turns out that having the model of the same size with multiple areas of physics does better than giving all of those parameters to each single physics. So if you had separate models and made them big enough as the original one, it still is worse.
Four Dimensions, Trillions of Tokens
Physical data grows across three spatial dimensions plus time, all independently, so a single sample can reach trillions of elements, far past the roughly million-token contexts language models handle.
when you multiply those numbers out, you're very quickly in the billions or even trillions in context. And that is something we've actually achieved with our models. Like we're able to train up to a trillion context input.
Neural Operators, Not Transformers
Transformer attention scales quadratically and cannot reach trillions of context, while neural operators stay resolution-invariant, so the same model can run a cheap low-detail pass or the most demanding physics.
using transformer architectures that have worked so well for language that just wouldn't be able to support a 5 trillion context length no matter all the compute in the world is thrown at it.
One Sample Will Not Fit a GPU
The models' layers and even single training samples are too large to fit on one accelerator or a full node, so standard sharding like FSDP breaks and the team rebuilt the sharding strategy from scratch.
our layers are so big you can't reassemble it inside a GPU. Our data samples are so big they don't fit an accelerator or even a full node. So we had to reinvent this whole sharding infrastructure, the whole sharding strategy
Physics Moves Into the Design Loop
In chip design, today's pipeline freezes the digital design and runs physics as a one-time check, leaving performance headroom that a physics model kept in the loop could use.
we're not taking advantage today of that headroom to get extra performance. So there's this potential to unlock much more of the skill set when you're also able to grab into what does the physics allow me to do to push the envelope a bit more