NVIDIA is seeking a senior software engineer to join the AI Networking co-design and benchmark R&D team. The role focuses on building and productizing machine learning tools for optimizing AI workloads across large GPU and CPU clusters. Candidates will work on distributed Deep Learning, particularly in LLM training and inference stacks, while engaging with diverse hardware and software layers. This position offers a unique opportunity to contribute to the infrastructure supporting next-generation large-scale AI systems.

• PhD or Master's degree in Computer Science, Software Engineering, or equivalent experience.
• 4+ years of experience applying machine learning techniques to computer architecture and system optimization problems.
• Hands-on experience developing and deploying various learning algorithms to tackle optimization challenges.
• Proficiency in building and using ML models with leading frameworks such as PyTorch or TensorFlow.
• Strong programming capabilities in Python, Bash, and C++.

• Design and implement resource allocation and combinatorial optimization techniques to optimize LLM models at datacenter scale.
• Research, develop, and deploy AI/ML techniques to optimize large-scale Deep Learning training and inference.
• Build and productionize ML-based tools for performance prediction and optimization.
• Develop and deploy a scalable, reliable data curation pipeline for complex data types.
• Collaborate across hardware and software teams to deliver valuable performance analysis insights.

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