MDN-VJEPA: Multimodal Probabilistic Joint-Embedding World Models

MDN-VJEPA is a multimodal probabilistic JEPA that uses a mixture density network to represent future states with a Gaussian mixture encoding alternative futures, their probabilities, and within-mode uncertainty.