Just Align $\bm{x}$: Aligning Predictions, Not Representations
JAx (Just Align x), a prediction-supervision method that aligns clean-image predictions across noise levels, shows that prediction-space supervision provides a simple and principled alternative to representation alignment for pixel-space generative models.