Senior Systems Software Engineer, TAO Software and API

We are the NVIDIA-Metropolis TAO (Train-Adapt-Optimize) group. We are hiring a Senior Systems Software Engineer to join our TAO Software and API team. The TAO toolkit includes scalable and user-friendly modules for training, fine-tuning, and optimization of machine learning models, particularly in Computer Vision and Multi-Modal AI domains. If you love exploring new tech and want to help create ethical AI solutions, we'd love for you to think about joining our team at NVIDIA.

In this role, you will contribute to the development and optimization of advanced machine learning software and frameworks. You will also play a critical role in enhancing our CI/CD pipelines and DevOps practices to ensure efficient and reliable delivery of our software solutions. We are looking for someone with extensive experience in machine learning, strong leadership skills, and a deep understanding of CI/CD and cloud infrastructure.

TAO Toolkit | NVIDIA Developer | https://developer.nvidia.com/tao-toolkit

What you’ll be doing:

• Contribute the development and maintenance of advanced machine learning software and frameworks, optimizing for performance and scalability.

• Enhance CI/CD pipelines to streamline the development, testing, and deployment of large-scale machine learning models.

• Implement and manage cloud infrastructure for continuous integration, delivery, and deployment, ensuring high availability and scalability.

• Collaborate with cross-functional teams, including engineering, QA, and research, to improve development workflows and enhance software delivery speed and quality.

• Troubleshoot and resolve complex issues related to software development, containerization, and cloud infrastructure in production environments.

• Write and maintain robust documentation for development and deployment processes.

• Communicate effectively with technical and non-technical stakeholders to set shared expectations and ensure visibility around the release and deployment process.

• Lead code reviews, testing, and debugging to ensure high-quality code and efficient workflows.

• Mentor and guide junior engineers, fostering professional growth and enhancing team capabilities.

What we need to see:

• Bachelor’s or master’s degree or equivalent experience in Computer Science, Information Systems, Engineering, or other related fields.

• 7+ years of experience in software engineering, with hands-on experience in CI/CD, cloud infrastructure, and advanced machine learning frameworks.

• Proficiency with automation and orchestration tools, including Docker, Kubernetes, Jenkins, and Terraform or Ansible, or similar CI Tools.

• Experience with cloud platforms like AWS, Azure, or GCP.

• Strong programming skills in Python and/or other relevant languages.

• Experience in developing and deploying scalable software solutions.

• Strong analytical and problem-solving skills with a focus on practical and scalable solutions.

• Ability to work effectively in a collaborative environment and manage multiple tasks and projects.

• Familiarity with version control systems and configuration management.

Ways to stand out from the crowd:

• Demonstrated ability to learn quickly and adapt to new technologies.

• Strong academic record, with achievements in competitive programming or related areas.

• Extensive experience with advanced AI tools and frameworks, including LLMs and NVIDIA Blueprints.

• Contributions to open-source projects, demonstrating a collaborative and innovative mindset.

• Experience in deploying machine learning models on edge devices or platforms.

With competitive salaries and a generous benefits package, NVIDIA is widely considered to be one of the technology industry's most desirable employers. We have some of the most forward-thinking and versatile people in the world working with us, and our engineering teams are growing fast in some of the most impactful fields of our generation: Deep Learning, Artificial Intelligence, and Autonomous Vehicles. If you're a creative engineer who enjoys autonomy and shares our passion for technology, we want to hear from you.

The base salary range is 184,000 USD - 356,500 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.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.