Senior Compiler Engineer – MLIR - Austin, United States - NVIDIA

Description

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 the 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. 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. You will be solving challenging problems working alongside some of the top minds in GPU computing and systems software. See your efforts in action as DL and Generative AI developers use new PTX compiler enhancements adopt simpler programming model and to achieve best performance of their applications.

What you will be doing:

• Provide stewardship for PTX Compiler infrastructure for Generative AI and DL.
• 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.
• Working with customers/partners to gather feedback and drive innovative ideas and features to incorporate into the product.

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 Compilers/Linkers.
• Prior work experience in LLVM and MLIR compiler stack
• 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.
• Good track record of developing and delivering software products.
• Ability to work independently and lead your own development effort.

Ways to stand out from the crowd:

• Development background in Compiler Front end
• Development experience in LLVM IR, MLIR
• Development experience in JIT compilers

NVIDIA is widely considered to be one of the technology world's most desirable employers. We have some of the most brilliant and hardworking people in the world working for us. If you're creative and autonomous, we want to hear from you

With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us and, due to unprecedented growth, our exclusive engineering teams are rapidly growing. If you're a creative and autonomous engineer with a real passion for technology, we want to hear from you.

The base salary range is 148,000 USD - 276,000 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and . NVIDIA accepts applications on an ongoing basis.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.