Design Principles of the Hippocampal Cognitive Map. - Kimberly L. Stachenfeld; Matthew M. Botvinick; Samuel Gershman (NIPS, 2014)

Spectral Inference Networks: Unifying Spectral Methods With Deep Learning. - David Pfau; Stig Petersen; Ashish Agarwal; David G. T. Barrett; Kimberly L. Stachenfeld (CoRR, 2018)

A probabilistic approach to discovering dynamic full-brain functional connectivity patterns. - Jeremy R. Manning; Xia Zhu; Theodore L. Willke; Rajesh Ranganath; Kimberly L. Stachenfeld; Uri Hasson; David M. Blei; Kenneth A. Norman (NeuroImage, 2018)

Spectral Inference Networks: Unifying Deep and Spectral Learning. - David Pfau; Stig Petersen; Ashish Agarwal; David G. T. Barrett; Kimberly L. Stachenfeld (ICLR (Poster), 2019)

Structured agents for physical construction. - Victor Bapst; Alvaro Sanchez-Gonzalez; Carl Doersch; Kimberly L. Stachenfeld; Pushmeet Kohli; Peter W. Battaglia; Jessica B. Hamrick (ICML, 2019)

Probabilistic Successor Representations with Kalman Temporal Differences. - Jesse P. Geerts; Kimberly L. Stachenfeld; Neil Burgess (CoRR, 2019)

Object-oriented state editing for HRL. - Victor Bapst; Alvaro Sanchez-Gonzalez; Omar Shams; Kimberly L. Stachenfeld; Peter W. Battaglia; Satinder Singh 001; Jessica B. Hamrick (CoRR, 2019)

Graph Networks with Spectral Message Passing. - Kimberly L. Stachenfeld; Jonathan Godwin; Peter W. Battaglia (CoRR, 2021)

Learned Coarse Models for Efficient Turbulence Simulation. - Kimberly L. Stachenfeld; Drummond B. Fielding; Dmitrii Kochkov; Miles D. Cranmer; Tobias Pfaff; Jonathan Godwin; Can Cui; Shirley Ho; Peter W. Battaglia; Alvaro Sanchez-Gonzalez (CoRR, 2021)

Graph network simulators can learn discontinuous, rigid contact dynamics. - Kelsey R. Allen; Tatiana Lopez-Guevara; Yulia Rubanova; Kimberly L. Stachenfeld; Alvaro Sanchez-Gonzalez; Peter W. Battaglia; Tobias Pfaff (CoRL, 2022)

Learned Simulators for Turbulence. - Kimberly L. Stachenfeld; Drummond Buschman Fielding; Dmitrii Kochkov; Miles D. Cranmer; Tobias Pfaff; Jonathan Godwin; Can Cui; Shirley Ho; Peter W. Battaglia; Alvaro Sanchez-Gonzalez (ICLR, 2022)

Inverse Design for Fluid-Structure Interactions using Graph Network Simulators. - Kelsey R. Allen; Tatiana Lopez-Guevara; Kimberly L. Stachenfeld; Alvaro Sanchez-Gonzalez; Peter W. Battaglia; Jessica B. Hamrick; Tobias Pfaff (NeurIPS, 2022)

Physical Design using Differentiable Learned Simulators. - Kelsey R. Allen; Tatiana Lopez-Guevara; Kimberly L. Stachenfeld; Alvaro Sanchez-Gonzalez; Peter W. Battaglia; Jessica B. Hamrick; Tobias Pfaff (CoRR, 2022)

Compositional Sequence Generation in the Entorhinal-Hippocampal System. - Daniel C. McNamee; Kimberly L. Stachenfeld; Matthew M. Botvinick; Samuel J. Gershman (Entropy, 2022)

A generative model of the hippocampal formation trained with theta driven local learning rules. - Tom M. George; Kimberly L. Stachenfeld; Caswell Barry; Claudia Clopath; Tomoki Fukai (NeurIPS, 2023)

Diffusion Generative Inverse Design. - Marin Vlastelica; Tatiana Lopez-Guevara; Kelsey R. Allen; Peter W. Battaglia; Arnaud Doucet; Kimberly L. Stachenfeld (CoRR, 2023)

Predictive auxiliary objectives in deep RL mimic learning in the brain. - Ching Fang; Kim Stachenfeld (ICLR, 2024)

Learning 3D Particle-based Simulators from RGB-D Videos. - William F. Whitney; Tatiana Lopez-Guevara; Tobias Pfaff; Yulia Rubanova; Thomas Kipf; Kim Stachenfeld; Kelsey R. Allen (ICLR, 2024)

Learning rigid-body simulators over implicit shapes for large-scale scenes and vision. - Yulia Rubanova; Tatiana Lopez-Guevara; Kelsey R. Allen; Will Whitney; Kimberly L. Stachenfeld; Tobias Pfaff (NeurIPS, 2024)

How does the primate brain combine generative and discriminative computations in vision? - Benjamin Peters; James J. DiCarlo; Todd M. Gureckis; Ralf Haefner; Leyla Isik; Joshua B. Tenenbaum; Talia Konkle; Thomas Naselaris; Kimberly L. Stachenfeld; Zenna Tavares; Doris Tsao; Ilker Yildirim; et al. (CoRR, 2024)

Scaling Face Interaction Graph Networks to Real World Scenes. - Tatiana Lopez-Guevara; Yulia Rubanova; William F. Whitney; Tobias Pfaff; Kimberly L. Stachenfeld; Kelsey R. Allen (CoRR, 2024)

DeepSpeech models show Human-like Performance and Processing of Cochlear Implant Inputs. - Cynthia R. Steinhardt; Menoua Keshishian; Nima Mesgarani; Kim Stachenfeld (CoRR, 2024)

SIMPL: Scalable and hassle-free optimisation of neural representations from behaviour. - Tom M. George; Pierre Glaser; Kim Stachenfeld; Caswell Barry; Claudia Clopath (ICLR, 2025)

When does compositional structure yield compositional generalization? A kernel theory. - Samuel Lippl; Kim Stachenfeld (ICLR, 2025)

ReCogLab: a framework testing relational reasoning & cognitive hypotheses on LLMs. - Andrew Liu; Henry Prior; Gargi Balasubramaniam; Rivka Moroshko; Amir Zait; Ilia Labzovsky; Danny Karmon; Ishita Dasgupta 001; Kim Stachenfeld; Kenneth Marino (ICLR, 2025)

Relational reasoning and inductive bias in transformers trained on a transitive inference task. - Jesse P. Geerts; Stephanie Chan; Claudia Clopath; Kimberly L. Stachenfeld (CoRR, 2025)