Diamond Maps: Efficient Reward Alignment via Stochastic Flow Maps. - Peter Holderrieth; Douglas Chen; Luca Eyring; Ishin Shah; Giri Anantharaman; Yutong He; Zeynep Akata; Tommi S. Jaakkola; Nicholas Matthew Boffi; Max Simchowitz (CoRR, 2026)