Die KI-Suchmaschine für Jobs
Senior/Staff AI Graph Compiler Engineer(m/w/x)
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.
Anforderungen
- 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
Aufgaben
- 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
Berufserfahrung
- 3 - 5 Jahre
Ausbildung
- Master-Abschluss
Sprachen
- Englisch – verhandlungssicher
Tools & Technologien
- MLIR
- PyTorch
- ONNX
- TensorFlow
- C++
- Python
- AI accelerators
- CPU
- NPU
- GPU
- Torch-MLIR
- TVM
- XLA
- Glow
Benefits
Betriebliche Altersvorsorge
- Pension plan
Gesundheits- & Fitnessangebote
- Extensive employee insurances
Attraktive Vergütung
- Company shares option
Noch nicht perfekt?
- Axelera AIVollzeitRemoteSeniorZürich
- Axelera AI
Senior/Staff Applications Engineer - Embedded AI(m/w/x)
VollzeitRemoteBerufserfahrenZürich, München - Dentsply Sirona
Senior Software Engineer (AI and Machine Learning)(m/w/x)
Vollzeitmit HomeofficeManagementZürich - Imnoo AG
CAD AI Machine Learning / Deep Learning Engineer – Coding Wizard(m/w/x)
Vollzeitmit HomeofficeSeniorOpfikon - ANYbotics
Senior AI Research Engineer, Visual Perception(m/w/x)
Vollzeitmit HomeofficeSeniorZürich
Senior/Staff AI Graph Compiler Engineer(m/w/x)
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.
Anforderungen
- 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
Aufgaben
- 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
Berufserfahrung
- 3 - 5 Jahre
Ausbildung
- Master-Abschluss
Sprachen
- Englisch – verhandlungssicher
Tools & Technologien
- MLIR
- PyTorch
- ONNX
- TensorFlow
- C++
- Python
- AI accelerators
- CPU
- NPU
- GPU
- Torch-MLIR
- TVM
- XLA
- Glow
Benefits
Betriebliche Altersvorsorge
- Pension plan
Gesundheits- & Fitnessangebote
- Extensive employee insurances
Attraktive Vergütung
- Company shares option
Über das Unternehmen
Axelera AI
Branche
IT
Beschreibung
The company is creating the next-generation AI platform to support advancements in humanity and improve the world.
Noch nicht perfekt?
- Axelera AI
Senior/Staff Backend Compiler Engineer(m/w/x)
VollzeitRemoteSeniorZürich - Axelera AI
Senior/Staff Applications Engineer - Embedded AI(m/w/x)
VollzeitRemoteBerufserfahrenZürich, München - Dentsply Sirona
Senior Software Engineer (AI and Machine Learning)(m/w/x)
Vollzeitmit HomeofficeManagementZürich - Imnoo AG
CAD AI Machine Learning / Deep Learning Engineer – Coding Wizard(m/w/x)
Vollzeitmit HomeofficeSeniorOpfikon - ANYbotics
Senior AI Research Engineer, Visual Perception(m/w/x)
Vollzeitmit HomeofficeSeniorZürich