You can deploy this model and start earning money today!

sogeeking/
disco-models

Views

No views yet

DISCO model checkpoints

Pretrained checkpoints for the ICML 2026 paper Test-Time Generalization via Neural Operator Splitting (Serrano et al.).

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/last.ckpt`

Load with PyTorch Lightning:

python

from train.train_generic import DISCOLitModule
model = DISCOLitModule.load_from_checkpoint("last.ckpt")