Axelera AI
Senior/Staff Backend Compiler Engineer(m/w/x)
Full-timeRemoteSenior
Zürich
New Job?Nejo!
Your personal AI career agent
AXAxelera AI
Senior/Staff AI Graph Compiler Engineer(m/w/x)
Zürich
Full-timeRemoteExperienced
AI/ML
Nejo AI Summary
Design and optimize MLIR components for the Metis™ AI Platform at an AI platform developer. MLIR, compiler design, and graph-based IRs experience essential. Company shares option included.
Requirements
- Master's or PhD in Computer Science or related technical field
- 3-5 years Software Engineering experience; ideally 2 years Deep Learning/AI
- Experience with MLIR, dialects, passes, compiler infrastructure
- Solid understanding of compiler design principles, graph-based IRs
- Experience with AI model frameworks (PyTorch, ONNX, TensorFlow)
- Proven experience implementing graph-level optimizations
- Strong C++ and Python programming skills
- Experience in collaborative engineering teams
- Excellent communication, knowledge sharing, and mentoring skills
- Experience with custom AI accelerators or specialized hardware
- Background in computer architecture, heterogeneous systems
- Experience with AI compiler stacks (Torch-MLIR, TVM, XLA, Glow)
- Experience optimizing AI workloads
- Familiarity with frontend model ingestion, compiler pipelines
Tasks
- Design, implement, and maintain frontend/graph-level MLIR components
- Develop and optimize performance-critical graph transformations
- Extend and maintain MLIR dialects, passes, and infrastructure
- Integrate AI models from PyTorch, ONNX, TensorFlow
- Lower AI models into internal compiler representations
- Collaborate with hardware and backend compiler teams
- Ensure efficient mapping to heterogeneous architectures
- Support and mentor team members on MLIR infrastructure
- Analyze model graphs for optimization
- Implement optimizations for performance and efficiency
- Contribute to compiler architecture and tooling design
- Debug and profile compiler performance
- Improve compiler performance and correctness
Work Experience
- 3 - 5 years
Education
- Master's degree
Languages
- English – Business Fluent
Tools & Technologies
- MLIR
- PyTorch
- ONNX
- TensorFlow
- C++
- Python
- AI accelerators
- CPU
- NPU
- GPU
- Torch-MLIR
- TVM
- XLA
- Glow
Benefits
Retirement Plans
- Pension plan
Healthcare & Fitness
- Extensive employee insurances
Competitive Pay
- Company shares option
Find the original job posting in its most current version here. Nejo automatically captured this job from the website of Axelera AI and processed the information on Nejo with the help of AI for you. Despite careful analysis, some information may be incomplete or inaccurate. Please always verify all details in the original posting! Content and copyrights of the original posting belong to the advertising company.
Not a perfect match?
- Axelera AI
Senior/Staff Applications Engineer - Embedded AI(m/w/x)
Full-timeRemoteExperiencedZürich, München - Imnoo AG
CAD AI Machine Learning / Deep Learning Engineer – Coding Wizard(m/w/x)
Full-timeWith HomeofficeSeniorOpfikon - Dentsply Sirona
Senior Software Engineer (AI and Machine Learning)(m/w/x)
Full-timeWith HomeofficeManagementZürich - ANYbotics
Senior AI Research Engineer, Visual Perception(m/w/x)
Full-timeWith HomeofficeSeniorZürich
AXAxelera AI
Senior/Staff AI Graph Compiler Engineer(m/w/x)
Zürich
Full-timeRemoteExperienced
AI/ML
Nejo AI Summary
Design and optimize MLIR components for the Metis™ AI Platform at an AI platform developer. MLIR, compiler design, and graph-based IRs experience essential. Company shares option included.
Requirements
- Master's or PhD in Computer Science or related technical field
- 3-5 years Software Engineering experience; ideally 2 years Deep Learning/AI
- Experience with MLIR, dialects, passes, compiler infrastructure
- Solid understanding of compiler design principles, graph-based IRs
- Experience with AI model frameworks (PyTorch, ONNX, TensorFlow)
- Proven experience implementing graph-level optimizations
- Strong C++ and Python programming skills
- Experience in collaborative engineering teams
- Excellent communication, knowledge sharing, and mentoring skills
- Experience with custom AI accelerators or specialized hardware
- Background in computer architecture, heterogeneous systems
- Experience with AI compiler stacks (Torch-MLIR, TVM, XLA, Glow)
- Experience optimizing AI workloads
- Familiarity with frontend model ingestion, compiler pipelines
Tasks
- Design, implement, and maintain frontend/graph-level MLIR components
- Develop and optimize performance-critical graph transformations
- Extend and maintain MLIR dialects, passes, and infrastructure
- Integrate AI models from PyTorch, ONNX, TensorFlow
- Lower AI models into internal compiler representations
- Collaborate with hardware and backend compiler teams
- Ensure efficient mapping to heterogeneous architectures
- Support and mentor team members on MLIR infrastructure
- Analyze model graphs for optimization
- Implement optimizations for performance and efficiency
- Contribute to compiler architecture and tooling design
- Debug and profile compiler performance
- Improve compiler performance and correctness
Work Experience
- 3 - 5 years
Education
- Master's degree
Languages
- English – Business Fluent
Tools & Technologies
- MLIR
- PyTorch
- ONNX
- TensorFlow
- C++
- Python
- AI accelerators
- CPU
- NPU
- GPU
- Torch-MLIR
- TVM
- XLA
- Glow
Benefits
Retirement Plans
- Pension plan
Healthcare & Fitness
- Extensive employee insurances
Competitive Pay
- Company shares option
Find the original job posting in its most current version here. Nejo automatically captured this job from the website of Axelera AI and processed the information on Nejo with the help of AI for you. Despite careful analysis, some information may be incomplete or inaccurate. Please always verify all details in the original posting! Content and copyrights of the original posting belong to the advertising company.
About the Company
Axelera AI
Industry
IT
Description
The company is creating the next-generation AI platform to support advancements in humanity and improve the world.
Not a perfect match?
- Axelera AI
Senior/Staff Backend Compiler Engineer(m/w/x)
Full-timeRemoteSeniorZürich - Axelera AI
Senior/Staff Applications Engineer - Embedded AI(m/w/x)
Full-timeRemoteExperiencedZürich, München - Imnoo AG
CAD AI Machine Learning / Deep Learning Engineer – Coding Wizard(m/w/x)
Full-timeWith HomeofficeSeniorOpfikon - Dentsply Sirona
Senior Software Engineer (AI and Machine Learning)(m/w/x)
Full-timeWith HomeofficeManagementZürich - ANYbotics
Senior AI Research Engineer, Visual Perception(m/w/x)
Full-timeWith HomeofficeSeniorZürich