# Senior Deep Learning Compiler Engineer - XLA

**Company**: NVIDIA
**Experience**: senior
**Job type**: full-time
**Category**: Engineering
**Industry**: Technology

**Apply**: https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Senior-Deep-Learning-Compiler-Engineer---XLA_JR2019844?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply
**Canonical**: https://yubhub.co/jobs/job_a25b8c04-1a7

## Description

NVIDIA is seeking a Senior Deep Learning Compiler Engineer to join its XLA team. The successful candidate will develop compiler optimization algorithms for deep learning workloads, optimizing inference and training performance for the JAX framework and the OpenXLA compiler on NVIDIA GPUs at scale.

Responsibilities:

- Craft and implement compiler 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.

- Generate code for NVIDIA GPU backends using open-source compilers such as MLIR, LLVM, and OpenAI Triton.

- Design user-facing features in JAX and related libraries, and perform general software engineering tasks.

- Collaborate with GPU hardware engineering teams to design AI compiler software features for next-generation GPUs.

Requirements:

- 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, performance analysis, and test design.

- Strong foundation in CPU, GPU, or other high-performance hardware accelerator architecture.

- Knowledge of high-performance computing and distributed programming.

- CUDA or OpenCL programming experience is desired but not required.

- Experience with XLA, TVM, MLIR, LLVM, OpenAI Triton, deep learning models and algorithms, and deep learning framework design is a plus.

Benefits:

- Competitive salaries

- Generous benefits package

- Equity eligibility

## Skills

### Required
- C/C++
- compiler optimizations
- performance analysis
- GPU architecture
- high-performance computing
- distributed programming

### Nice to have
- CUDA
- OpenCL
- XLA
- TVM
- MLIR
- LLVM
- OpenAI Triton
- deep learning models and algorithms
- deep learning framework design

---

Source: [Apply at nvidia.wd5.myworkdayjobs.com](https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Senior-Deep-Learning-Compiler-Engineer---XLA_JR2019844?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply)