Senior Deep Learning Compiler Engineer – Open Source Impact

NVIDIA; Warszawa; On-site; PLN 293,000 - 507,000; Full time

Job summary

NVIDIA is seeking passionate engineers to build the next generation of tools for AI developers and researchers. You will lead the design and optimization of core compiler technologies that accelerate large DL workloads on NVIDIA GPUs, collaborating with teams behind CUDA, nvFuser, TVM, and TorchInductor.

Ideal candidates have a strong background in Python, compiler concepts, and DL frameworks like PyTorch or JAX, with 8+ years of experience and excellent collaboration skills in a distributed,

Qualifications

• A Bachelor's, Master's, or Ph.D. in Computer Science or a related technical field (or equivalent experience).
• 8+ years of relevant work experience.
• Strong command of Python and experience building complex, well-tested software systems.
• Hands-on experience with deep learning frameworks like PyTorch or JAX. You understand how models are built and where the performance challenges lie.
• A solid foundation in compiler concepts such as abstract syntax trees (ASTs), intermediate representations (e.g., SSA form), program analysis, and code generation.
• Excellent communication and collaboration skills, essential for working effectively in a distributed, open-source environment.

Responsibilities

• Lead the design, implementation, optimization, and maintenance of core compiler technologies that accelerate deep learning workloads.
• Collaborate with engineers who built PyTorch for NVIDIA hardware to add new features.
• Perform performance analysis on workloads across GPUs to identify optimization opportunities.

Skills

Python

Deep learning frameworks

PyTorch/JAX

Compiler concepts (AST/SSA)

English communication

Education

Bachelor's/Master's/PhD in CS or related field

Tools

PyTorch

JAX

CUDA

TorchDynamo/TorchInductor

TVM/MLIR experience

Job description

NVIDIA is seeking passionate engineers to build the next generation of tools for AI developers and researchers. You will lead the design and optimization of core compiler technologies that accelerate large DL workloads on NVIDIA GPUs, collaborating with teams behind CUDA, nvFuser, TVM, and TorchInductor.

Ideal candidates have a strong background in Python, compiler concepts, and DL frameworks like PyTorch or JAX, with 8+ years of experience and excellent collaboration skills in a distributed,