Daniel Koceja

koceja [at] stanford.edu

I am a first-year PhD student at Stanford University. I am interested in test-time training and models that learn continuously without a fixed boundary between training and inference.

Selected Papers

Learning to Discover at Test Time

Mert Yuksekgonul*, Daniel Koceja*, Xinhao Li*, Federico Bianchi*, Jed McCaleb, Xiaolong Wang, Jan Kautz, Yejin Choi, James Zou†, Carlos Guestrin†, Yu Sun*

ICML 2026 (Spotlight)

paper · code

End-to-End Test-Time Training for Long Context

Arnuv Tandon*, Karan Dalal*, Xinhao Li*, Daniel Koceja*, Marcel Rød*, Sam Buchanan, Xiaolong Wang, Jure Leskovec, Sanmi Koyejo, Tatsunori Hashimoto, Carlos Guestrin, Jed McCaleb, Yejin Choi, Yu Sun*

paper · code

* core contributors

Last updated: September 2026