Machine Learning Researcher
Current- Led the development of scalable, distributed multi-billion parameter machine learning architectures leveraging PyTorch and advanced techniques such as Fully Sharded Data Parallel (FSDP), torch compile, hardware aware.
- Co-designed novel model-in-the loop methods for molecular design via discrete optimization achieving SOTA on a variety of molecular optimization tasks.
- Designed and implemented experiment and code tracking systems enabling facile experimentation, reproducibility and collaboration with stakeholders