His current research interests include distributed systems for tensors with sparsity. He has developed FastMoE, the world's first open-source distributed training framework for Mixture-of-Experts models based on PyTorch. He has a few papers published on ASPLOS, PPoPP, and other conferences.