Publications

Yue Guan, Xinwei Qiang, Zaifeng Pan, Daniels Johnson, Yuanwei Fang, Keren Zhou, Yuke Wang, Wanlu Li, Yufei Ding, Adnan Aziz (2025). Mercury: Unlocking Multi-GPU Operator Optimization for LLMs via Remote Memory Scheduling. Proceedings of the 31st ACM Symposium on Operating Systems Principles (SOSP).

Cite DOI PDF

Bowen Cui, Tejas Ramesh, Oscar Hernandez, Keren Zhou (2025). Comprehensive Evaluation of LLMs in HPC Code Performance Optimization. Proceedings of the Workshop on AI Assisted Software Development for HPC (AI4Dev).

Cite arXiv

Yue Guan, Yuanwei Fang, Keren Zhou, Corbin Robeck, Manman Ren, Zhongkai Yu, Yufei Ding, Adnan Aziz (2025). KPerfIR: Towards an Open and Compiler-centric Ecosystem for GPU Kernel Performance Tooling on Modern AI Workloads. Proceedings of the 20th USENIX Symposium on Operating Systems Design and Implementation (OSDI).

Cite DOI PDF arXiv

Qidong Zhao, Hao Wu, Yuming Hao, Zilingfeng Ye, Jiajia Li, Xu Liu, Keren Zhou (2025). DeepContext: A Context-aware, Cross-platform, and Cross-framework Tool for Performance Profiling and Analysis of Deep Learning Workloads. Proceedings of the 30th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS).

Cite DOI arXiv

Tejas Ramesh, Alexander Rush, Xu Liu, Binqian Yin, Keren Zhou, Shuyin Jiao (2025). Triton-Viz: Visualizing GPU Programming in AI Courses. Proceedings of the 56th ACM Technical Symposium on Computer Science Education (SIGCSE).

Cite Project URL

Aditya Desai, Kimia Saedi, Apoorv Walia, Jihyeong Lee, Keren Zhou, Anshumali Shrivastava (2024). SS1: Accelerating Inference with Fast and Expressive Sketch Structured Transform. The Thirty-eighth Annual Conference on Neural Information Processing Systems (NeuIPS).

Cite Project

Jason Ansel, Edward Yang, Horace He, Natalia Gimelshein, Animesh Jain, Michael Voznesensky, Bin Bao, Peter Bell, David Berard, Evgeni Burovski, Geeta Chauhan, Anjali Chourdia, Will Constable, Alban Desmaison, Zachary DeVito, Elias Ellison, Will Feng, Jiong Gong, Michael Gschwind, Brian Hirsh, Sherlock Huang, Kshiteej Kalambarkar, Laurent Kirsch, Michael Lazos, Mario Lezcano, Yanbo Liang, Jason Liang, Yinghai Lu, C. K. Luk, Bert Maher, Yunjie Pan, Christian Puhrsch, Matthias Reso, Mark Saroufim, Marcos Yukio Siraichi, Helen Suk, Shunting Zhang, Michael Suo, Phil Tillet, Xu Zhao, Eikan Wang, Keren Zhou, Richard Zou, Xiaodong Wang, Ajit Mathews, William Wen, Gregory Chanan, Peng Wu, Soumith Chintala (2024). PyTorch 2: Faster Machine Learning Through Dynamic Python Bytecode Transformation and Graph Compilation. Proceedings of the 29th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS).

Cite Project DOI URL

Keren Zhou, Karthik Ganapathi Subramanian, Po-Hsun Lin, Matthias Fey, Binqian Yin, Jiajia Li (2024). FASTEN: Fast GPU-accelerated Segmented Matrix Multiplication for Heterogenous Graph Neural Networks. Proceedings of the 38th ACM International Conference on Supercomputing (ICS).

Cite Project DOI URL

Zhen Xie, Murali Emani, Xiaodong Yu, Dingwen Tao, Xin He, Pengfei Su, Keren Zhou, Venkatram Vishwanath (2024). Centimani: Enabling Fast AI Accelerator Selection for DNN Training with a Novel Performance Predictor. USENIX Annual Technical Conference (USENIX ATC).

Cite Project URL

Aditya Desai, Keren Zhou, Anshumali Shrivastava (2023). Hardware-Aware Compression with Random Operation Access Specific Tile (ROAST) Hashing. Proceedings of the 40th International Conference on Machine Learning (ICML).

Cite Project URL

Binqian Yin, Qinhong Hu, Yingying Zhu, Keren Zhou (2023). Semi-supervised learning for shale image segmentation with fast normalized cut loss. Geoenergy Science and Engineering.

Cite DOI URL

Mao Lin, Keren Zhou, Pengfei Su (2023). DrGPUM: Guiding Memory Optimization for GPU-Accelerated Applications. Proceedings of the 28th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS).

Cite Project DOI URL

Keren Zhou, Yueming Hao, John Mellor-Crummey, Xiaozhu Meng, Xu Liu (2022). ValueExpert: Exploring Value Patterns in GPU-Accelerated Applications. Proceedings of the 27th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS).

Cite Project DOI URL

Binqian Yin, Qinhong Hu, Yingying Zhu, Chen Zhao, Keren Zhou (2022). Paw-Net: Stacking Ensemble Deep Learning for Segmenting Scanning Electron Microscopy Images of Fine-grained Shale Samples. Computers & Geosciences.

Cite DOI URL

Keren Zhou, Jonathon Anderson, Xiaozhu Meng, John Mellor-Crummey (2022). Low Overhead and Context Sensitive Profiling of GPU-Accelerated Applications. Proceedings of the 36th ACM International Conference on Supercomputing (ICS).

Cite Project DOI URL

Keren Zhou, Xiaozhu Meng, Ryuichi Sai, Dejan Grubisic, John Mellor-Crummey (2022). An Automated Tool for Analysis and Tuning of GPU-Accelerated Code in HPC Applications. IEEE Transactions on Parallel and Distributed Systems (TPDS).

Cite Project DOI URL

Ryuichi Sai, John Mellor-Crummey, Xiaozhu Meng, Keren Zhou, Mauricio Araya-Polo, Jie Meng (2022). Accelerating High-order Stencils on GPUs. Concurrency and Computation: Practice and Experience (CCPE).

Cite Project DOI URL

Keren Zhou, Xiaozhu Meng, Ryuichi Sai, John Mellor-Crummey (2021). GPA: A GPU Performance Advisor Based on Instruction Sampling. IEEE/ACM International Symposium on Code Generation and Optimization (CGO).

Cite Project DOI URL

Barbara Chapman, Buu Pham, Charlene Yang, Christopher Daley, Colleen Bertoni, Dhruva Kulkarni, Dossay Oryspayev, Ed D'Azevedo, Johannes Doerfert, Keren Zhou, Kiran Ravikumar, Mark Gordon, Mauro Del Ben, Meifeng Lin, Melisa Alkan, Michael Kruse, Oscar Hernandez, P. K. Yeung, Paul Lin, Peng Xu, Swaroop Pophale, Tosaporn Sattasathuchana, Vivek Kale, William Huhn, Yun (Helen) He (2021). Outcomes of OpenMP Hackathon: OpenMP Application Experiences with the Offloading Model. International Workshop on OpenMP (IWOMP).

Cite Project DOI URL

Aaron Thomas Cherian, Keren Zhou, Dejan Grubisic, Xiaozhu Meng, John Mellor-Crummey (2021). Measurement and Analysis of GPU-Accelerated OpenCL Computations on Intel GPUs. IEEE/ACM International Workshop on Programming and Performance Visualization Tools (ProTools).

Cite Project DOI URL

Keren Zhou, Laksono Adhianto, Jonathon Anderson, Aaron Cherian, Dejan Grubisic, Mark Krentel, Yumeng Liu, Xiaozhu Meng, John Mellor-Crummey (2021). Measurement and Analysis of GPU-accelerated Applications with HPCToolkit. Parallel Computing (PARCO).

Cite Project DOI URL

Keren Zhou, Mark W. Krentel, John Mellor-Crummey (2020). Tools for Top-down Performance Analysis of GPU-Accelerated Applications. Proceedings of the 34th ACM International Conference on Supercomputing (ICS).

Cite Project DOI URL

Keren Zhou, Yueming Hao, John Mellor-Crummey, Xiaozhu Meng, Xu Liu (2020). GVPROF: A Value Profiler for GPU-Based Clusters. International Conference for High Performance Computing, Networking, Storage and Analysis (SC).

Cite Project DOI URL

Keren Zhou, Mark Krentel, John Mellor-Crummey (2020). A Tool for Top-down Performance Analysis of GPU-Accelerated Applications. Proceedings of the 25th ACM SIGPLAN Symposium on Principles and Practice of Parallel Programming (PPoPP).

Cite Project DOI URL

Keren Zhou, John Mellor-Crummey (2019). A Tool for Performance Analysis of GPU-Accelerated Applications. IEEE/ACM International Symposium on Code Generation and Optimization (CGO).

Cite Project DOI URL

Keren Zhou, Guangming Tan, Wei Zhou (2018). Quadboost: A Scalable Concurrent Quadtree. IEEE Transactions on Parallel and Distributed Systems (TPDS).

Cite DOI URL

Xiuxia Zhang, Guangming Tan, Shuangbai Xue, Jiajia Li, Keren Zhou, Mingyu Chen (2017). Understanding the GPU Microarchitecture to Achieve Bare-Metal Performance Tuning. Proceedings of the 22nd ACM SIGPLAN Symposium on Principles and Practice of Parallel Programming (PPoPP).

Cite Project DOI URL

Keren Zhou, Guangming Tan, Xiuxia Zhang, Chaowei Wang, Ninghui Sun (2017). A Performance Analysis Framework for Exploiting GPU Microarchitectural Capability. Proceedings of the International Conference on Supercomputing (ICS).

Cite Project DOI URL

Qiang Li, Maojie Gu, Keren Zhou, Xiaoming Sun (2015). Multi-Classes Feature Engineering with Sliding Window for Purchase Prediction in Mobile Commerce. IEEE International Conference on Data Mining Workshop (ICDMW).

Cite DOI URL

Zilong Tan, Keren Zhou, Hao Zhang, Wei Zhou (2015). BF-MapReduce: A Bloom Filter Based Efficient Lightweight Search. IEEE Conference on Collaboration and Internet Computing (CIC).

Cite DOI URL