NVIDIA · 1 week ago
Senior Deep Learning Compiler Engineer - XLA
United States
Full-time
Remote
Mid, Senior Level
$152K/yr - $242K/yr
4+ years expNVIDIA is at the forefront of the AI revolution, seeking versatile software engineers for their XLA team. The role involves developing compiler optimization algorithms for deep learning workloads and optimizing performance for the JAX framework and OpenXLA compiler on NVIDIA GPUs.
Artificial Intelligence (AI)TransportationCloud ComputingEmbedded SoftwareGamingHardwareQuantum ComputingSemiconductorAI InfrastructureAutonomous VehiclesEmbedded SystemsFoundational AI
Responsibilities
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
Qualification
Required
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
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
Preferred
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
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
Benefits
Equity
Benefits
Company
NVIDIA
NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.
10001+ employees
H1B Sponsorship
NVIDIA has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
89%
Represents job field similar to this job
Engineering and Development
Management and Executive
Product Management
Marketing
Sales
Accounting and Finance
Customer Service and Support
Creatives and Design
Human Resources
Arts and Entertainment
Legal and Compliance
Trends of Total Sponsorships
*2026 (1247)
2025 (1868)
2024 (1353)
2023 (976)
2022 (835)
2021 (601)
2020 (529)
Funding
Current Stage
Public CompanyTotal Funding
$29.09BKey Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2026-06-15Post Ipo Debt· $25B
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
Leadership Team
Jensen Huang
Founder and CEO
Michael Kagan
Chief Technology Officer
Recent News
2026-07-12
Company data provided by crunchbase