Senior Deep Learning Compiler Engineer for PyTorch

NVIDIA; Zürich; Vor Ort; CHF 292.500 - 507.000; Vollzeit

Benefits dieser Stelle

Highly competitive salaries

Extensive benefits package

Diversity and inclusion in workplace

Zusammenfassung

A leading AI technology company in Zurich is seeking experienced engineers to innovate AI compiler technology for accelerated computing. You will lead compiler design and optimization efforts to enhance deep learning frameworks like PyTorch, collaborating closely with experts in the field. Ideal candidates will possess deep learning and compiler expertise, along with strong Python skills. The role offers competitive salaries and a supportive workplace focused on diversity and inclusion.

Qualifikationen

• 8+ years of relevant work experience.
• Hands-on experience with deep-learning frameworks like PyTorch or JAX.
• Excellent communication and collaboration skills.

Aufgaben

• Lead design, implementation, optimisation, and maintenance of compiler technologies.
• Conduct performance analysis on workloads running on thousands of GPUs.
• Translate research into practical open-source solutions.

Kenntnisse

Python

Deep-learning frameworks (PyTorch, JAX)

Compiler concepts

Performance analysis

Collaboration skills

Ausbildung

Bachelor’s, Master’s, or Ph.D. in Computer Science or related field

Jobbeschreibung

A leading AI technology company in Zurich is seeking experienced engineers to innovate AI compiler technology for accelerated computing. You will lead compiler design and optimization efforts to enhance deep learning frameworks like PyTorch, collaborating closely with experts in the field. Ideal candidates will possess deep learning and compiler expertise, along with strong Python skills. The role offers competitive salaries and a supportive workplace focused on diversity and inclusion.