Hi! I’m Rohan. I recently graduated with a BS in Computer Science from UCLA, and work as a software engineer. I’m originally from San Jose, California.
Work
I currently work at Facebook AI on enabling large-scale distributed training within the PyTorch framework. I’ve recently been working on helping build out model parallelism support within PyTorch, as well as investigating the scalability of data parallel training.
Previously, I’ve worked on building mortgage lending products at Blend, risk models at Affirm, and messaging applications for sports teams at Hudl.
Writing
This blog is mostly a place for me to collect random notes and tips that I’ve found interesting and useful over the years. Below is a reverse-chronological set of links to some essays that I’ve written over the years!
2020
How to use PyTorch’s RPC Framework
2019
2018
My Internship at Affirm: Crafting a Reliable Metrics and Alerting Framework
2017
Getting Started in Machine Learning
Misc
This section is still in progress!