Klein, Michal

17 publications

ICLR 2026 Completed Hyperparameter Transfer Across Modules, Width, Depth, Batch and Duration Bruno Kacper Mlodozeniec, Pierre Ablin, Louis Béthune, Dan Busbridge, Michal Klein, Jason Ramapuram, Marco Cuturi

ICLR 2026 Flow Matching with Semidiscrete Couplings Alireza Mousavi-Hosseini, Stephen Y. Zhang, Michal Klein, Marco Cuturi

TMLR 2026 Multivariate Conformal Prediction Using Optimal Transport Michal Klein, Louis Béthune, Eugene Ndiaye, Marco Cuturi

TMLR 2026 On Fitting Flow Models with Large Sinkhorn Couplings Stephen Y. Zhang, Alireza Mousavi-Hosseini, Michal Klein, Marco Cuturi

ICLR 2025 Controlling Language and Diffusion Models by Transporting Activations Pau Rodriguez, Arno Blaas, Michal Klein, Luca Zappella, Nicholas Apostoloff, Marco Cuturi, Xavier Suau

NeurIPS 2025 LinEAS: End-to-End Learning of Activation Steering with a Distributional Loss Pau Rodriguez, Michal Klein, Eleonora Gualdoni, Valentino Maiorca, Arno Blaas, Luca Zappella, Marco Cuturi, Xavier Suau

CVPR 2025 Multimodal Autoregressive Pre-Training of Large Vision Encoders Enrico Fini, Mustafa Shukor, Xiujun Li, Philipp Dufter, Michal Klein, David Haldimann, Sai Aitharaju, Victor G. Turrisi da Costa, Louis Béthune, Zhe Gan, Alexander Toshev, Marcin Eichner, Moin Nabi, Yinfei Yang, Joshua Susskind, Alaaeldin El-Nouby

ICLR 2025 Simple ReFlow: Improved Techniques for Fast Flow Models Beomsu Kim, Yu-Guan Hsieh, Michal Klein, Marco Cuturi, Jong Chul Ye, Bahjat Kawar, James Thornton

ICML 2024 Careful with That Scalpel: Improving Gradient Surgery with an EMA Yu-Guan Hsieh, James Thornton, Eugene Ndiaye, Michal Klein, Marco Cuturi, Pierre Ablin

ICML 2024 Contrasting Multiple Representations with the Multi-Marginal Matching Gap Zoe Piran, Michal Klein, James Thornton, Marco Cuturi

NeurIPS 2024 Learning Elastic Costs to Shape Monge Displacements Michal Klein, Aram-Alexandre Pooladian, Pierre Ablin, Eugène Ndiaye, Jonathan Niles-Weed, Marco Cuturi

NeurIPS 2024 Progressive Entropic Optimal Transport Solvers Parnian Kassraie, Aram-Alexandre Pooladian, Michal Klein, James Thornton, Jonathan Niles-Weed, Marco Cuturi

ICML 2024 Scalable Pre-Training of Large Autoregressive Image Models Alaaeldin El-Nouby, Michal Klein, Shuangfei Zhai, Miguel Ángel Bautista, Vaishaal Shankar, Alexander T Toshev, Joshua M. Susskind, Armand Joulin

ICML 2023 Monge, Bregman and Occam: Interpretable Optimal Transport in High-Dimensions with Feature-Sparse Maps Marco Cuturi, Michal Klein, Pierre Ablin

ICML 2023 Predicting Ordinary Differential Equations with Transformers Sören Becker, Michal Klein, Alexander Neitz, Giambattista Parascandolo, Niki Kilbertus

NeurIPS 2023 Unbalanced Low-Rank Optimal Transport Solvers Meyer Scetbon, Michal Klein, Giovanni Palla, Marco Cuturi

NeurIPSW 2022 Discovering Ordinary Differential Equations That Govern Time-Series Sören Becker, Michal Klein, Alexander Neitz, Giambattista Parascandolo, Niki Kilbertus