Keren Zhou
Keren Zhou
Home
Experience
Projects
Featured
Publications
Talks
Students
Tags
News
Light
Dark
Automatic
Triton
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
Proton: Introduction and Development
Presented the ongoing work on Proton
Oct 21, 2024 9:41 PM — 9:41 PM
Virtual
Yuanwei Fang
,
Corbin Robeck
,
Keren Zhou
Project
Slides
Dev Tools: Proton/Interpreter
Presented the Proton and Interpreter tools in the Triton project.
Sep 17, 2024 9:41 PM — 9:41 PM
Virtual
Keren Zhou
Project
Slides
Video
Update on Triton's Interpreter
Review Triton’s Interpreter’s progress and future plans
Apr 3, 2024 10:03 PM — 10:03 PM
Virtual
Keren Zhou
Project
Slides
Proton: A Profiler for Triton
Went through Proton’s design overview
Feb 20, 2024 10:03 PM — 10:03 PM
Virtual
Keren Zhou
Project
Slides
PyTorch 2: Faster Machine Learning Through Dynamic Python Bytecode Transformation and Graph Compilation
This paper introduces two extensions to the popular PyTorch machine learning framework, TorchDynamo and TorchInductor, which implement …
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
Cite
Project
DOI
URL
Cite
×