Swarnim Jain

researcher and engineercomputer science @ University of Cambridgepreviously, math @ IIT Bombay

about

i'm excited by the intersection of ml research and engineering across domains. these days, i'm exploring the transformer architecture along with how we can optimize training and inference. i'm also running CUAI .

if you want to talk about any cool projects or opportunities, feel free to reach out!

experience

researcherPietro Liò Group, University of Cambridge

working on vision language models (VLMs), reinforcement learning, and robotics.
previously worked on graph neural networks and diffusion-based graph generation.

researcherVisual Geometry Group, University of Oxford

working on an efficient ResNet architecture and video point-tracking.

machine learning engineerMyrtle.ai

worked on LLM post-training, evals, and RAG.

connect