Ziheng Jiang
- Learning to Optimize Tensor Programs
2018/05/21 by Tianqi Chen, Lianmin Zheng, Chen, Tianqi +14 · 1 voice · 25 citations
Computer Science · Physics and Astronomy · #Parallel Computing and Optimization Techniques #Computational Physics and Python Applications #Model Reduction and Neural Networks
- TVM: An Automated End-to-End Optimizing Compiler for Deep Learning
2018/02/12 by Tianqi Chen, Thierry Moreau, Chen, Tianqi +22 · 1 voice · 12 citations
Computer Science · #Parallel Computing and Optimization Techniques #Advanced Neural Network Applications #Advanced Data Storage Technologies
- MegaScale: Scaling Large Language Model Training to More Than 10,000 GPUs
2024/02/23 by Ziheng Jiang, Haibin Lin, Jiang, Ziheng +61 · 30 citations
Computer Science · #Topic Modeling #Natural Language Processing Techniques
- EfficientPhys: Enabling Simple, Fast and Accurate Camera-Based Vitals Measurement
2021/10/09 by Xin Liu, Liu, Xin, Brian L. Hill +7 · 8 citations
Engineering · Medicine · #Artificial Intelligence (cs.AI) #Computer Vision and Pattern Recognition (cs.CV) #ECG Monitoring and Analysis #FOS: Computer and information sciences #Human-Computer Interaction (cs.HC) #Non-Invasive Vital Sign Monitoring #Optical Imaging and Spectroscopy Techniques
- FLUX: Fast Software-based Communication Overlap On GPUs Through Kernel Fusion
2024/06/11 by Li‐Wen Chang, Chang, Li-Wen, Wenlei Bao +22 · 13 citations
Computer Science · Neuroscience · #Parallel Computing and Optimization Techniques #Embedded Systems Design Techniques #Brain Tumor Detection and Classification
- Comet: Fine-grained Computation-communication Overlapping for Mixture-of-Experts
2025/02/27 by S. J. Zhang, Zhang, Shulai, Ningxin Zheng +20 · 11 citations
Computer Science · #Advanced Neural Network Applications #Mobile Crowdsensing and Crowdsourcing #Big Data and Digital Economy
- MegaScale-Infer: Serving Mixture-of-Experts at Scale with Disaggregated Expert Parallelism
2025/04/03 by Ruidong Zhu, Zhu, Ruidong, Ziheng Jiang +35 · 14 citations
Computer Science · #Advanced Neural Network Applications #Big Data and Digital Economy #Domain Adaptation and Few-Shot Learning
- Triton-distributed: Programming Overlapping Kernels on Distributed AI Systems with the Triton Compiler
2025/04/28 by Size Zheng, Wenlei Bao, Zheng, Size +40 · 9 citations
Computer Science · #Parallel Computing and Optimization Techniques #Advanced Neural Network Applications #Distributed and Parallel Computing Systems
- A Hardware-Software Blueprint for Flexible Deep Learning Specialization
2018/07/11 by Thierry Moreau, Tianqi Chen, Moreau, Thierry +19 · 1 voice
Computer Science · Mathematics · #Distributed #FOS: Computer and information sciences #Machine Learning (cs.LG) #Machine Learning (stat.ML) #Parallel #and Cluster Computing (cs.DC) #cs.DC #cs.LG #stat.ML
- Federated Remote Physiological Measurement with Imperfect Data
2022/03/11 by Xin Liu, Liu, Xin, Mingchuan Zhang +7 · 1 citation
Computer Science · Environmental Science · #Privacy-Preserving Technologies in Data #Air Quality Monitoring and Forecasting #IoT and Edge/Fog Computing