Senior MLOps Engineer - DSX Enablement

NVIDIA AI; Indiana (PA); On-site; USD 160,000 - 210,000; Full time

Job summary

NVIDIA is seeking a Senior MLOps Engineer to join the DSX Enablement team, collaborating with customers to implement and enhance AI workloads. You will build AI solutions on NeoCloud and DGX/NG Cloud platforms, tune distributed training, and help optimize production ML pipelines across frameworks.

You will engage as a primary technical contact for customers, guide joint engagements, and contribute to open-source tools and reference architectures for scalable AI systems, including exploration of

Qualifications

• BS/MS/PhD in CS, EE, or related field, or equivalent experience.
• 8+ years in data science, data engineering, or ML engineering for large-scale systems.
• AI/ML lifecycle experience from exploration to production.
• Linux, batch schedulers, Kubernetes, distributed filesystems, datacenter networking.
• Scripting in Bash and Python; systems programming in C++, Go, or Rust.
• Experience with ML frameworks for training and inference.
• Excellent communication and ability to present architectures to engineers and leadership.
• Strong engineering discipline and track record on meaningful projects.

Responsibilities

• Develop solutions advancing AI infrastructure capabilities and ML workloads.
• Advise on demands of ML workloads and diagnose full-stack AI/ML issues.
• Collaborate with teams building infrastructure software and accelerated frameworks.
• Profile and tune large-scale training/inference workloads on NVIDIA platforms.
• Contribute open-source tools and reference architectures for scalable ML workloads.

Skills

Communication & presentation

Scripting & programming

Linux OS

Python

C++

Go

Rust

bash

Kubernetes

ML frameworks

Education

BS/MS/PhD in CS/EE or related

Tools

DGX Cloud

Linux

Distributed filesystems

Networking (datacenter)

Job description

Job Requisition ID JR2024864

Job Category Engineering

Time Type Full time

NVIDIA is seeking a Senior MLOps Engineer to join our DSX Enablement team, collaborating closely with strategic customers to implement and enhance groundbreaking AI workloads. We partner with the world's most innovative AI companies and open-source communities to address their most challenging technical problems.

What You Will Be Doing

In this role, you will develop innovative solutions that advance AI infrastructure capabilities, advise infrastructure experts on the demands of ML workloads, help practitioners diagnose and solve full-stack AI and ML system problems, and work on a team with direct responsibility for the success of internal and external customers’ AI and ML initiatives, including LLM performance evaluation and supporting new hardware in open-source frameworks. You will:

• Build and deploy custom AI solutions on NeoCloud platforms and NVIDIA Cloud Partners (NCPs), including distributed training, inference optimization, and MLOps pipelines,
• Act as a primary technical contact for internal and external customers and partners, guiding joint engagements, ensuring the success of initiatives on DGX Cloud, and solving complex problems in production,
• Work closely with the teams building the infrastructure software and accelerated frameworks that support today’s most compelling AI applications,
• Profile and tune large-scale training and inference workloads on NCP platforms, leading efforts to reduce latency, cost, and operational risk, and
• Develop open-source tools and reference architectures to make it easier to build and manage machine learning and AI workloads, pipelines, and systems at scale.

What We Need To See

• BS, MS, or Ph.D. in Computer Science, Computer/Electrical Engineering, or a related technical field, or equivalent experience.
• 8+ years of experience in technical roles such as data science, data engineering, or ML engineering, ideally targeting large- scale production systems.
• Demonstrated AI/ML experience across multiple phases of the machine learning lifecycle, from exploratory analysis to production systems.
• Facility with systems topics including Linux, batch schedulers, Kubernetes, distributed filesystems, and advanced networking at datacenter scale.
• Solid scripting and programming skills in languages like bash and Python and solid systems programming skills in a language like C++, Go, or Rust.
• Experience using machine learning or deep learning frameworks for training and inference.
• Excellent communication and technical presentation skills, with the ability to clearly articulate architectures, trade-offs, and recommendations to both engineering and leadership audiences.
• A clear record of engineering discipline and execution on interesting projects, whether you’re working alone or collaborating on a team.

Ways To Stand Out From The Crowd

• Experience contributing to and working in open-source communities.
• Experience with the NVIDIA ecosystem, including DGX systems, CUDA, NeMo, RAPIDS, Triton, NIM, and NVIDIA networking technologies such as InfiniBand, NVLink, and RoCE.
• Experience and familiarity building machine learning systems in a security-critical environment and distributed training and inference frameworks.
• Familiarity with MLOps practices in a cloud-native context: containerization, CI/CD pipelines, workflow automation, observability stacks, and GitOps workflows.
• Direct experience drawing on deep systems knowledge to diagnose and fix performance or correctness problems that span multiple layers of the application stack, like hardware, networking, accelerator, hypervisor or OS, compilers or runtimes, application code, and libraries.

NVIDIA offers competitive salaries and a generous benefits package. It is recognized as one of the technology world’s most desirable employers. We have some of the most innovative and dedicated people working here. Due to rapid growth, our outstanding teams are expanding quickly. Join us to make a lasting impact on the world!