Generalised Lookahead Optimiser. - Costin-Andrei Oncescu; Jack Valmadre; João F. Henriques (Tiny Papers @ ICLR, 2023)

Feature emergence via margin maximization: case studies in algebraic tasks. - Depen Morwani; Benjamin L. Edelman; Costin-Andrei Oncescu; Rosie Zhao; Sham M. Kakade (ICLR, 2024)

Flash Inference: Near Linear Time Inference for Long Convolution Sequence Models and Beyond. - Costin-Andrei Oncescu; Sanket Purandare; Stratos Idreos; Sham M. Kakade (ICLR, 2025)

A Simplified Analysis of SGD for Linear Regression with Weight Averaging. - Alexandru Meterez; Depen Morwani; Costin-Andrei Oncescu; Jingfeng Wu; Cengiz Pehlevan; Sham M. Kakade (CoRR, 2025)