Compiler Engineer (Frontend)
Core
Design and build a compiler converting AI models from popular ML frameworks into assembly code for accelerator hardware, implementing novel scheduling algorithms.
Role type
Senior IC compiler engineer (AI hardware)
Builds
Production compiler and runtime systems for AI accelerators
Domain
AI hardware, compiler infrastructure, high-performance computing
Deliverable
production ML models
Required skills
C++, Python, compiler development, scheduling algorithms, model architecture analysis, hardware optimization, multithreaded programming, large codebase management
Preferred skills
Auto-scheduling, program synthesis, AI tools usage, mentoring
Technologies
LLVM, MLIR, TVM, XLA, IREE, PyTorch, TensorFlow, JAX, Halide, Ansor
Responsibilities
Design and build the compiler converting AI models to assembly code, develop novel scheduling algorithms, collaborate on hardware/software solutions, analyze system efficiency and scalability, provide technical leadership and mentorship
Seniority
Senior, hands-on IC with leadership responsibilities