> Markdown version of [/jobs/ext/2074191-senior-deep-learning-compiler-engineer-xla](https://www.wearedevelopers.com/jobs/ext/2074191-senior-deep-learning-compiler-engineer-xla). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md). --- # Senior Deep Learning Compiler Engineer - XLA - **Company:** NVIDIA Ltd. - **Location:** Santa Clara, CA, United States - **Experience:** Expert - **Salary:** $152,000.0 - $241,500.0 - **Contract:** Permanent contract - **Skills:** Artificial Intelligence, C++ (Programming Language), Code Generation, Nvidia CUDA, Computer Engineering, Software Debugging, Open Source Technology, OpenCL, Performance Tuning, Tensorflow, Software Engineering, Graphics Processing Unit (GPU), High Performance Computing, Pytorch, Distributed Programming, Deep Learning, Information Technology, Optimization Algorithms, Hardware Acceleration - **Published:** August 15, 2026 - **Apply:** https://www.disabledperson.com/jobs/74260749-senior-deep-learning-compiler-engineer-xla ## About the Role * 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. ## Description We are looking for versatile software engineers for our XLA team. NVIDIA is at the center for the AI revolution that's transforming how people live, work, and interact with technology. Come join us to build high-performance, production-grade software that's at the core of next-generation AI systems. 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. ## Related Videos - [From Model to Metal: An Open Source Stack for Accelerating Intelligence](https://www.wearedevelopers.com/videos/1636-from-model-to-metal-an-open-source-stack-for-accelerating-intelligence) - [Machine learning in the browser with TensorFlowjs](https://www.wearedevelopers.com/videos/155-machine-learning-in-the-browser-with-tensorflowjs) - [Photonic Computing: Programming a New Class of AI Accelerators (incl. Live Coding)](https://www.wearedevelopers.com/videos/100196-photonic-computing-programming-a-new-class-of-ai-accelerators-incl-live-coding) - [Developing an AI.SDK](https://www.wearedevelopers.com/videos/198-developing-an-ai-sdk) - [Making neural networks portable with ONNX](https://www.wearedevelopers.com/videos/301-making-neural-networks-portable-with-onnx) - [How We Built a Machine Learning-Based Recommendation System (And Survived to Tell the Tale)](https://www.wearedevelopers.com/videos/752-how-we-built-a-machine-learning-based-recommendation-system-and-survived-to-tell-the-tale) ## Related Articles - [What Are Large Language Models?](https://www.wearedevelopers.com/magazine/304-what-are-large-language-models) - [How to Become an AI Engineer](https://www.wearedevelopers.com/magazine/331-how-to-become-an-ai-engineer) - [MLOps And AI Driven Development](https://www.wearedevelopers.com/magazine/82-mlops-and-ai-driven-development) - [Highest Paying Tech Companies for Developers](https://www.wearedevelopers.com/magazine/220-highest-paying-tech-companies-for-developers) - [Best US AI Conferences for CTOs in 2026: Build vs. Buy, Vendor Evaluation, and Peer Intelligence](https://www.wearedevelopers.com/magazine/736-best-us-ai-conferences-for-ctos-in-2026-build-vs-buy-vendor-evaluation-and-peer-intelligence) - [What’s the latest in NVIDIA CUDA Python](https://www.wearedevelopers.com/magazine/568-what-s-the-latest-in-nvidia-cuda-python)