We are looking for experienced Systems SW Compiler Engineers for an exciting role in our PTX (Parallel Thread Execution) Compiler Development team. Join the PTX Compiler team and help drive PTX language design and PTX compiler evolution. PTX enables all GPU Computing applications including HPC, Deep Learning and Autonomous Driving. PTX provides a stable programming model and portable instruction set Architecture (ISA) for NVIDIA GPUs and used by all Compute programming languages compiled to NVIDIA GPUs. PTX is also used as a compiler target by various non-NVIDIA compilers. Work with NVIDIA GPU Architecture and CUDA Programming model teams to build abstractions to expose new GPU features in portable and performant ways in PTX ISA. PTX Compiler (PTXAS) apart from implementing PTX ISA is responsible for PTX Compiler Front End, interaction with optimizer and runtime aspects involving object files, debug information, linkers, loaders and Driver Compiler Interface.
As a senior member of the team, you will be responsible for leading efforts to enhance PTX Compiler infrastructure to enhance it to support new compilation models for DL and Generative AI codes. You will be contributing towards evolving programming model for Generative AI and DL applications on GPUs.
What you will be doing:
Provide stewardship for PTX ISA and PTX Compiler infrastructure for Generative AI and DL.
Collaborating with architecture and programming model teams to design and implement programming models for next generation GPUs.
Working closely with others to help design compilation stack and strategies for AI and DL workloads.
Collaborate closely with teams developing other related components to ensure compatibility, robustness and high-quality code generation.
What we need to see:
BS (or equivalent experience), MS or Ph.D. in Computer Science, Computer Engineering, or related fields.
4+ years of experience in the area of compiler front end, programming language designs, Compilers/Linkers.
Superb analytical and C/C++ programming skills.
Experience in any one area of compiler development including feature support, code generation and compiler infrastructure.
Excellent and strong interactive, verbal and written communications skills.
Understanding of any Processor ISA (GPU ISA a plus).
Good track record of developing, driving and delivering software products.
Ways to stand out from the crowd:
Experience in Programming Languages design and drafting programming language standards.
Knowledge of GPU development and compute APIs such as CUDA, and OpenCL.
Development experience in LLVM IR, MLIR
Top Skills
NVIDIA Pune, Mahārāshtra, IND Office
Survey No.144 145, Commerzone No.5, Off, Airport Rd, Yerawada, Pune, Maharashtra, India, 411006
