Wei Cui

Principal Researcher

Publications

View by:

- MetroRLHF: Enabling Memory-Effective Training for On-Policy RLHF via Adaptive Sequence Streaming
  
  Wei Cui, Peng Cheng
  
  NeurIPS 2025 The First Workshop of Efficient Reasoning | December 2025
  
  Publication
- FlashFFTStencil: Bridging Fast Fourier Transforms to Memory-Efficient Stencil Computations on Tensor Core Units
  
  Haozhi Han, Kun Li, Wei Cui, Donglin Bai, Yiwei Zhang, Liang Yuan, Yifeng Chen, Yunquan Zhang, Ting Cao, Mao Yang
  
  ACM SIGPLAN Symposium on Principles and Practice of Parallel Programming (PPoPP) | March 2025
  
  DOI
- Empowering In-Browser Deep Learning Inference on Edge Devices with Just-in-Time Kernel Optimizations
  
  Fucheng Jia, Shiqi Jiang, Ting Cao, Wei Cui, Tianrui Xia, Xu Cao, Yuanchun Li, Qipeng Wang, Deyu Zhang, Ju Ren, Yunxin Liu, Lili Qiu, Mao Yang
  
  the 22nd Annual International Conference on Mobile Systems, Applications and Services (MobiSys'24) | June 2024
  
  Publication
- Query Processing on Gaming Consoles
  
  Wei Cui, Qianxi Zhang, Jesús Camacho-Rodríguez, Spyros Blanas, Brandon Haynes, Yinan Li, Peng Cheng, Ravishankar Ramamurthy, Rathijit Sen, Matteo Interlandi
  
  June 2023
  
  DOI Publication
- Roller: Fast and Efficient Tensor Compilation for Deep Learning
  
  Hongyu Zhu , Ruofan Wu , Yijia Diao , Shanbin Ke , Haoyu Li , Chen Zhang , Jilong Xue, Lingxiao Ma, Yuqing Xia, Wei Cui, Fan Yang, Mao Yang, Lidong Zhou, Asaf Cidon , Gennady Pekhimenko
  
  The 16th USENIX Symposium on Operating Systems Design and Implementation (OSDI'22) | July 2022
  
  Publication Project
- Tutel: Adaptive Mixture-of-Experts at Scale
  
  Changho Hwang, Wei Cui, Yifan Xiong, Ziyue Yang, Ze Liu, Han Hu, Zilong Wang, Rafael Salas, Jithin Jose, Prabhat Ram, Joe Chau, Peng Cheng, Fan Yang, Mao Yang, Yongqiang Xiong
  
  July 2022
  
  Publication
- OpEvo: An Evolutionary Method for Tensor Operator Optimization
  
  Xiaotian Gao, Wei Cui, Lintao Zhang, Mao Yang
  
  AAAI 2021 | May 2021
  
  Publication
- Adaptive Page Migration Policy With Huge Pages in Tiered Memory Systems
  
  Taekyung Heo, Yang Wang, Wei Cui, Jaehyuk Huh, Lintao Zhang
  
  IEEE Transactions on Computers | November 2020, Vol 71(1): pp. 53-68
  
  DOI Publication
- Rammer: Enabling Holistic Deep Learning Compiler Optimizations with rTasks
  
  Lingxiao Ma, Zhiqiang Xie, Zhi Yang, Jilong Xue, Youshan Miao, Wei Cui, Wenxiang Hu, Fan Yang, Lintao Zhang, Lidong Zhou
  
  14th USENIX Symposium on Operating Systems Design and Implementation (OSDI'20) | November 2020
  
  Publication Project

- MetroRLHF: Enabling Memory-Effective Training for On-Policy RLHF via Adaptive Sequence Streaming
  
  Wei Cui, Peng Cheng
  
  NeurIPS 2025 The First Workshop of Efficient Reasoning | December 2025
  
  Publication
- MetroRLHF: Enabling Memory-Effective Training for On-Policy RLHF via Adaptive Sequence Streaming
  
  Wei Cui, Peng Cheng
  
  NeurIPS 2025 The First Workshop of Efficient Reasoning | December 2025
  
  Publication
- FlashFFTStencil: Bridging Fast Fourier Transforms to Memory-Efficient Stencil Computations on Tensor Core Units
  
  Haozhi Han, Kun Li, Wei Cui, Donglin Bai, Yiwei Zhang, Liang Yuan, Yifeng Chen, Yunquan Zhang, Ting Cao, Mao Yang
  
  ACM SIGPLAN Symposium on Principles and Practice of Parallel Programming (PPoPP) | March 2025
  
  DOI
- Empowering In-Browser Deep Learning Inference on Edge Devices with Just-in-Time Kernel Optimizations
  
  Fucheng Jia, Shiqi Jiang, Ting Cao, Wei Cui, Tianrui Xia, Xu Cao, Yuanchun Li, Qipeng Wang, Deyu Zhang, Ju Ren, Yunxin Liu, Lili Qiu, Mao Yang
  
  the 22nd Annual International Conference on Mobile Systems, Applications and Services (MobiSys'24) | June 2024
  
  Publication
- Tutel: Adaptive Mixture-of-Experts at Scale
  
  Changho Hwang, Wei Cui, Yifan Xiong, Ziyue Yang, Ze Liu, Han Hu, Zilong Wang, Rafael Salas, Jithin Jose, Prabhat Ram, Joe Chau, Peng Cheng, Fan Yang, Mao Yang, Yongqiang Xiong
  
  July 2022
  
  Publication
- Roller: Fast and Efficient Tensor Compilation for Deep Learning
  
  Hongyu Zhu , Ruofan Wu , Yijia Diao , Shanbin Ke , Haoyu Li , Chen Zhang , Jilong Xue, Lingxiao Ma, Yuqing Xia, Wei Cui, Fan Yang, Mao Yang, Lidong Zhou, Asaf Cidon , Gennady Pekhimenko
  
  The 16th USENIX Symposium on Operating Systems Design and Implementation (OSDI'22) | July 2022
  
  Publication Project
- OpEvo: An Evolutionary Method for Tensor Operator Optimization
  
  Xiaotian Gao, Wei Cui, Lintao Zhang, Mao Yang
  
  AAAI 2021 | May 2021
  
  Publication
- Adaptive Page Migration Policy With Huge Pages in Tiered Memory Systems
  
  Taekyung Heo, Yang Wang, Wei Cui, Jaehyuk Huh, Lintao Zhang
  
  IEEE Transactions on Computers | November 2020, Vol 71(1): pp. 53-68
  
  DOI Publication
- Rammer: Enabling Holistic Deep Learning Compiler Optimizations with rTasks
  
  Lingxiao Ma, Zhiqiang Xie, Zhi Yang, Jilong Xue, Youshan Miao, Wei Cui, Wenxiang Hu, Fan Yang, Lintao Zhang, Lidong Zhou
  
  14th USENIX Symposium on Operating Systems Design and Implementation (OSDI'20) | November 2020
  
  Publication Project
- FlashFFTStencil: Bridging Fast Fourier Transforms to Memory-Efficient Stencil Computations on Tensor Core Units
  
  Haozhi Han, Kun Li, Wei Cui, Donglin Bai, Yiwei Zhang, Liang Yuan, Yifeng Chen, Yunquan Zhang, Ting Cao, Mao Yang
  
  ACM SIGPLAN Symposium on Principles and Practice of Parallel Programming (PPoPP) | March 2025
  
  DOI
- Empowering In-Browser Deep Learning Inference on Edge Devices with Just-in-Time Kernel Optimizations
  
  Fucheng Jia, Shiqi Jiang, Ting Cao, Wei Cui, Tianrui Xia, Xu Cao, Yuanchun Li, Qipeng Wang, Deyu Zhang, Ju Ren, Yunxin Liu, Lili Qiu, Mao Yang
  
  the 22nd Annual International Conference on Mobile Systems, Applications and Services (MobiSys'24) | June 2024
  
  Publication
- Query Processing on Gaming Consoles
  
  Wei Cui, Qianxi Zhang, Jesús Camacho-Rodríguez, Spyros Blanas, Brandon Haynes, Yinan Li, Peng Cheng, Ravishankar Ramamurthy, Rathijit Sen, Matteo Interlandi
  
  June 2023
  
  DOI Publication

- MetroRLHF: Enabling Memory-Effective Training for On-Policy RLHF via Adaptive Sequence Streaming
  
  Wei Cui, Peng Cheng
  
  NeurIPS 2025 The First Workshop of Efficient Reasoning | December 2025
  
  Publication
- FlashFFTStencil: Bridging Fast Fourier Transforms to Memory-Efficient Stencil Computations on Tensor Core Units
  
  Haozhi Han, Kun Li, Wei Cui, Donglin Bai, Yiwei Zhang, Liang Yuan, Yifeng Chen, Yunquan Zhang, Ting Cao, Mao Yang
  
  ACM SIGPLAN Symposium on Principles and Practice of Parallel Programming (PPoPP) | March 2025
  
  DOI
- Empowering In-Browser Deep Learning Inference on Edge Devices with Just-in-Time Kernel Optimizations
  
  Fucheng Jia, Shiqi Jiang, Ting Cao, Wei Cui, Tianrui Xia, Xu Cao, Yuanchun Li, Qipeng Wang, Deyu Zhang, Ju Ren, Yunxin Liu, Lili Qiu, Mao Yang
  
  the 22nd Annual International Conference on Mobile Systems, Applications and Services (MobiSys'24) | June 2024
  
  Publication
- Query Processing on Gaming Consoles
  
  Wei Cui, Qianxi Zhang, Jesús Camacho-Rodríguez, Spyros Blanas, Brandon Haynes, Yinan Li, Peng Cheng, Ravishankar Ramamurthy, Rathijit Sen, Matteo Interlandi
  
  June 2023
  
  DOI Publication
- Tutel: Adaptive Mixture-of-Experts at Scale
  
  Changho Hwang, Wei Cui, Yifan Xiong, Ziyue Yang, Ze Liu, Han Hu, Zilong Wang, Rafael Salas, Jithin Jose, Prabhat Ram, Joe Chau, Peng Cheng, Fan Yang, Mao Yang, Yongqiang Xiong
  
  July 2022
  
  Publication
- Roller: Fast and Efficient Tensor Compilation for Deep Learning
  
  Hongyu Zhu , Ruofan Wu , Yijia Diao , Shanbin Ke , Haoyu Li , Chen Zhang , Jilong Xue, Lingxiao Ma, Yuqing Xia, Wei Cui, Fan Yang, Mao Yang, Lidong Zhou, Asaf Cidon , Gennady Pekhimenko
  
  The 16th USENIX Symposium on Operating Systems Design and Implementation (OSDI'22) | July 2022
  
  Publication Project
- OpEvo: An Evolutionary Method for Tensor Operator Optimization
  
  Xiaotian Gao, Wei Cui, Lintao Zhang, Mao Yang
  
  AAAI 2021 | May 2021
  
  Publication
- Rammer: Enabling Holistic Deep Learning Compiler Optimizations with rTasks
  
  Lingxiao Ma, Zhiqiang Xie, Zhi Yang, Jilong Xue, Youshan Miao, Wei Cui, Wenxiang Hu, Fan Yang, Lintao Zhang, Lidong Zhou
  
  14th USENIX Symposium on Operating Systems Design and Implementation (OSDI'20) | November 2020
  
  Publication Project
- Adaptive Page Migration Policy With Huge Pages in Tiered Memory Systems
  
  Taekyung Heo, Yang Wang, Wei Cui, Jaehyuk Huh, Lintao Zhang
  
  IEEE Transactions on Computers | November 2020, Vol 71(1): pp. 53-68
  
  DOI Publication

Wei Cui

Publications

MetroRLHF: Enabling Memory-Effective Training for On-Policy RLHF via Adaptive Sequence Streaming

FlashFFTStencil: Bridging Fast Fourier Transforms to Memory-Efficient Stencil Computations on Tensor Core Units

Empowering In-Browser Deep Learning Inference on Edge Devices with Just-in-Time Kernel Optimizations

Query Processing on Gaming Consoles

Roller: Fast and Efficient Tensor Compilation for Deep Learning

Tutel: Adaptive Mixture-of-Experts at Scale

OpEvo: An Evolutionary Method for Tensor Operator Optimization

Adaptive Page Migration Policy With Huge Pages in Tiered Memory Systems

Rammer: Enabling Holistic Deep Learning Compiler Optimizations with rTasks

MetroRLHF: Enabling Memory-Effective Training for On-Policy RLHF via Adaptive Sequence Streaming

MetroRLHF: Enabling Memory-Effective Training for On-Policy RLHF via Adaptive Sequence Streaming

FlashFFTStencil: Bridging Fast Fourier Transforms to Memory-Efficient Stencil Computations on Tensor Core Units

Empowering In-Browser Deep Learning Inference on Edge Devices with Just-in-Time Kernel Optimizations

Tutel: Adaptive Mixture-of-Experts at Scale

Roller: Fast and Efficient Tensor Compilation for Deep Learning

OpEvo: An Evolutionary Method for Tensor Operator Optimization

Adaptive Page Migration Policy With Huge Pages in Tiered Memory Systems

Rammer: Enabling Holistic Deep Learning Compiler Optimizations with rTasks

FlashFFTStencil: Bridging Fast Fourier Transforms to Memory-Efficient Stencil Computations on Tensor Core Units

Empowering In-Browser Deep Learning Inference on Edge Devices with Just-in-Time Kernel Optimizations

Query Processing on Gaming Consoles

MetroRLHF: Enabling Memory-Effective Training for On-Policy RLHF via Adaptive Sequence Streaming

FlashFFTStencil: Bridging Fast Fourier Transforms to Memory-Efficient Stencil Computations on Tensor Core Units

Empowering In-Browser Deep Learning Inference on Edge Devices with Just-in-Time Kernel Optimizations

Query Processing on Gaming Consoles

Tutel: Adaptive Mixture-of-Experts at Scale

Roller: Fast and Efficient Tensor Compilation for Deep Learning

OpEvo: An Evolutionary Method for Tensor Operator Optimization

Rammer: Enabling Holistic Deep Learning Compiler Optimizations with rTasks

Adaptive Page Migration Policy With Huge Pages in Tiered Memory Systems