Samyak Jain

Interested in AI Safety (Looking for PhD positions)

Hi! I am Samyak. I am currently working at Five AI at Oxford, where I am advised by Puneet Dokania. Previously, I have worked with David Krueger at Cambridge University and Venkatesh Babu at Indian Institute of Science, Bangalore. Recently, I completed my bachelors and masters in Computer Science at the Indian Institute of Technology (BHU) Varansi, where my thesis focussed on understanding and improving adversarial robustness of neural networks. For the last few years I have been broadly working on Machine Learning with a particular focus on Adversarial Robustness, Generalization and Mechanistic Interpretability. Moving forward, I am excited about contributing in the field of AI Safety.

Recently, I completed my interenship at Cambridge University, where I worked on mechanistic interpretability. Before that I worked at the Vision and AI Lab where my projects were related to adversarial robustness and understanding loss landscape of neural networks, which have been published at the following venues.

Besides this I actively participate in machine learning and computer vision conferences and have served as a reviewer at multiple conferences and won outstanding reviewer awards at ICLR 2022 and CVPR 2022.

Here is the link to my CV.

Broadly I am interested in research topics related to AI safety, in particular:
Alignment: Unlearning, RLHF fine-tuning
Mechanistic Interpretability: Science of Deep Learning
Robustness: Adversarial Robustness, OOD Robustness
Generalization and Foundations of ML: Mode Connectivity, Loss landscape sharpness, Learning dynamics
Role of Data in Learning: Curating data to build safe models