Skip to content
Neuer Job?Nejo!

Die KI-Suchmaschine für Jobs

AXAxelera AI

Senior/Staff AI Graph Compiler Engineer(m/w/x)

Zürich
VollzeitRemoteBerufserfahren
AI/ML

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

  • Englischverhandlungssicher

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
Die Originalanzeige dieses Stellenangebotes in der aktuellsten Version findest du hier. Nejo hat diesen Job automatisch von der Website des Unternehmens Axelera AI erfasst und die Informationen auf Nejo mit Hilfe von KI für dich aufbereitet. Trotz sorgfältiger Analyse können einzelne Informationen unvollständig oder ungenau sein. Bitte prüfe immer alle Angaben in der Originalanzeige! Inhalte und Urheberrechte der Originalanzeige liegen beim ausschreibenden Unternehmen.

  • Axelera AI

    Senior/Staff Backend Compiler Engineer(m/w/x)

    VollzeitRemoteSenior
    Zürich
  • Axelera AI

    Senior/Staff Applications Engineer - Embedded AI(m/w/x)

    VollzeitRemoteBerufserfahren
    Zürich, München
  • Dentsply Sirona

    Senior Software Engineer (AI and Machine Learning)(m/w/x)

    Vollzeitmit HomeofficeManagement
    Zürich
  • Imnoo AG

    CAD AI Machine Learning / Deep Learning Engineer – Coding Wizard(m/w/x)

    Vollzeitmit HomeofficeSenior
    Opfikon
  • ANYbotics

    Senior AI Research Engineer, Visual Perception(m/w/x)

    Vollzeitmit HomeofficeSenior
    Zürich
Alle 100+ ähnlichen Jobs ansehen

Nejo ist eine KI – Ergebnisse können unvollständig sein oder Fehler enthalten