PyTorch
flow-matching
wavelets
physics-emulation
pde
generative-model
neurips-2026