Senior Python DL Automation Engineer, Deep Learning Algorithms

Shape the future of AI by contributing to software used by the global community. Collaborate with top-tier software engineers to develop a comprehensive toolset that rigorously tests deep learning models and frameworks on the most powerful computers. Ability to work in a multifaceted, fast-paced environment is required, as well as strong social skills. In this role, you will collaborate with internal partners, users, and the open-source community to develop and implement solutions for building, testing, integrating, and releasing NVIDIA AI services and deep learning frameworks on the most powerful, enterprise-grade GPU clusters, capable of hundreds of petaflops. You will work with internal engineering teams to deploy and operationalize AI models and services at scale by driving adoption for end-to-end AI solutions in the cloud and on-prem.

We are seeking passionate and hardworking Python developers to help us scale our AI services, platforms, models, and internal tools. You will be responsible for implementing and maintaining tools and infrastructure that enable our teams to productize the NVIDIA software stack, from deep learning frameworks to deep learning models and AI services. Are you ready for this challenge?

What you’ll be doing:

• Automating and optimizing testing of deep learning models and AI services from different domains, with a focus on inference.

• Developing shared utilities for setting up systems, running tests, and recording results.

• Configuring, maintaining, and building upon deployments of industry-standard tools.

• Lead best practices for building, testing, and releasing software, including AI services and Deep Learning models, and documenting them.

• Identifying infrastructure needs and translating them into action.

• Building tools for automatic content generation mechanisms that save dozens of engineering hours.

What we need to see:

• BSc or MS degree in Computer Science, Software Architecture or related engineering field (or equivalent experience).

• 5+ years of work experience in software development.

• Excellent Python programming and system design skills.

• Understanding of Deep Learning foundations, allowing benchmarking on DL models and AI services.

• Strong analytical and problem-solving skills, and a proactive, data-driven approach.

• Effective time-management and organization skills for coordinating multiple initiatives, priorities, and implementations of new technology and products into very complex projects.

• Effective communication, an open-minded attitude, and comprehensive documentation practices.

Ways to stand out from the crowd:

• Proficiency in Linux Environments and Containerization: 

• Expertise in Continuous Integration/Deployment (CI/CD) and Large-Scale Automation

• Familiarity with Front-end and Backend Python Frameworks

• Experience with High-Performance Computing (HPC) Clusters and Orchestration Solutions  like Slurm and Kubernetes

• Understanding of Cloud Services, MLOps, DevOps, SRE, and AI Agentic Tools

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 148,000 USD - 235,750 USD for Level 3, and 184,000 USD - 287,500 USD for Level 4.Applications for this job will be accepted at least until October 10, 2025.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.