Branch-JEPA: Finite-Support Predictive Distributions for JEPA World Models
Branch-JEPA is introduced, which replaces this point-valued transition with a context-weighted finite set of latent successors, and preserves more distinct futures, while full-set scoring improves the quality of the resulting predictive distribution.