Frequency-aware SGD for Efficient Embedding Learning with Provable Benefits
Yan Li, Dhruv Choudhary, Xiaohan Wei, Baichuan Yuan, Bhargav Bhushanam, Tuo Zhao, Guanghui Lan
People
Dhruv Choudhary
Machine Learning Engineer
I work on large scale Machine Learning efficiency at Facebook. My research interests include inference and training efficiency for real time ML systems. At Facebook, I work on improving the efficiency of the Recommendations stack.
Interests
Recommender systems, artificial intelligence & systems
Latest Publications
ICLR - 2022
Training Recommender Systems at Scale: Communication-Efficient Model and Data Parallelism
Vipul Gupta, Dhruv Choudhary, Peter Tang, Xiaohan Wei, Xing Wang, Yuzhen Huang, Arun Kejariwal, Kannan Ramchandran, Michael W. Mahoney
Hierarchical Training: Scaling Deep Recommendation Models on Large CPU Clusters
Yuzhen Huang, Xiaohan Wei, Xing Wang, Jiyan Yang, Bor-Yiing Su, Shivam Bharuka, Dhruv Choudhary, Zewei Jiang, Hai Zheng, Jack Langman