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

Load with PyTorch Lightning:

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

Downloads last month: -; Downloads are not tracked for this model. How to track

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support