Ax Will Schwarzer, Jordan Schneider, Philip S. Thomas, Scott Niekum 7/8/2026

Supervised Reward Inference

Supervised reward inference approach handling diverse human behaviors for goal inference.

Ax Lorenzo Steccanella, Joshua B. Evans, \"Ozg\"ur \c{S}im\c{s}ek, Anders Jonsson 7/8/2026

Learning The Minimum Action Distance

Learning minimum action distance as state representation metric from trajectories without rewards or actions.

Ax Gang Liao, Hongsen Qin, Ying Wang, Alicia Golden, Michael Kuchnik, Yavuz Yetim, Jia Jiunn Ang, Chunli Fu, Yihan He, Samuel Hsia, Zewei Jiang, Dianshi Li, Uladzimir Pashkevich, Varna Puvvada, Feng Shi, Matt Steiner, Ruichao Xiao, Liyuan Li, Nathan Yan, Xiayu Yu, Zhou Fang, Roman Levenstein, Kunming Ho, Haishan Zhu, Alec Hammond, Richard Li, Ajit Mathews, Kaustubh Gondkar, Abdul Zainul-Abedin, Ketan Singh, Hongtao Yu, Wenyuan Chi, Barney Huang, Sean Zhang, Noah Weller, Zach Marine, Wyatt Cook, Carole-Jean Wu, Gaoxiang Liu 7/8/2026

KernelEvolve: Scaling Agentic Kernel Coding for Heterogeneous AI Accelerators at Meta

Agentic framework for automated kernel code generation across heterogeneous AI accelerators at scale for recommendation models.

Ax Ramnath Kumar, Kyle Ritscher, Junmin Judy, Lawrence Liu, Cho-Jui Hsieh 7/8/2026

FlexAct: Why Learn when you can Pick?

Research on learning discrete activation functions using Gumbel-Softmax for task-specific neural networks.

Ax Gloria Felicia (University of Virginia), Zitha Sasindran (Indian Institute of Science Bangalore), Jinfeng He (Cornell University), Michael Eniolade (University of the Cumberlands), Hemant Kumar (University of Arizona), Milan Hussain Angati (California State University Northridge) 7/8/2026

StepShield: When, Not Whether to Intervene on Rogue Agents

StepShield benchmark for agent safety measuring detection timeliness of rogue agent behavior on 9,429 code-agent trajectories.

Ax Stefano Woerner, Seong Joon Oh, Christian F. Baumgartner 7/8/2026

Universal Algorithm-Implicit Learning

Meta-learning framework defining practical universality and generalizable learning across arbitrary task distributions.

Ax Huaiyang Wang, Xiaojie Li, Xiaohan Wang, Zhixia Zhang, Xiaodong Lu, Zixuan Huang, Jiajun Chai, Guojun Yin, Deqing Wang, Haoyi Zhou, Yaodong Yang, Jianxin Li, Yikun Ban 7/8/2026

Policy Improvement Reinforcement Learning

Policy Improvement RL method for post-training LLMs and agents that explicitly verifies policy improvements over baselines.

Ax Kathan Shah 7/8/2026

Token Geometry

Lightweight optimizer exploiting gradient geometry of embedding tables and LM-heads for improved training efficiency across finetuning and pretraining.

Ax Joachim Tomasi, Sandrine Anthoine, Hachem Kadri 7/8/2026

Benign Overfitting with Quantum Kernels

Theoretical study of benign overfitting in quantum kernel methods for machine learning on quantum computers.