Avatar
Lei Zhang
Senior Software Engineer Google
AI Frameworks & Compilers. Now: Vulkan compute, IREE, MLIR. Previous: Vulkan graphics, SPIR-V toolchain.

About This Website

Hello, thanks for your interest in my website. 😊 I try to log down my learnings and thoughts about compilers (SPIR-V, MLIR, etc.), GPU (Vulkan, Metal, etc.), toolchain (ML frameworks, productivity tools, etc.), and others sporadically.

About Me

I’m Lei Zhang. I work on machine learning frameworks and compilers within Google.

Currently my main focus is pushing forward leveraging a compilation-based approach and Vulkan compute for ML inference on GPUs, together with many Google colleagues and open source contributors. It involves developing a compiler stack and various other tools; for example, I created µVkCompute, maintain the SPIR-V dialect and conversions in MLIR, and contribute to IREE, a MLIR-based end-to-end compiler and runtime that draws lots of design principles from Vulkan.

Previously I contributed to various general Vulkan and SPIR-V toolchain projects; for example, tech lead and initial enabler of HLSL to SPIR-V compilation in DirectXShaderCompiler, many fundamental features in SPIRV-Tools, and also SPIRV-Headers, Shaderc, Glslang, VK-GL-CTS.

I also created a few SPIR-V projects in the Rust programming language: rspirv—Rust implementation of SPIR-V module processing functionalities, and shaderc-rs—Rust bindings for the shaderc library.