Function graph transformers lift functions to graph measures to universally approximate nonlinear operators between function spaces via standard attention and MLPs.
Njord is a probabilistic graph neural network that produces efficient ensemble ocean forecasts via single-pass sampling, achieving lowest average upper-ocean errors globally and in the Baltic Sea.
Spatiotemporal Noise-Contrastive Estimation learns energy-based models via joint spatiotemporal differences to avoid failure modes of spatial or temporal methods alone, matching state-of-the-art density estimation.
SCALLOP introduces a Hutchinson-free likelihood distillation objective for few-step Boltzmann generators, reducing training variance and time while achieving up to 10x inference speedup.