Staff Software Engineer - AI Compiler
Core
Design and implementation of an AI/ML compiler integrating with the ML software stack, covering model ingestion, low-level hardware programming, memory allocation, and performance optimization.
Role type
Staff Software Engineer (AI Compiler) (via careerplan.io/jobs/101241234368-staff-software-engineer-ai-compiler-at-arm)
Builds
AI/ML compiler and ML software stack components
Domain
Semiconductors / AI Hardware / Compiler Technology
Deliverable
production ML models
Required skills
C++, Python, Compiler development, MLIR, TVM, Software lifecycle management, Rigorous software testing, AI tool usage
Preferred skills
FPGA development, Inference engines (llama.cpp, vLLM), NPU/GPU architecture, Performance analysis of ML workloads
Technologies
MLIR, TVM, C++, Python, FPGAs, NPU, GPU
Responsibilities
Design and implement AI/ML compiler components, Integrate compiler with ML software stack, Optimize memory allocation and performance, Collaborate with specialists across the company
Seniority
Staff, hands-on IC with strategic impact
