PDE-JEPA: Predictive Representation Learning of Latent Dynamics Modeling for Parametric PDEs
PDE-JEPA uses masked-latent predictive pretraining with geometry alignment for parametric PDE dynamics, beating SOTA by 33.4% in-distribution and 51.4% extrapolating.