The AI Job Search Engine
Senior Deep Learning Compiler Engineer - PyTorch(m/w/x)
Leading compiler design for deep learning workloads on thousands of GPUs at an AI/VR solutions developer. 8+ years experience with compiler concepts and deep learning frameworks required. Collaboration with engineers optimizing deep learning frameworks for NVIDIA hardware.
Requirements
- Bachelor's, Master's, or Ph.D. in Computer Science or related field, or equivalent experience
- 8+ years of relevant work experience
- Strong command of Python
- Experience with deep learning frameworks like PyTorch or JAX
- Solid foundation in compiler concepts
- Excellent communication and collaboration skills
- Previous contributions to deep learning compiler projects or frameworks
- Deep expertise in PyTorch internals
- Experience with JAX-like functional transformations
- Familiarity with parallel programming and distributed systems
- Track record of impactful participation in open-source communities
Tasks
- Lead the design and implementation of core compiler technologies
- Optimize and maintain compiler technologies for deep learning workloads
- Conduct performance analysis on workloads running on thousands of GPUs
- Identify optimization opportunities to enhance Thunder's design
- Collaborate with engineers who developed PyTorch for NVIDIA hardware
- Pioneer new features in compiler innovation
- Work with compiler, library, and systems teams to translate research into solutions
Work Experience
- 8 years
Education
- Bachelor's degreeOR
- Master's degree
Languages
- English – Business Fluent
Tools & Technologies
- Python
- PyTorch
- JAX
- CUDA
Not a perfect match?
- NVIDIAFull-timeOn-siteSeniorBerlin
- NVIDIA
Senior Developer Technology Engineer, Artificial Intelligence(m/w/x)
Full-timeOn-siteSeniorBerlin - Perplexity
Machine Learning Research Engineer - Search(m/w/x)
Full-timeOn-siteExperiencedBerlin - Helsing
AI Research Engineer - ML Engineering(m/w/x)
Full-timeOn-siteExperiencedBerlin, München - Prior Labs
MLOps / ML Systems Engineer(m/w/x)
Full-timeOn-siteSeniorBerlin, Freiburg im Breisgau
Senior Deep Learning Compiler Engineer - PyTorch(m/w/x)
Leading compiler design for deep learning workloads on thousands of GPUs at an AI/VR solutions developer. 8+ years experience with compiler concepts and deep learning frameworks required. Collaboration with engineers optimizing deep learning frameworks for NVIDIA hardware.
Requirements
- Bachelor's, Master's, or Ph.D. in Computer Science or related field, or equivalent experience
- 8+ years of relevant work experience
- Strong command of Python
- Experience with deep learning frameworks like PyTorch or JAX
- Solid foundation in compiler concepts
- Excellent communication and collaboration skills
- Previous contributions to deep learning compiler projects or frameworks
- Deep expertise in PyTorch internals
- Experience with JAX-like functional transformations
- Familiarity with parallel programming and distributed systems
- Track record of impactful participation in open-source communities
Tasks
- Lead the design and implementation of core compiler technologies
- Optimize and maintain compiler technologies for deep learning workloads
- Conduct performance analysis on workloads running on thousands of GPUs
- Identify optimization opportunities to enhance Thunder's design
- Collaborate with engineers who developed PyTorch for NVIDIA hardware
- Pioneer new features in compiler innovation
- Work with compiler, library, and systems teams to translate research into solutions
Work Experience
- 8 years
Education
- Bachelor's degreeOR
- Master's degree
Languages
- English – Business Fluent
Tools & Technologies
- Python
- PyTorch
- JAX
- CUDA
About the Company
NVIDIA
Industry
IT
Description
The company is developing groundbreaking solutions in Virtual Reality, Artificial Intelligence, Deep Learning, and Autonomous Vehicles.
Not a perfect match?
- NVIDIA
Principal AI Developer Technology Engineer(m/w/x)
Full-timeOn-siteSeniorBerlin - NVIDIA
Senior Developer Technology Engineer, Artificial Intelligence(m/w/x)
Full-timeOn-siteSeniorBerlin - Perplexity
Machine Learning Research Engineer - Search(m/w/x)
Full-timeOn-siteExperiencedBerlin - Helsing
AI Research Engineer - ML Engineering(m/w/x)
Full-timeOn-siteExperiencedBerlin, München - Prior Labs
MLOps / ML Systems Engineer(m/w/x)
Full-timeOn-siteSeniorBerlin, Freiburg im Breisgau