NVIDIA is seeking software engineers to join its Deep Learning models performance engineering team. The team focuses on building and optimizing libraries and tools that enhance the efficiency of AI applications. Engineers will work on open-source frameworks like PyTorch and JAX, contributing to state-of-the-art performance in deep learning. The role involves collaboration with various teams and the open-source community to drive innovations in AI technology.

• BS or equivalent experience in Computer Science, Electrical Engineering, or a related field.
• 3+ years of experience in C++ and Python programming.
• Strong background, experience, or coursework in parallel systems programming, preferably on GPUs.
• Knowledge of Computer Architecture, Code Optimization, and/or Operating Systems.
• Proven experience in developing large software projects.
• Excellent verbal and written communication skills.

• Build and support Transformer Engine, the open-source library for accelerating the training of Large Language Models.
• Collaborate on systems research that improves Deep Learning model performance.
• Implement, benchmark, and optimize new Deep Learning models to scale efficiently on NVIDIA GPUs and systems.
• Build and contribute to NVIDIA submissions on community benchmarks such as MLPerf.
• Engage with the open-source community and support enterprise customers and partners.
• Influence the design of new hardware generations and core platform software components for NVIDIA hardware and systems.

• 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.