Systems Software Engineer, Tao Toolkit

NVIDIA is hiring Systems Software Engineers to join the TAO Toolkit engineering team. Our team crafts building blocks to make AI easy to develop, integrate, and deploy. We are looking for dedicated engineers who can contribute to build and maintain the infrastructure, tools, and processes vital to support the machine learning model lifecycle in a production environment. If you possess a passion for cutting-edge technologies and a drive to optimize AI development, we invite you to join our dynamic team at NVIDIA!

What You’ll Be Doing

• Design, implement, and maintain an efficient and scalable CI/CD platform that would enable model versioning, packaging, and integration with existing systems.
• Automate the training, retraining and fine-tuning processes, ensuring regular model updates and improvements based on new data and performance feedback.
• Collaborate with data scientists and deep learning engineers to build and deploy machine learning models.

What We Need To See

• Experience developing and deploying automated CI systems using Jenkins, GitLab CI, etc.
• Familiarity with Agile and DevOps methodologies.
• BS or MS in Computer Science or a related field (or equivalent experience).
• Excellent communication and collaboration skills to work effectively in a cross-functional team environment.
• 1-3 years of experience
• Experience with cloud platforms such as AWS, Azure, or Google Cloud.
• Knowledge of containerization technologies like Docker and orchestration tools like Kubernetes.
• Strong programming and debugging skills with proficiency in scripting languages such as Python, Bash, or similar.
• Familiarity with deep learning frameworks such as TensorFlow, PyTorch, etc.
• Strong understanding of software development principles, practices, and methodologies.

Ways To Stand Out From The Crowd

• Proven work experience as a CI/CD Engineer or a similar role, preferably in the domain of deep learning applications.
• Exude high energy and a positive attitude.

NVIDIA is widely considered to be one of the technology industry's most desirable employers. We have some of the most forward-thinking and hard-working people working with us and our engineering teams. If you're a creative engineer with a real passion for building scalable and robust infrastructure, we want to hear from you!

The base salary range is $116,000 - $224,250. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and benefits .

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.