Senior PyTorch Deep Learning Compiler Engineer

NVIDIA Corporation; Warszawa; On-site; PLN 292,500 - 507,000; Full time

Benefits offered by this job

Extensive benefits package

Flexible work environment

Diversity and inclusion initiatives

Job summary

NVIDIA Corporation is seeking a Senior Deep Learning Compiler Engineer to enhance AI compiler technology, particularly for PyTorch models on NVIDIA GPUs. This role involves designing and optimizing compiler technologies while collaborating closely with leading engineers in the field.

Ideal candidates will have 8+ years of experience, expertise in Python, and proficiency with deep learning frameworks. A competitive salary range in Poland is offered, between 292,500 PLN and 507,000 PLN.

Qualifications

• 8+ years of relevant work experience.
• Hands-on experience with PyTorch or JAX.
• Strong command of Python.

Responsibilities

• Contribute to the design, implementation, and optimization of compiler technologies.
• Collaborate with engineers to enhance PyTorch for NVIDIA GPUs.
• Analyze workloads on GPUs for optimization.

Skills

Python

Deep learning frameworks

Collaborative skills

Compiler concepts

Education

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

Tools

PyTorch

JAX

Job description

NVIDIA Corporation is seeking a Senior Deep Learning Compiler Engineer to enhance AI compiler technology, particularly for PyTorch models on NVIDIA GPUs. This role involves designing and optimizing compiler technologies while collaborating closely with leading engineers in the field.

Ideal candidates will have 8+ years of experience, expertise in Python, and proficiency with deep learning frameworks. A competitive salary range in Poland is offered, between 292,500 PLN and 507,000 PLN.