New Job?Nejo!

The AI Job Search Engine

NV
NVIDIA
3mo ago

Senior Deep Learning Compiler Engineer - PyTorch(m/w/x)

Berlin
Full-timeOn-siteSenior
AI/ML

Description

In this role, you will be at the cutting edge of AI compiler technology, driving the design and optimization of tools that enhance deep learning performance. Your work will involve collaboration with top engineers and performance analysis to shape the future of the PyTorch ecosystem.

Let AI find the perfect jobs for you!

Upload your CV and Nejo AI will find matching job offers for you.

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

Education

Bachelor's degree
OR
Master's degree
OR
Doctoral / PhD

Work Experience

8 years

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

Tools & Technologies

PythonPyTorchJAXCUDA

Languages

English – Business Fluent

Find the original job posting in its most current version here. Nejo automatically captured this job from the website of NVIDIA and processed the information on Nejo with the help of AI for you. Despite careful analysis, some information may be incomplete or inaccurate. Please always verify all details in the original posting! Content and copyrights of the original posting belong to the advertising company.
Not a perfect match?
100+ Similar Jobs in Berlin
  • NVIDIA

    Senior Developer Technology Engineer, Artificial Intelligence(m/w/x)

    Full-timeOn-siteSenior
    Berlin
  • NVIDIA

    Principal AI Developer Technology Engineer(m/w/x)

    Full-timeOn-siteSenior
    Berlin
  • Perplexity

    Machine Learning Research Engineer - Search(m/w/x)

    Full-timeOn-siteExperienced
    Berlin
  • AUTO1 Group

    Senior Computer Vision Engineer(m/w/x)

    Full-timeOn-siteSenior
    Berlin
  • RepRisk AG

    Senior AI Engineer(m/w/x)

    Full-timeOn-siteSenior
    Berlin
100+ View all similar jobs
Apply now