NVIDIA is seeking versatile software engineers for its XLA team, which is at the forefront of the AI revolution. The role focuses on developing compiler optimization algorithms for deep learning workloads, specifically optimizing performance for the JAX framework and OpenXLA compiler on NVIDIA GPUs. Engineers will collaborate with deep learning framework teams and hardware architecture teams to enhance next-generation AI systems. This position offers the opportunity to work on high-performance, production-grade software that is crucial for advancing AI technology.

• Bachelor's, Master's, or Ph.D. in Computer Science, Computer Engineering, or a related field.
• 4+ years of relevant work or research experience in performance analysis and compiler optimizations.
• Excellent C/C++ programming and software design skills, including debugging and performance analysis.
• Strong foundation in the architecture of CPU, GPUs, or other high-performance hardware accelerators.
• Knowledge of high-performance computing and distributed programming.

• Develop compiler optimization algorithms for deep learning workloads.
• Optimize inference and training performance for the JAX framework and OpenXLA compiler on NVIDIA GPUs.
• Collaborate with partners in deep learning framework teams and hardware architecture teams.
• Craft and implement compiler optimization techniques for deep learning network graphs.
• Design novel graph partitioning and tensor sharding techniques for distributed training and inference.

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.