Variational Flow Maps:
Make Some Noise for One-Step Conditional Generation

Abbas Mammadov^1,*    So Takao^2,*    Bohan Chen²    Ricardo Baptista³    Morteza Mardani⁴    Yee Whye Teh¹    Julius Berner⁴

¹Department of Statistics, University of Oxford    ²California Institute of Technology
³University of Toronto    ⁴NVIDIA
^*Equal contribution

Official Implementation of "Variational Flow Maps: Make Some Noise for One-Step Conditional Generation"

Code will be released soon! 🏗️

Citation

If you find this work useful, please cite our paper:

@misc{mammadov2026variationalflowmapsmake,
      title={Variational Flow Maps: Make Some Noise for One-Step Conditional Generation}, 
      author={Abbas Mammadov and So Takao and Bohan Chen and Ricardo Baptista and Morteza Mardani and Yee Whye Teh and Julius Berner},
      year={2026},
      eprint={2603.07276},
      archivePrefix={arXiv},
      primaryClass={cs.CV},
      url={[https://arxiv.org/abs/2603.07276](https://arxiv.org/abs/2603.07276)}, 
}