The Mondrian Process for Machine Learning. - Matej Balog; Yee Whye Teh (CoRR, 2015)

The Mondrian Kernel. - Matej Balog; Balaji Lakshminarayanan; Zoubin Ghahramani; Daniel M. Roy 001; Yee Whye Teh (UAI, 2016)

DeepCoder: Learning to Write Programs. - Matej Balog; Alexander L. Gaunt; Marc Brockschmidt; Sebastian Nowozin; Daniel Tarlow (ICLR (Poster), 2017)

Lost Relatives of the Gumbel Trick. - Matej Balog; Nilesh Tripuraneni; Zoubin Ghahramani; Adrian Weller (ICML, 2017)

Differentially Private Database Release via Kernel Mean Embeddings. - Matej Balog; Ilya O. Tolstikhin; Bernhard Schölkopf (ICML, 2018)

Fast Training of Sparse Graph Neural Networks on Dense Hardware. - Matej Balog; Bart van Merriënboer; Subhodeep Moitra; Yujia Li; Daniel Tarlow (CoRR, 2019)

Neural Program Synthesis with a Differentiable Fixer. - Matej Balog; Rishabh Singh; Petros Maniatis; Charles Sutton (CoRR, 2020)

Converting to optimization in machine learning: Perturb-and-MAP, differential privacy, and program synthesis. - Matej Balog (2020)

Discovering faster matrix multiplication algorithms with reinforcement learning. - Alhussein Fawzi; Matej Balog; Aja Huang; Thomas Hubert; Bernardino Romera-Paredes; Mohammadamin Barekatain; Alexander Novikov 001; Francisco J. R. Ruiz; Julian Schrittwieser; Grzegorz Swirszcz; David Silver; Demis Hassabis; et al. (Nat., 2022)

Finding Increasingly Large Extremal Graphs with AlphaZero and Tabu Search. - Abbas Mehrabian; Ankit Anand; Hyunjik Kim; Nicolas Sonnerat; Matej Balog; Gheorghe Comanici; Tudor Berariu; Andrew Lee; Anian Ruoss; Anna Bulanova; Daniel Toyama; Sam Blackwell; et al. (IJCAI, 2024)

Amplifying human performance in combinatorial competitive programming. - Petar Velickovic; Alex Vitvitskyi; Larisa Markeeva; Borja Ibarz; Lars Buesing; Matej Balog; Alexander Novikov 001 (CoRR, 2024)

Mathematical discoveries from program search with large language models. - Bernardino Romera-Paredes; Mohammadamin Barekatain; Alexander Novikov 001; Matej Balog; M. Pawan Kumar; Emilien Dupont; Francisco J. R. Ruiz; Jordan S. Ellenberg; Pengming Wang 001; Omar Fawzi; Pushmeet Kohli; Alhussein Fawzi (Nat., 2024)

Quantum circuit optimization with AlphaTensor. - Francisco J. R. Ruiz; Tuomas Laakkonen; Johannes Bausch; Matej Balog; Mohammadamin Barekatain; Francisco J. H. Heras; Alexander Novikov 001; Nathan Fitzpatrick; Bernardino Romera-Paredes; John van de Wetering; Alhussein Fawzi; Konstantinos Meichanetzidis; et al. (Nat. Mac. Intell., 2025)

AlphaEvolve: A coding agent for scientific and algorithmic discovery. - Alexander Novikov 001; Ngân Vu; Marvin Eisenberger; Emilien Dupont; Po-Sen Huang; Adam Zsolt Wagner; Sergey Shirobokov; Borislav Kozlovskii; Francisco J. R. Ruiz; Abbas Mehrabian; M. Pawan Kumar; Abigail See; et al. (CoRR, 2025)