omniDiffusion-PDE generates coupled PDE solutions using latent diffusion models trained only on inexpensive decoupled single-field simulations. Independent conditional models are composed with Product-of-Experts inference and iterative coupling schemes to produce physically consistent, scalable spatiotemporal solutions and long time-series. -
View it on GitHub