• Propagates Wasserstein distances through graph structures to capture higher-order uncertainty.
  • Demonstrates gains on synthetic physics datasets and molecular property prediction tasks.