Description: Moksh Jain. PhD Student in Machine Learning at Mila and UdeM.
Moksh Jain
I am a first year Ph.D. student at Mila and Université de Montréal supervised by Prof. Yoshua Bengio. I am interested in developing machine learning algorithms for effective experimental design, incorporating tools from probabilistic inference and modern deep learning. My work focuses on the probabilistic inference framework of GFlowNets. I am interested in applications of these algorithms to accelerate the scientific discovery process. I lead various efforts for developing novel machine learning approaches
Before joining Mila as a graduate student, I was a visiting researcher with Prof. Yoshua Bengio, working on uncertainty estimation and drug discovery. I also spent a year at Microsoft Turing working on compressing and optimizing large langauge models for deployment across Bing and Office.