Avatar
Lei Zhang
Director of Engineering AMD AI Group
LLMs GPU compilers, kernels, systems: Triton, TokenSpeed, ROCm. ex-Google: IREE, MLIR, TensorFlow, SPIR-V, Vulkan.

About Me

I work on open-source AI compiler and runtime solutions within the AMD AI Group. I lead Triton and related compiler and kernel efforts, co-founded LightSeek, and co-created TokenSpeed. Together with many other talented colleagues and community contributors, we are building domain-specific languages, compilers, kernels, and inference systems that help AMD GPUs shine.

Previously, I was fortunate to work with many great Googlers on projects at Google spanning machine learning toolchains (MLIR, IREE, TensorFlow, and TFLite), SPIR-V toolchains (DirectXShaderCompiler, SPIRV-Tools, Glslang, Shaderc, rspirv, and shaderc-rs), and the Vulkan ecosystem (VK-GL-CTS, GAPID, µVkCompute, and hardware-perfcounter). You can find a summary here.