Senior GPU Compiler Development Engineer- Pune-Nvidia Graphics Pvt Ltd-4-7 - TimesJobs.com

Nvidia Graphics Pvt Ltd; Pune District; On-site; INR 12,00,000 - 18,00,000; Full time

Job summary

A leading technology firm in Pune is seeking a Senior GPU Compiler Development Engineer to enhance PTX Compiler infrastructure for AI and Deep Learning applications. The ideal candidate will have a strong background in compiler development and C/C++ programming, with at least 4 years of experience. Responsibilities include collaborating with teams to design next-generation GPU programming models and ensuring high-quality code generation.

Qualifications

• 4 years of experience in compiler front end, programming language designs.
• Experience in any area of compiler development.
• Understanding of Processor ISA.

Responsibilities

• Provide stewardship for PTX ISA and Compiler infrastructure.
• Collaborate to design programming models for GPUs.
• Ensure compatibility and robustness in code generation.

Skills

C/C++ programming skills

Analytical skills

Verbal and written communication

Education

BS/MS/Ph.D. in Computer Science

Job description

Senior GPU Compiler Development Engineer

We are looking for an experienced Systems Software Compiler Engineer 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 is 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 support new compilation models for DL and Generative AI codes. You will be contributing towards evolving programming models 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.
• Collaborate with architecture and programming model teams to design and implement programming models for next generation GPUs.
• Work 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

• 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.

Locations

• India, Pune