PhD student @ AIMM, EPFL under Prof. Charlotte Bunne
Bachelor's + Master's @ IIT Bombay, Institute Silver Medalist (Rank #2)
Exchange student @ EPFL, Switzerland in Computer Science for Autumn 2022-23
[ Eeshaan Jain ] [ eeshaan-jain ] [ eeshaan_jain ] [ eeshaanjain ]

I have a keen interest in applications of machine learning to life sciences, especially multi-scale representation learning and drug discovery along with geometric deep learning. Previously, I have worked on optimization, graph neural networks, graph retrieval, and fair learning under Prof. Abir De and Prof. Soumen Charkabarti at IIT Bombay. I have also worked with Google Research for a year, and with AWL, Inc. and Sony, Japan.

I have acted as a Teaching Assistant in the following courses:

1. CS 461: Foundation models and generative AI (250+ students) [EPFL]
2. CS 119: Information, calculation, communication (300+ students) [EPFL]
3. CS 419M (Introduction to Machine Learning) (100+ students) [IIT Bombay]
4. CS 768 (Learning with Graphs) (50+ students) [IIT Bombay]
5. CS 769 (Optimization for Machine Learning: 100+ students) [IIT Bombay]
6. MA 207 (Partial Differential Equations: 200+ students) [IIT Bombay]
7. MA 108 (Ordinary Differential Equations: 400+ students) [IIT Bombay]
8. CH 107 (Quantum Chemistry: 400+ students) [IIT Bombay]

I have received scholarships and grants from:

• Broad School of MIT and Harvard for ICLR 2025
• NeurIPS Travel Grant for NeurIPS 2023
• Google Research (Google Conference Scholarship)
• IIT Bombay (Institute Academic Prize)
• Govt. of India (Institute Silver Medal)
• Heyning-Roelli Foundation (Semeseter Exchange Scholarship)