Name
TK Wave: A Symbolic Python/MLIR DSL and Compiler for High Performance Machine Learning
Session Type
Technical Talk
Date & Time
Tuesday, April 15, 2025, 2:15 PM - 2:45 PM
Abstract/s
We introduce Wave, a Python DSL and compiler for high performance machine learning. Wave exposes a high level kernel language and user constraints to specify the distribution strategy for the kernel while using the existing IREE/MLIR stack for the low-level optimizations and code generation. The language and compiler make extensive use of symbolic data types to represent tensor shapes and memory access patterns that make it easier to reason about the kernel.
Speakers
Location Name
Pavilion Room