NVIDIA Corporation

Senior Deep Learning Compiler Engineer - XLA

NVIDIA Corporation • $152K — $287K *
US-AnywhereRemote in United States
Enterprise Technology
Less than 5 years of experience
Today
Be an Early Applicant
Job Overview by Ladders

Qualifications

  • Bachelor's, Master's or Ph.D. in Computer Science, Computer Engineering, or a related field (or equivalent experience).
  • 4+ years of relevant work or research experience in performance analysis and compiler optimizations.
  • Ability to work independently, define project goals and scope, and lead development efforts.
  • Excellent C/C++ programming and software design skills, including debugging and performance analysis.
  • Strong foundation in CPU, GPU architecture, high-performance hardware accelerators, and distributed programming.
  • Familiarity with CUDA or OpenCL is desirable but not mandatory.

Responsibilities

  • Develop compiler optimization algorithms for deep learning workloads.
  • Optimize inference and training performance for JAX and OpenXLA on NVIDIA GPUs.
  • Craft and implement optimization techniques for deep learning network graphs.
  • Design novel graph partitioning and tensor sharding techniques for distributed training and inference.
  • Perform performance tuning and analysis of deep learning applications.
  • Generate code for NVIDIA GPU backends using MLIR, LLVM, and OpenAI Triton.
  • Collaborate with GPU hardware engineers to design AI compiler software features.

Benefits

  • Equity options in the company.
  • Access to various employee benefits.
  • Opportunity to work on cutting-edge technology in deep learning.
  • Supportive environment for career development and mentorship.
Full Job Description
What you will be doing:

In this role, develop compiler optimization algorithms for deep learning workloads. You will optimize inference and training performance for the JAX framework and the OpenXLA compiler on NVIDIA GPUs at scale. You'll collaborate with our partners in deep learning framework teams and our hardware architecture teams to accelerate the next generation of deep learning software. The scope of these efforts include:
  • Crafting and implementing compiler optimization techniques for deep learning network graphs.
  • Designing novel graph partitioning and tensor sharding techniques for distributed training and inference.
  • Performance tuning and analysis.
  • Code-generation for NVIDIA GPU backends using open-source compilers such as MLIR, LLVM and OpenAI Triton.
  • Designing user facing features in JAX and related libraries and other general software engineering work.
  • Working closely with GPU hardware engineering teams to design AI compiler software features for next-generation GPUs.


What we need to see:
  • Bachelors, Masters or Ph.D. in Computer Science, Computer Engineering, related field (or equivalent experience).
  • 4+ years of relevant work or research experience in performance analysis and compiler optimizations.
  • Ability to work independently, define project goals and scope, and lead your own development effort adopting clean software engineering and testing practices.
  • Excellent C/C++ programming and software design skills, including debugging, performance analysis, and test design.
  • Strong foundation in architecture of CPU, GPUs or other high performance hardware accelerators. Knowledge of high-performance computing and distributed programming.
  • CUDA or OpenCL programming experience is desired but not required.
  • Experience with the following technologies is a huge plus: XLA, TVM, MLIR, LLVM, OpenAI Triton, deep learning models and algorithms, and deep learning framework design.
  • Strong interpersonal skills are required along with the ability to work in a dynamic product-oriented team. A history of mentoring junior engineers and interns is a bonus.


Ways to stand out from the crowd:
  • Experience working deep learning frameworks such as JAX, PyTorch or TensorFlow.
  • Extensive experience with CUDA or with GPUs in general.
  • Experience with open-source compilers such as XLA, LLVM, MLIR or TVM.


Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 152,000 USD - 241,500 USD for Level 3, and 184,000 USD - 287,500 USD for Level 4.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until August 18, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

#deeplearning

About NVIDIA Corporation

Nvidia, a global leader in graphics, gaming, and AI technology, offers Nvidia careers and internship opportunities for those passionate about driving innovation in the tech industry. you'll find a company committed to growth, teamwork, and leadership in computer science and machine learning domains.

About Nvidia

A Pioneer in Technology and Innovation

Nvidia has cemented its reputation as a powerhouse in developing advanced graphics processing units (GPUs) and has significantly contributed to the gaming industry's evolution. Moreover, its foray into AI and machine learning has opened new frontiers in technology, making Nvidia a beacon of innovation and a desirable workplace for ambitious tech professionals.

Job Opportunities

Diverse Positions in a Dynamic Field

Nvidia is continuously on the lookout for talented individuals across various domains, including hardware and software engineering, product design, marketing, and sales. Employment opportunities at Nvidia are vast, catering to a wide range of expertise and career aspirations.

Employment in Hardware and Graphics

For those fascinated by the intricacies of hardware and graphics technology, Nvidia offers positions that sit at the forefront of gaming and computing advancements.

Growth in Machine Learning and AI

Nvidia's leadership in AI and machine learning has created numerous vacancies for specialists eager to contribute to groundbreaking projects.

Recruitment in Computer Science

With the constant demand for innovation, Nvidia's recruitment efforts focus on computer science experts capable of pushing the boundaries of what's possible.

Internship Program

Opening Doors to Future Innovators

Nvidia's internship program is designed to nurture the next generation of technology leaders, offering hands-on experience in a culture that celebrates creativity and teamwork.

Benefits and Culture

Interns at Nvidia enjoy a plethora of benefits, from competitive stipends to mentorship opportunities, all within an environment that values growth and learning.

Opportunities for Students

Whether you're an undergraduate, a master's student, or a Ph.D. candidate, Nvidia's internships provide a real-world glimpse into the tech industry, offering valuable experience in various technology fields.

Pathways to Full-Time Employment

Many interns have transitioned into full-time positions, marking the start of successful careers at Nvidia. The internship program is more than a stepping stone into the company; it’s an investment in the professional development of interns. The goal is to ensure that interns are well-equipped for future challenges.

Nvidia Careers: More Than Just a Job

Nvidia offers more than just a job to its employees; it provides a front-row seat on the journey into the future of technology. Nvidia stands as a pillar of innovation with its vast opportunities in hardware, graphics, gaming, machine learning, and computer science. Nvidia careers serve as a launching pad for talented workers who aim to redefine the technological landscape. Whether through full-time positions or internships, joining Nvidia means contributing to a legacy of breakthroughs and becoming part of a global community dedicated to pushing the boundaries of what's possible.
Learn more about NVIDIA Corporation
Size
22,473 employees
Market Cap
$350.4 billion
Industry
Manufacturing & Automotive
Net Income
$4.3 billion
Founded
1993
5 Year Trend
+31.3%
Revenue
$16.6 billion
NASDAQ
NVDA
* Ladders Estimates

Similar Jobs

Ad banner background

Get Ready For Your
Next Interview

More Jobs at NVIDIA Corporation

More Enterprise Technology Jobs

Find similar Senior Deep Learning Compiler Engineer - XLA jobs:

NationwideRemote