Gustaf Ahdritz

I'm a second-year PhD student in Computer Science at Harvard University. I'm a member of the Machine Learning Foundations Group and am advised by Jonathan Frankle and Boaz Barak. I'm supported by a fellowship from Harvard's Kempner Institute.

I'm broadly interested in empirical investigations of the properties of realistic deep neural networks. At the moment, I'm thinking about uncertainty in large language models.

I graduated from Columbia with a B.A. in Computer Science & History (2020) and an M.S. in Computer Science (2021). There, I work with Mohammed AlQuraishi on the applied task of protein structure prediction and lead the development of OpenFold. I also spent time in Kathleen McKeown's lab and the History Lab.

Here's my CV.


* denotes equal contribution


  • OpenProteinSet: Training data for structural biology at scale
    Gustaf Ahdritz, Nazim Bouatta, Sachin Kadyan, Lukas Jarosch, Daniel Berenberg, Ian Fisk, Andrew M. Watkins, Stephen Ra, Richard Bonneau, Mohammed AlQuraishi
    arXiv, 2023.
    [paper] [data] [bibtex]
  • OpenFold: Retraining AlphaFold2 yields new insights into its learning mechanisms and capacity for generalization
    Gustaf Ahdritz, Nazim Bouatta, Christina Floristean, Sachin Kadyan, Qinghui Xia, William Gerecke, Timothy J. O'Donnell, Daniel Berenberg, Ian Fisk, Niccolò Zanichelli, Bo Zhang, Arkadiusz Nowaczynski, Bei Wang, Marta M. Stepniewska-Dziubinska, Shang Zhang, Adegoke Ojewole, Murat Efe Guney, Stella Biderman, Andrew M. Watkins, Stephen Ra, Pablo Ribalta Lorenzo, Lucas Nivon, Brian Weitzner, Yih-En Andrew Ban, Peter K. Sorger, Emad Mostaque, Zhao Zhang, Richard Bonneau, Mohammed AlQuraishi
    bioRxiv, 2022.
    [paper] [code] [talk] [bibtex]

Workshop papers

  • Soft prompting might be a bug, not a feature
    Luke Bailey, Gustaf Ahdritz, Anat Kleiman, Siddharth Swaroop, Finale Doshi-Velez, Weiwei Pan
    Workshop on Challenges in Deployable Generative AI, ICML 2023.
    [paper] [bibtex]


  • Single-sequence protein structure prediction using a language model and deep learning
    Ratul Chowdhury, Nazim Bouatta, Surojit Biswas, Christina Floristean, Anant Kharkar, Koushik Roy, Charlotte Rochereau, Gustaf Ahdritz, Joanna Zhang, George M. Church, Peter K. Sorger, Mohammed AlQuraishi
    Nature Biotechnology, 2022.
    [paper] [code] [coverage] [bibtex]


I've served as a teaching fellow/assistant for the following courses at Harvard and Columbia:
  • Spring 2023: Foundations of Deep Learning (Harvard COMPSCI 229br) with Boaz Barak
  • Spring 2019 - Spring 2021: Advanced Programming (Columbia COMS 3157) with Jae Woo Lee

Awards & Fellowships


[github] [google scholar] [twitter]