sogeeking/

disco-rd

Trajectories used in the ICML 2026 paper Test-time Generalization for Physics through Neural Operator Splitting (Serrano, Han, Oyallon, Ho, Morel).

Each HDF5 file contains:

trajectories: shape (N, T, C, spatial), float32 env_id: shape (N,), int64 — environment index for each trajectory env_params/: optional metadata mapping env_id back to PDE coefficients

Load via the project's train_generic.py / test_generic.py scripts: from… See the full description on the dataset page: https://huggingface.co/datasets/sogeeking/disco-rd.