Yash Mehta

PhD student, Johns Hopkins University.



I am currently pursuing a PhD in Computational Cognitive Science at Johns Hopkins University and am a visiting PhD student at Harvard Medical School. My research aims to bridge cognitive science and artificial intelligence to improve AI applications in healthcare.

Throughout my academic and professional journey, I’ve had the privilege of collaborating with many talented scientists and engaging in diverse research projects. My areas of interest include foundation models, cognition, and computational neuroscience. Below is the list of labs and mentors I have worked with. Prior to my research career, I gained industry experience as a developer at Amazon.

I have a strong commitment to mentoring and spend time engaging with students who are passionate about research. If you are interested in exploring research opportunities or seeking advice, please don’t hesitate to reach out to me directly.


Oct 15, 2023 Update. Started at Harvard Medical School with Pranav Rajpurkar to work on knowledge graphs for automated radiology report generation 🏥
Sep 1, 2023 Talk. “Multimodal Foundation Models @ICML’23” at SenticLab in NTU Singapore!
Aug 19, 2023 Visit. Research visit to the Flatiron Institute and PolymathicAI team - multimodal foundation models for science!
Jul 20, 2023 Conference. Attended ICML 2023 in Hawaii! 🏄🏻‍♂️🌸
Mar 23, 2023 General. My application for US immigration has been approved under EB1A category - individual of outstanding ability! 🇺🇸
Dec 1, 2022 Conference. Co-presented work on node perturbation at NeurIPS 2022 in New Orleans! ✨🎷
Oct 1, 2022 Visit. 1 month as a student researcher in Larry Abbott’s lab at the Zuckerman Institute, Columbia University! 🗽
May 27, 2022 Got Married! :ring: :sparkles: :sunny:
May 1, 2022 Editorial. Our special issue “Future-generation personality prediction from digital footprints” was accepted at FGCS international journal! Managing guest editor 🤝
Apr 25, 2022 Conference. Presented our NAS-BenchSuite paper on neural architecture search at ICLR’22.
Jan 10, 2022 Update. Started at HHMI Janelia Research Campus working on modeling synaptic plasticity with James Fitzgerald and Jan Funke 🪰
Nov 1, 2021 Teaching. Co-created attention and transformers module with Prof. Frank Hutter. Also served as TA for Deep Learning Lab, and Deep Learning MSc. courses.
Oct 15, 2020 Update. Joined the AutoML group with Frank Hutter at University of Freiburg, Germany, to work on neural architecture search.
Jan 10, 2019 Update. Joined Gatsby Unit, UCL in London as a Research Assistant working on biologically plausiblle credit assignment algorithms with Peter Latham and Tim Lillicrap (DeepMind)! ⚡️🙌🏼
Jul 15, 2018 Update. Started working at Amazon as a software development engineer in Bangalore.
Jul 1, 2018 Update. Graduated in Computer Science from BITS Pilani Goa! 🎓
Jan 15, 2018 Update. Started undergraduate research thesis with Erik Cambria on using large language models for personality trait prediction.

Selected Publications

  1. Towards Biologically Plausible Convolutional Networks
    Roman Pogodin, Yash MehtaTimothy Lillicrap, and Peter Latham
    In Advances in Neural Information Processing Systems (NeurIPS) 2021
  2. NAS-Bench-Suite: NAS Evaluation is (Now) Surprisingly Easy
    Yash Mehta, Colin White, Arber Zela, Arjun Krishnakumar, and 5 more authors
    In International Conference on Learning Representations (ICLR) 2022
  3. Stability and Scalability of Node Perturbation Learning
    Naoki Hiratani, Yash MehtaTimothy Lillicrap, and Peter Latham
    In Advances in Neural Information Processing Systems (NeurIPS) 2022