Senior Python DL Automation Engineer, Deep Learning Algorithms

Join NVIDIA’s AI Revolution: Shape the Future of Deep Learning

Ready to be at the forefront of artificial intelligence? At NVIDIA, we’re not just building technology; we’re shaping the future. We’re seeking a talented and passionate Software Engineer to join our team and play a critical role in developing tools that empower the global AI community.

Note: This is a summarized version. Click the ‘Apply’ button for the full job description.

About the Role: AI Software Engineer

As an AI Software Engineer, you’ll be instrumental in building and maintaining the infrastructure that enables our teams to productize NVIDIA’s cutting-edge software stack. This includes everything from deep learning frameworks to sophisticated AI services.

Your primary focus will be on automating and optimizing the testing of deep learning models and AI services, with a specific emphasis on inference. You’ll develop shared utilities for system setup, test execution, and result recording, ensuring the quality and reliability of our AI solutions.

Key Responsibilities:

• Collaborate with internal partners, users, and the open-source community to drive the development and implementation of solutions for building, testing, integrating, and releasing NVIDIA AI services and deep learning frameworks.
• Work closely with internal engineering teams to deploy and operationalize AI models and services at scale, ensuring seamless integration into our existing infrastructure.
• Champion the adoption of end-to-end AI solutions in both cloud and on-premise environments.
• Implement and maintain the tools and infrastructure essential for the productization of NVIDIA’s software stack.
• Lead the charge in establishing and documenting best practices for building, testing, and releasing software, including AI services and Deep Learning models.
• Identify infrastructure needs and translate them into actionable plans, driving continuous improvement in our development processes.
• Develop innovative tools for automatic content generation, unlocking significant time savings for our engineering teams.

What You’ll Bring to the Table:

We’re looking for a highly motivated individual with a strong foundation in software development and a passion for AI. The ideal candidate will possess the following:

• BSc or MS degree in Computer Science, Software Architecture, or a related engineering field.
• 5+ years of experience in software development, with a proven track record of success.
• Exceptional Python programming and system design skills.
• A solid understanding of Deep Learning foundations, enabling effective benchmarking of DL models and AI services.
• Strong analytical and problem-solving skills, coupled with a proactive and data-driven approach.
• Excellent time-management and organizational skills, capable of coordinating multiple initiatives and priorities within complex projects.
• Outstanding communication skills, an open-minded attitude, and meticulous documentation practices.

Required Expertise:

To excel in this role, you should possess a strong working knowledge of the following technologies:

• 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.
• A deep understanding of Cloud Services, MLOps, DevOps, SRE, and AI Agentic Tools.

Why NVIDIA?

NVIDIA is a world leader in accelerated computing, pioneering solutions that tackle challenges others can’t. Our work in AI and digital twins is transforming industries and impacting society in profound ways. We offer a dynamic and collaborative environment where you can learn, grow, and make a real difference. If you’re a creative and autonomous individual, we encourage you to apply and join our team of brilliant and forward-thinking professionals.

NVIDIA is an equal opportunity employer and values 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.

Learn more about NVIDIA here.