Ax Ali Taghibakhshi, Ruisi Cai, Saurav Muralidharan, Sharath Turuvekere Sreenivas, Aditya Vavre, Ameya Sunil Mahabaleshwarkar, Bilal Kartal, Sheldon Liang, Marcin Chochowski, Zijia Chen, Akhiad Bercovich, Ran Zilberstein, Ran El-Yaniv, Yonatan Geifman, Daniel Korzekwa, Yoshi Suhara, Oluwatobi Olabiyi, Ashwath Aithal, Nima Tajbakhsh, Pavlo Molchanov 5/11/2026

Star Elastic: Many-in-One Reasoning LLMs with Efficient Budget Control

Post-training method for creating multiple nested reasoning LLM variants efficiently with computational budget control.

Ax Fred Zhangzhi Peng, Avishek Joey Bose, Anru R. Zhang, Alexander Tong 5/11/2026

Coupling Models for One-Step Discrete Generation

One-step generative model for discrete sequences using coupling between discrete structures and Gaussian latent variables.

Ax Ryan Thompson, He Zhao, Daniel M. Steinberg, Edwin V. Bonilla 5/11/2026

Arrow: A Foundation Model for Causal Discovery

Foundation model for zero-shot causal discovery on tabular data using transformer architecture with structured graph inference.

Ax Johan Mathe, Adele Myers, Simon Mataigne, Nina Miolane 5/11/2026

bispectrum: Selective $G$-Bispectra Made Practical

Practical implementation of G-bispectrum group invariants for machine learning on signals, images, and spherical data with applications to classification and pooling layers.

Ax Junfeng Fang, Zhepei Hong, Mao Zheng, Mingyang Song, Gengsheng Li, Houcheng Jiang, Dan Zhang, Haiyun Guo, Xiang Wang, Tat-Seng Chua 5/11/2026

Rubric-based On-policy Distillation

ROPD framework enables on-policy distillation using semantic rubrics instead of teacher logits for scalable black-box model alignment.

Ax Viktar Dubovik, Patryk Marsza{\l}ek, Jacek Tabor, Tomasz Ku\'smierczyk 5/11/2026

Bayesian Fine-tuning in Projected Subspaces

Extends LoRA with Bayesian fine-tuning in projected subspaces for uncertainty quantification in large model adaptation.

Ax Nils Gr\"unefeld, Bertram H{\o}jer, Philipp Mondorf, Barbara Plank, Anna Rogers, Christian Hardmeier, Stefan Heinrich, Jes Frellsen 5/11/2026

Tracing Uncertainty in Language Model "Reasoning"

Studies uncertainty dynamics in LM reasoning through Chain-of-Thought by analyzing intermediate token sequences as evolving model states.