Kohler, Jonas

14 publications

AAAI 2025 Adaptive Guidance: Training-Free Acceleration of Conditional Diffusion Models Angela Castillo, Jonas Kohler, Juan C. Pérez, Juan Pablo Pérez, Albert Pumarola, Bernard Ghanem, Pablo Arbeláez, Ali K. Thabet

CVPR 2025 Autoregressive Distillation of Diffusion Transformers Yeongmin Kim, Sotiris Anagnostidis, Yuming Du, Edgar Schönfeld, Jonas Kohler, Markos Georgopoulos, Albert Pumarola, Ali Thabet, Artsiom Sanakoyeu

CVPR 2025 FlexiDiT: Your Diffusion Transformer Can Easily Generate High-Quality Samples with Less Compute Sotiris Anagnostidis, Gregor Bachmann, Yeongmin Kim, Jonas Kohler, Markos Georgopoulos, Artsiom Sanakoyeu, Yuming Du, Albert Pumarola, Ali Thabet, Edgar Schönfeld

CVPR 2024 Cache Me if You Can: Accelerating Diffusion Models Through Block Caching Felix Wimbauer, Bichen Wu, Edgar Schoenfeld, Xiaoliang Dai, Ji Hou, Zijian He, Artsiom Sanakoyeu, Peizhao Zhang, Sam Tsai, Jonas Kohler, Christian Rupprecht, Daniel Cremers, Peter Vajda, Jialiang Wang

ICML 2023 Rigid Body Flows for Sampling Molecular Crystal Structures Jonas Köhler, Michele Invernizzi, Pim De Haan, Frank Noe

AISTATS 2022 Vanishing Curvature in Randomly Initialized Deep ReLU Networks Antonio Orvieto, Jonas Kohler, Dario Pavllo, Thomas Hofmann, Aurelien Lucchi

ICCV 2021 Learning Generative Models of Textured 3D Meshes from Real-World Images Dario Pavllo, Jonas Kohler, Thomas Hofmann, Aurelien Lucchi

NeurIPS 2021 Smooth Normalizing Flows Jonas Köhler, Andreas Krämer, Frank Noe

NeurIPS 2020 Batch Normalization Provably Avoids Ranks Collapse for Randomly Initialised Deep Networks Hadi Daneshmand, Jonas Kohler, Francis R. Bach, Thomas Hofmann, Aurelien Lucchi

NeurIPS 2020 Stochastic Normalizing Flows Hao Wu, Jonas Köhler, Frank Noe

AISTATS 2019 Exponential Convergence Rates for Batch Normalization: The Power of Length-Direction Decoupling in Non-Convex Optimization Jonas Kohler, Hadi Daneshmand, Aurelien Lucchi, Thomas Hofmann, Ming Zhou, Klaus Neymeyr

UAI 2019 The Role of Memory in Stochastic Optimization Antonio Orvieto, Jonas Kohler, Aurelien Lucchi

ICML 2018 Escaping Saddles with Stochastic Gradients Hadi Daneshmand, Jonas Kohler, Aurelien Lucchi, Thomas Hofmann

ICLR 2018 Spherical CNNs Taco S. Cohen, Mario Geiger, Jonas Köhler, Max Welling