AREX: Affine-Residual Exponential Integrator for Few-Step Sampling in Flow Matching

We introduce AREX, a training-free sampler for pretrained flow matching models that uses the target mean and covariance to capture an analytically tractable part of the sampling dynamics.