publications

Publications in reversed chronological order. For a complete list, see Google Scholar.

2026

  1. arXiv
    Stop Training for the Worst: Progressive Unmasking Accelerates Masked Diffusion Training
    Jaeyeon Kim*, Jonathan Geuter*, David Alvarez-Melis, Sham Kakade, and Sitan Chen
    2026
    In submission
  2. ICLR
    Boomerang Distillation Enables Zero-Shot Model Size Interpolation
    Sara Kangaslahti, Nihal V. Nayak*, Jonathan Geuter*, Marco Fumero, Francesco Locatello, and David Alvarez-Melis
    Proceedings of the International Conference on Learning Representations (ICLR) , 2026
    Previously accepted at 3rd Workshop on Unifying Representations in Neural Models at NeurIPS 2025
  3. ICLR
    Guided Speculative Inference for Efficient Test-Time Alignment of LLMs
    Jonathan Geuter, Youssef Mroueh, and David Alvarez-Melis
    Proceedings of the International Conference on Learning Representations (ICLR) , 2026
    Previously accepted at 3rd Workshop for Efficient Systems for Foundation Models at ICML 2025 (Spotlight)

2025

  1. NeurIPS (Workshop)
    RoBoN: Routed Online Best-of-n for Test-Time Scaling with Multiple LLMs
    Jonathan Geuter and Gregor Kornhardt
    1st Workshop on Foundations of Reasoning in Language Models at NeurIPS 2025, San Diego, USA , 2025
  2. ICML (Workshop)
    Entropy-Driven Pre-Tokenization for Byte-Pair Encoding
    Yifan Hu, Frank Liang, Dachuan Zhao, Jonathan Geuter, Varshini Reddy, Craig W. Schmidt, and Chris Tanner
    Tokenization Workshop at ICML , 2025
  3. AISTATS
    DDEQs: Distributional Deep Equilibrium Models through Wasserstein Gradient Flows
    Jonathan Geuter, Clément Bonet, Anna Korba, and David Alvarez-Melis
    The 28th International Conference on Artificial Intelligence and Statistics (AISTATS) , 2025
  4. ICML
    Universal Neural Optimal Transport
    Jonathan Geuter, Gregor Kornhardt, Ingimar Tomasson, and Vaios Laschos
    Forty-second International Conference on Machine Learning (ICML) , 2025

2023

  1. EMNLP (Workshop)
    Jina Embeddings: A Novel Set of High-Performance Sentence Embedding Models
    Michael Günther, Louis Milliken, Jonathan Geuter, Georgios Mastrapas, Bo Wang, and Han Xiao
    3rd Workshop for Natural Language Processing Open Source Software at EMNLP , 2023

2022

  1. A Sinkhorn-NN Hybrid Algorithm for Optimal Transport
    Jonathan Geuter
    Technische Universität Berlin and Weierstrass Institute. Master’s thesis, supervised by Dr. Vaios Laschos and Prof. Dr. Martin Skutella , 2022

2020

  1. Nonlinear Korn Inequalities
    Jonathan Geuter
    Bachelor’s thesis, supervised by Prof. Dr. Barbara Zwicknagl and Dr. Hans-Christian Kreusler , 2020