Your personal AI career agent
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
Like this job?
BetaYour Career Agent finds similar jobs for you every day.
Noch nicht perfekt?
- NVIDIAVollzeitnur vor OrtSeniorBerlin
- Helsing
AI Research Engineer - ML Engineering(m/w/x)
Vollzeitnur vor OrtBerufserfahrenBerlin, München - NVIDIA
Principal AI Developer Technology Engineer(m/w/x)
Vollzeitnur vor OrtSeniorBerlin - AUTO1 Group
Senior Computer Vision Engineer(m/w/x)
Vollzeitnur vor OrtSeniorBerlin - Perplexity
Machine Learning Research Engineer(m/w/x)
Vollzeitnur vor OrtBerufserfahrenBerlin
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
Like this job?
BetaYour Career Agent finds similar jobs for you every day.
About the Company
NVIDIA
Industry
IT
Description
The company is developing groundbreaking solutions in Virtual Reality, Artificial Intelligence, Deep Learning, and Autonomous Vehicles.
Noch nicht perfekt?
- NVIDIA
Senior Developer Technology Engineer, Artificial Intelligence(m/w/x)
Vollzeitnur vor OrtSeniorBerlin - Helsing
AI Research Engineer - ML Engineering(m/w/x)
Vollzeitnur vor OrtBerufserfahrenBerlin, München - NVIDIA
Principal AI Developer Technology Engineer(m/w/x)
Vollzeitnur vor OrtSeniorBerlin - AUTO1 Group
Senior Computer Vision Engineer(m/w/x)
Vollzeitnur vor OrtSeniorBerlin - Perplexity
Machine Learning Research Engineer(m/w/x)
Vollzeitnur vor OrtBerufserfahrenBerlin