metadata
license: mit
tags:
- pytorch-lightning
- neural-pde-solver
- operator-splitting
DISCO model checkpoints
Pretrained checkpoints for the ICML 2026 paper Test-time Generalization for Physics through Neural Operator Splitting (Serrano, Han, Oyallon, Ho, Morel).
| Equation | Checkpoint |
|---|---|
| Advection-diffusion (1D) | advection-diffusion/last.ckpt |
| Combined equation (1D) | combined-equation/last.ckpt |
| Reaction-diffusion (Gray-Scott, 2D) | reaction-diffusion/last.ckpt |
| Navier-Stokes / Euler (2D) | navier-stokes/best-checkpoint.ckpt |
Load with PyTorch Lightning:
from train.train_generic import DISCOLitModule
model = DISCOLitModule.load_from_checkpoint("last.ckpt")