A selection framework for LHCb's upgrade trigger - Niklas Nolte (2020)

Towards Understanding Grokking: An Effective Theory of Representation Learning. - Ziming Liu; Ouail Kitouni; Niklas Nolte; Eric J. Michaud; Max Tegmark; Mike Williams (NeurIPS, 2022)

Finding NEEMo: Geometric Fitting using Neural Estimation of the Energy Mover's Distance. - Ouail Kitouni; Niklas Nolte; Mike Williams (CoRR, 2022)

Expressive Monotonic Neural Networks. - Niklas Nolte; Ouail Kitouni; Mike Williams (ICLR, 2023)

NuCLR: Nuclear Co-Learned Representations. - Ouail Kitouni; Niklas Nolte; Sokratis Trifinopoulos; Subhash Kantamneni; Mike Williams (CoRR, 2023)

Expressive Monotonic Neural Networks. - Ouail Kitouni; Niklas Nolte; Michael Williams (CoRR, 2023)

KBFormer: A Diffusion Model for Structured Entity Completion. - Ouail Kitouni; Niklas Nolte; James Hensman; Bhaskar Mitra 001 (CoRR, 2023)

Robust and provably monotonic networks. - Ouail Kitouni; Niklas Nolte; Mike Williams (Mach. Learn. Sci. Technol., 2023)

From Neurons to Neutrons: A Case Study in Interpretability. - Ouail Kitouni; Niklas Nolte; Víctor Samuel Pérez-Díaz; Sokratis Trifinopoulos; Mike Williams (ICML, 2024)

The Factorization Curse: Which Tokens You Predict Underlie the Reversal Curse and More. - Ouail Kitouni; Niklas Nolte; Adina Williams; Michael Rabbat; Diane Bouchacourt; Mark Ibrahim (NeurIPS, 2024)

Transformers Can Navigate Mazes With Multi-Step Prediction. - Niklas Nolte; Ouail Kitouni; Adina Williams; Mike Rabbat; Mark Ibrahim (CoRR, 2024)

The cool and the cruel: separating hard parts of LWE secrets. - Niklas Nolte; Mohamed Malhou; Emily Wenger; Samuel Stevens 001; Cathy Yuanchen Li; François Charton; Kristin E. Lauter (IACR Cryptol. ePrint Arch., 2024)

SALSA FRESCA: Angular Embeddings and Pre-Training for ML Attacks on Learning With Errors. - Samuel Stevens 001; Emily Wenger; Cathy Yuanchen Li; Niklas Nolte; Eshika Saxena; François Charton; Kristin E. Lauter (IACR Cryptol. ePrint Arch., 2024)

Transforming the bootstrap: using transformers to compute scattering amplitudes in planar N = 4 super Yang-Mills theory. - Tianji Cai; Garrett W. Merz; François Charton; Niklas Nolte; Matthias Wilhelm 003; Kyle Cranmer; Lance J. Dixon (Mach. Learn. Sci. Technol., 2024)

MagicPIG: LSH Sampling for Efficient LLM Generation. - Zhuoming Chen; Ranajoy Sadhukhan; Zihao Ye; Yang Zhou; Jianyu Zhang; Niklas Nolte; Yuandong Tian; Matthijs Douze; Léon Bottou; Zhihao Jia; Beidi Chen (ICLR, 2025)

Memory Mosaics. - Jianyu Zhang; Niklas Nolte; Ranajoy Sadhukhan; Beidi Chen; Léon Bottou (ICLR, 2025)

Learning Distributions over Permutations and Rankings with Factorized Representations. - Daniel Severo 001; Brian Karrer; Niklas Nolte (CoRR, 2025)

Accelerated Sampling from Masked Diffusion Models via Entropy Bounded Unmasking. - Heli Ben-Hamu; Itai Gat; Daniel Severo 001; Niklas Nolte; Brian Karrer (CoRR, 2025)