Nikhil Rao

Senior ML Scientist


My research interests lie in developing models and efficient algorithms for learning from high dimensional structured data, and their applications in information retrieval, natural language processing and detecting statistically rare events. I'm currently applying these principles at Amazon Search, where I build large scale ML algorithms on graphs. I was previously a Machine Learning Researcher in the Technicolor AI Lab, and prior to that, a recipient of the ICES Postdoctoral Fellowship at the University of Texas at Austin. I received my PhD in Electrical and Computer Engineering from the University of Wisconsin-Madison.


  • ANTHEM: Attentive Hyperbolic Entity Model for Product Search (WSDM 2022), with Nurendra Choudhary, Sumeet Katariya, Karthik Subbian and Chandan Reddy.

  • Online Bayesian Learning for E-Commerce Query Reformulation (NeuRIPS Bayesian Deep Learning workshop 2019), with Gaurush Hiranandani, Sumeet Katariya and Karthik Subbian

  • Short Text Classification using Graph Convolutional Networks (NeuRIPS Graph Workshop 2019), with Kshitij Tayal, Saurabh Agrawal and Karthik Subbian

  • Correlated Gaussian Designs for Compressive Imaging, (IEEE ICIP, 2012) with Robert Nowak


email: <firstname><lastname>86-at-gmail-dot-com

LinkedIn Google Scholar GitHub