.png){kind=logo}
Design, implement, and optimize compiler components for AI workloads using PyTorch, Triton, LLVM, and MLIR. Build IRs, codegen passes, and performance-tuned kernels for CPUs, GPUs, and custom accelerators. Contribute upstream to open-source projects and collaborate cross-functionally to enable scalable, high-performance deployment across diverse hardware.
Description
Role Description:
We are looking for an experienced, highly skilled and motivated Senior Compiler Engineer to join our compiler team. In this role, you will be responsible for designing, developing, and maintaining 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. This is a hands-on technical position where you will solve complex problems, optimize for performance, and contribute to the next generation of our technology.
What You’ll Do:
- 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
What We’re Looking For:
- Bachelor’s or Master’s in Computer Science, Computer Engineering, or a related field from a recognized university
- 7+ years experience in compiler engineering or closely related fields
- 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
Ways to Stand Out from the Crowd:
- 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
- Active contributions to open-source compiler frameworks, demonstrating leadership and community involvement
- Familiarity with distributed training strategies, graph compilers, and advanced memory models
Similar Jobs
eCommerce • Information Technology • Retail • Industrial
Outside B2B sales role managing ~85 medium customers (~$2M portfolio). Build territory and account plans, grow customer relationships, maintain CRM pipeline, identify opportunities, negotiate deals, and conduct face-to-face customer meetings within an assigned geography.
eCommerce • Information Technology • Retail • Industrial
Outside B2B sales role managing ~85 medium-size MRO customers across a territory. Build territory and account plans, grow sales through relationship development, pipeline management, CRM use, customer meetings, negotiation, and multi-channel outreach. Regular face-to-face customer visits with occasional overnight travel.
Top Skills:
CRM
eCommerce • Information Technology • Retail • Industrial
Outside B2B sales role managing ~85 medium-sized customers and a $2M portfolio. Build territory and account plans, grow customer relationships, maintain pipeline, use CRM to track opportunities, and conduct face-to-face meetings within an assigned geography. Negotiate and close sales while balancing customer satisfaction and profitability.
Top Skills:
CRM
What you need to know about the San Francisco Tech Scene
San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.
Key Facts About San Francisco Tech
- Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Google, Apple, Salesforce, Meta
- Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
- Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
- Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine
