Lead AI Graph Compiler Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Modular · 1 month ago

Lead AI Graph Compiler Engineer

positionUnited States / Canada
timeFull-time
remoteOnsite
senioritySenior Level, Lead/Staff
money$234K/yr - $273K/yr
date5+ years exp

Modular is on a mission to revolutionize AI infrastructure by rebuilding the AI software stack. The Lead AI Graph Compiler Engineer will extend and develop Modular's graph compiler to support more features and hardware, setting the technical direction and mentoring junior engineers.

AI InfrastructureArtificial Intelligence (AI)Generative AIMachine LearningSoftware
check
H1B Sponsor Likelynote

Responsibilities

Set the technical direction for a team building the best heterogenous multi-device compiler and runtime system available, using MLIR and LLVM technologies
Collaborate with the Mojo compiler team to influence and harness Mojo's powerful compile-time meta-programming capabilities
Develop and disseminate deep expertise in new hardware platforms, partnering with Mojo kernel developers to achieve state-of-the-art performance on the latest hardware
Tackle complex technical challenges to empower ML engineers and accelerate model development
Strategically identify technical opportunities to grow the business
Mentor and develop junior engineers, growing the capability of the team over time

Qualification

Compiler engineeringMLIRLLVMC++Machine learning frameworksCuriosityTeam-oriented attitudeCreativity

Required

5+ years of compiler engineering experience
A track record of challenging the status quo and delivering significant, measurable improvements
Experience working with compilers for machine learning frameworks, such as PyTorch
Knowledge of core compiler algorithms and data structures
Knowledge of and experience working with MLIR and LLVM
Knowledge of C++, as well as, knowledge of basic GitHub workflows like pull requests
Creativity and curiosity for solving complex problems, a team-oriented attitude that enables you to work well with others, and alignment with our culture

Preferred

Experience with ML graph optimizations, parallel / distributed programming, heterogeneous ML computation, and/or code generation
Advanced degree in Computer Science or a related area is a plus

Benefits

Premier insurance plans
Up to 5% 401k matching
Flexible paid time off
Annual target bonus
Equity
Benefits

Company

Modular

twittertwittertwitter
company-logo
Modular provides AI infrastructure for deployment, serving, and programming GPUs.
dateFounded in 2022
location
Palo Alto, California, USA
people-size51-200 employees
web-link
https://www.modular.com

H1B Sponsorship

Modular has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (10)
2024 (6)
2023 (8)
2022 (4)

Funding

Current Stage
Growth Stage
Total Funding
$380M
Key Investors
US Innovative Technology FundGeneral CatalystGoogle Ventures
2025-09-24Series C· $250M
2023-08-24Series B· $100M
2022-06-30Seed· $30M

Leadership Team

leader-logo
Chris Lattner
CEO + Co-Founder
linkedin
leader-logo
Tim Davis
Co-Founder & President
linkedin

Recent News

Greylock
Greylock Portfolio - Modular
2025-12-29
W&B
Modular Raises $30 Million To Create Platform-Agnostic AI Dev Platform
2025-12-01
Crunchbase News
Highest Count Of New Unicorns Join Crunchbase Board In Over 3 Years As Exits Also Gain Steam
2025-10-15
Company data provided by crunchbase