Senior Compiler Engineer (ML)

About the role:I'm working with a well-funded startup building a next-generation AI infrastructure platform from the ground up. Their team is developing a modern ML compiler stack that takes models from PyTorch through Triton down to CUDA and machine IR, targeting both today's GPUs and tomorrow's custom AI accelerators.This is an opportunity to work at the intersection of compilers, machine learning systems, GPUs, and hardware/software co-design, helping define how AI workloads are compiled and executed on cutting-edge hardware.You'll be:Developing and optimizing a modern ML compiler stackLowering PyTorch and Triton workloads into efficient machine IRBuilding compiler infrastructure for GPUs (CUDA/ROCm) and emerging AI acceleratorsWorking closely with hardware and systems engineers to optimize performance across the stackLooking for:Strong compiler fundamentals and performance optimization experienceExperience with LLVM, MLIR, or modern compiler infrastructureKnowledge of PyTorch, Triton, and low-level code generationBackground in compilers, ML systems, or high-performance computingGPU architecture or hardware/software co-design experience is a strong plus