Software Engineer - AI Research Clusters
NVIDIA Ltd.
yesterday
Role details
Contract type
Permanent contract Employment type
Full-time (> 32 hours) Working hours
Regular working hours Languages
English Experience level
Senior Compensation
$ 242KJob location
Tech stack
JavaScript
API
C++
CSS
Computer Clusters
Relational Databases
Linux
Distributed Systems
General-Purpose Computing on Graphics Processing Units
Python
Performance Tuning
Software Engineering
AI Platforms
Gitlab-ci
Kubernetes
Information Technology
Deployment Automation
Slurm
Machine Learning Operations
Virtual Agents
Software Coding
REST
Docker
Job description
- In this position, you will work with coworkers across the AI Platform organization to understand the pain points of validating, monitoring and operating GPU clusters at scale. Then you will design, develop and maintain engineering solutions to solve those pain points systematically.
- You will also research in traditional AIOps and the emerging Agentic AI, and leverage it to further reduce the operation toil.
- You will participate in on-call support for systems, platforms built and owned by the team.
Requirements
- BS/MS in Computer Science, Engineering, or equivalent experience.
- 5+ years in software/platform engineering, including 3+ years in ML infrastructure or distributed systems.
- Experience in software development lifecycle on Linux-based platforms.
- Strong coding skills in languages such as Python, C++ or Rust.
- Experience with Docker, Kubernetes, GitLab CI, automated deployments.
- Experience with AIOps or Agentic AI and apply it successfully in production environment.
Ways To Stand Out From The Crowd:
- Proficiency with full-stack development: Relational Data Modeling, DB optimization, REST API Semantics, Javascript, CSS, providing API as a service.
- Passion for building developer-centric platforms with great UX and strong operational reliability.
- Experience running Slurm or custom scheduling frameworks in production ML environments.
- Familiarity with GPU computing, Linux systems internals, and performance tuning at scale.
Benefits & conditions
Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 152,000 USD - 241,500 USD for Level 3, and 184,000 USD - 287,500 USD for Level 4.
About the company
NVIDIA is at the forefront of innovations in Artificial Intelligence, High-Performance Computing, and Visualization. Our invention-the GPU-functions as the visual cortex of modern computing and is central to groundbreaking applications from generative AI to autonomous vehicles. We are now looking for a Senior Software Engineer to help accelerate the next era of machine learning innovation.
In this role, you will propose and implement engineering solutions to ensure delivery of functional, reliable, secure, and performance-optimal GPU clusters to internal researchers, enable them to focus on training and development by reducing operational disruption and overhead, empower them for self-service continuous improvement on reliability, operational excellence & performance. Your work will empower scientists and engineers to train, fine-tune, and deploy the most advanced ML models on some of the world's most powerful GPU systems.