Senior Compiler Engineer (AI/Startup) – Silicon Valley

Our premier Silicon Valley AI startup client is hiring a Senior Compiler Engineer to design, develop, and maintain critical components of our AI-driven compilation stack. You will work on PyTorch, Triton, LLVM, and MLIR to build robust, scalable, and high-performance solutions for diverse hardware backends.

Note: All candidates MUST have at least 8 years of complier experience.

Responsibilities:

• Design, Implement, and Optimize Compiler Components: Architect and develop critical compiler modules to efficiently translate and optimize AI models for deployment across a variety of hardware platforms, including CPUs, GPUs, and emerging custom accelerators.
• Enhance and unify PyTorch Inductor, Triton, LLVM, and MLIR toolchains to support cutting-edge architectures, facilitate seamless interoperability, and enable rapid experimentation with new compiler features.
• Create and maintain custom IRs, code generation passes, and optimization strategies tailored for AI workloads, focusing on both general and domain-specific improvements.
• Profile and tune computational kernels—such as linear algebra operations, matrix multiplications, and elementwise computations—to achieve optimal performance, scalability, and resource efficiency on diverse hardware.
• Open-Source Engagement: Actively contribute to the LLVM, MLIR, Triton, and PyTorch open-source projects, sharing improvements, collaborating with the developer community, and driving the evolution of the AI compiler ecosystem.

Requirements:

• Bachelor’s or Master’s in Computer Science, Computer Engineering, or a related field.
• 8+ years experience in compiler engineering or closely related fields.
• Strong understanding of AI models – both training and inference pipelines
• Experience developing compiler support for custom hardware accelerators, including ASICs, FPGAs, or novel AI chips
• Deep knowledge of LLVM and MLIR internals, including IR transformations, code generation, and backend optimization techniques.
• Experience with PyTorch (Inductor/Dynamo) and Triton, including their compiler subsystems.
• Proven experience and expertise in C/C++ programming.
• Demonstrated expertise in performance optimization, including vectorization, parallelization, and hardware-specific tuning.
• Advanced debugging, analytical, and system-level thinking skills.
• Excellent communication skills with a strong track record of cross-functional collaboration.

Visa Requirements:

• Will transfer H1-B and TN Visas for candidates that resides in the San Francisco Bay/Silicon Valley Area.

#compiler #LLVM #MLIR #pytorch #artificialintelligence #softwareengineering #hiring #startup