Senior Software Engineer - AI Research Clusters

NvidiaSanta Clara, CA
82d$184,000 - $356,500

About The Position

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, enabling them to focus on training and development by reducing operational disruption and overhead, empowering 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.

Requirements

  • BS/MS in Computer Science, Engineering, or equivalent experience.
  • 8+ 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.

Nice To Haves

  • 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.

Responsibilities

  • Work with coworkers across the AI Platform organization to understand the pain points of validating, monitoring and operating GPU clusters at scale.
  • Design, develop and maintain engineering solutions to solve those pain points systematically.
  • Research in traditional AIOps and the emerging Agentic AI, and leverage it to further reduce the operation toil.
  • Participate in on-call support for systems and platforms built and owned by the team.

Benefits

  • Equity and benefits

Stand Out From the Crowd

Upload your resume and get instant feedback on how well it matches this job.

Upload and Match Resume

What This Job Offers

Job Type

Full-time

Career Level

Senior

Industry

Computer and Electronic Product Manufacturing

Education Level

Bachelor's degree

© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service