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.

• 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

NVIDIA is widely regarded as one of the most desirable employers in the technology world. We have some of the most brilliant and forward-thinking people in the world working for us. If you're creative and autonomous, we want to hear from you!

We are an equal opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status.