Software Engineer, AI Compiler
Benefits:
Offers Equity
Education Requirements:
BS, MS, or PhD in Computer Science, Electrical/Computer Engineering, or related field
Experience Requirements:
4+ years of hands-on ML compiler or systems engineering experience
Demonstrated experience building and owning an end-to-end compiler stack
Experience working with machine learning models, neural network graphs, and graph optimizations
Comfortable collaborating with hardware teams to map novel architectural primitives
Strong understanding of compiler performance trade-offs, profiling, and bottleneck analysis
Responsibilities:
Work across the full stack with software, systems, and hardware teams
Contribute to shaping the long-term compiler architecture and tooling strategy
Design and implement parts of the compiler stack targeting our novel AI accelerator
Build and evolve MLIR/LLVM based infrastructure
Collaborate closely with hardware architects, microarchitects, and research teams
Show more details