← All roles

Senior Deep Learning Compiler Engineer - HW-SW Codesign

NVIDIA· Engineer, Sys SW
Apply on company site
Hybrid (Remote / Local)$152k - $241.5kFull-timePosted 2mo ago

About the role

Senior-level software engineer role focusing on compiler optimization for deep learning workloads, optimizing inference and training performance for JAX and OpenXLA on NVIDIA GPUs, and collaborating with framework and hardware teams to accelerate next-generation AI systems.

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.

Qualifications

  • 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 XLA, TVM, MLIR, LLVM, OpenAI Triton, deep learning models and algorithms, and deep learning framework design is a huge plus.
  • Strong interpersonal skills and ability to work in a dynamic product-oriented team. Mentoring junior engineers and interns is a bonus.

Skills

Required:Compiler optimizationGraph partitioning and tensor shardingPerformance tuning and analysisC/C++ programming
Nice to have:CUDA/OpenCLXLA/MLIR/LLVM/OpenAI Triton

Benefits

  • equity
  • benefits
Last updated Sep 18, 2026. Interested? Apply on the company site.

About NVIDIA

World Leader in AI Computing

NVIDIA invents the GPU and drives advances in AI, HPC, gaming, creative design, autonomous vehicles, and robotics.

AIautonomous vehiclesgaminggpugraphics cardhigh performance computingliquid-ai-recruiting-targetllm
Company profileWebsite

More open roles at NVIDIA