Keren Zhou
Keren Zhou
Home
Experience
Lab
Projects
Featured
Publications
Talks
Students
Tags
News
Light
Dark
Automatic
Triton
Triton-Viz
Triton-Viz is an interactive visualization tool for understanding Triton programs, including execution structure, memory behavior, and GPU programming concepts in AI systems courses.
Code
Demo
Linear Layouts: Robust Code Generation of Efficient Tensor Computation Using F_2
Conference talk at ASPLOS'26 on Linear Layouts.
Mar 25, 2026 12:00 AM — 12:00 AM
Pittsburgh, USA
Keren Zhou
Project
Slides
Linear Layouts: Robust Code Generation of Efficient Tensor Computation Using F_2
Keren Zhou
,
Mario Lezcano
,
Adam Goucher
,
Akhmed Rakhmati
,
Jeff Niu
,
Justin Lebar
,
Pawel Szczerbuk
,
Peter Bell
,
Phil Tillet
,
Thomas Raoux
,
Zahi Moudallal
Cite
Project
arXiv
Triton-Sanitizer: A Fast and Device-Agnostic Memory Sanitizer for Triton with Rich Diagnostic Context
Hao Wu
,
Qidong Zhao
,
Songqing Chen
,
Yang Chen
,
Yueming Hao
,
Tony C. W. Liu
,
Sijia Chen
,
Adnan Aziz
,
Keren Zhou
Cite
Project
URL
Triton, Gluon, and the Future of Tile-Based Programming Models
Invited talk for UCSD CSE 291P on tile-based programming models.
Mar 1, 2026 12:00 AM — 12:00 AM
UC San Diego
Keren Zhou
Project
Slides
Proton: Towards Multi-level, Adaptive Profiling for Triton
Keren Zhou
,
Tianle Zhong
,
Hao Wu
,
Jihyeong Lee
,
Yue Guan
,
Yufei Ding
,
Corbin Robeck
,
Yuanwei Fang
,
Jeff Niu
,
Philippe Tillet
Cite
Project
URL
Tile-based Programming Models for AI
Introduction to Triton Programming Language for graduate students.
Apr 1, 2025 9:34 PM — 9:34 PM
Cambridge, MA
Keren Zhou
Project
Slides
Proton: Adaptive and Lightweight Profiling for Deep Learning Workloads
A gentle introduction to the Proton profiler and its updates.
Mar 1, 2025 9:34 PM — 9:34 PM
Las Vegas, NV
Keren Zhou
,
Corbin Robeck
,
Yuanwei Fang
Project
Slides
The Proton Dialect: An MLIR Dialect For AI Compiler GPU Kernel Profiling
Presented our work on multi-level instrumentation-based profiling for Triton kernels.
Mar 1, 2025 9:34 PM — 9:34 PM
Las Vegas, NV
Keren Zhou
,
Corbin Robeck
,
Yuanwei Fang
Project
Slides
SS1: Accelerating Inference with Fast and Expressive Sketch Structured Transform
Tensor multiplication with learned weight matrices is the fundamental building block in deep learning models. These matrices can often …
Aditya Desai
,
Kimia Saedi
,
Apoorv Walia
,
Jihyeong Lee
,
Keren Zhou
,
Anshumali Shrivastava
Cite
Project
»
Cite
×