NVIDIA Logo

NVIDIA

Senior Software Engineer, DGX Cloud AI Infrastructure

Reposted 2 Days Ago
In-Office or Remote
3 Locations
184K-357K Annually
Senior level
In-Office or Remote
3 Locations
184K-357K Annually
Senior level
Lead the optimization and performance analysis of distributed training and inference workloads on NVIDIA GPU platforms, with responsibilities including debugging, benchmarking, and ensuring reliability of large-scale AI systems.
The summary above was generated by AI

NVIDIA is at the forefront of the generative AI revolution, building the software and systems that power the world’s most advanced large language model workloads. We are looking for a Senior Software Engineer to lead the bring-up, triage, benchmarking, analysis, and optimization of distributed training and inference workloads across NVIDIA GPU platforms at the largest scales we run.

In this role you will set technical direction across communication libraries, model frameworks, and inference/training stacks to ensure state-of-the-art LLM workloads run efficiently and reliably at scale. You will lead deep performance and reliability investigations on multi-GPU and multi-node deployments, define how we benchmark and qualify new platforms, and build the resilience and failure-attribution capabilities that keep large clusters productive. This is a hands-on senior individual-contributor role for an engineer who operates at the intersection of deep learning systems, GPU performance, distributed computing, and large-scale operations — and who raises the bar for the engineers around them.

What you’ll be doing:

  • Lead bring-up, validation, and debugging of large-scale AI clusters, infrastructure, and end-to-end workloads, setting the standard for how the team operates.

  • Bring up, tune, and benchmark AI pre-training, post-training, and inference workloads using PyTorch, NeMo / Megatron, TensorRT-LLM, and adjacent NVIDIA AI software stacks.

  • Profile and optimize end-to-end workload performance across compute, memory, networking, and communication layers using tools such as Nsight Systems, NCCL tests, and custom microbenchmarks.

  • Analyze scaling efficiency for distributed LLM workloads using data, tensor, pipeline, and expert parallelism across modern GPU clusters, and translate findings into concrete tuning guidance.

  • Own root-cause analysis of complex failures — hangs, performance regressions, topology sensitivity  in large distributed environments.

  • Define and build the resilience and failure-attribution stack: detecting, triaging, and attributing node, fabric, and workload failures across the cluster at scale.

  • Build repeatable benchmark suites, automation, acceptance criteria, and qualification workflows on new platforms.

  • Tune runtime settings, communication parameters, and deployment configurations in close partnership with framework, systems, and platform teams.

  • Deliver actionable, data-driven recommendations based on profiling, benchmark results, and cluster characterization.

  • Mentor engineers, drive technical standards, and act as a force multiplier across the broader performance and infrastructure organization.

What we need to see:

  • Bachelor’s or Master’s in Computer Science or a related technical field (or equivalent experience).

  • 8+ years of experience developing software infrastructure for large-scale AI or HPC systems, including a track record of technical leadership.

  • Expertise debugging and triaging AI applications across the full stack — from the application layer down to the hardware.

  • Deep hands-on experience with NCCL, CUDA-aware distributed execution, and debugging multi-GPU and multi-node workloads at scale.

  • Proven track record of architecting, debugging, and scaling large-scale distributed systems.

  • Expert-level Python and C/C++ programming skills.

  • Experience operating workloads in scheduled, containerized cluster environments.

  • Excellent analytical, debugging, and communication skills, with the ability to influence across teams.

Ways to stand out from the crowd:

  • Demonstrated experience debugging and optimizing AI workloads at large scale.

  • Deep familiarity with the RDMA software stack (NCCL, IB verbs, UCX, libfabric).

  • Strong knowledge of GPU cluster fabrics and topology, including NVLink, NVSwitch, PCIe, RoCE, and InfiniBand.

  • Experience building acceptance tests, benchmark harnesses, regression gates, or cluster qualification tooling for AI platforms.

  • Experience building resilience, fault-detection, or failure-attribution systems for datacenter-scale infrastructure.

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us. If you’re creative, autonomous, and love a challenge, we want to hear from you.

Your base salary will be determined based on your location, experience, and other objective job-related factors. The expected base salary ranges for this position are:
184,000 USD – 287,500 USD for Level 4; and 224,000 USD – 356,500 USD for Level 5.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until November 3, 2026.

This posting is for an existing vacancy. 

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering an inclusive 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.

NVIDIA Austin, Texas, USA Office

Austin, United States

Similar Jobs

2 Days Ago
In-Office or Remote
3 Locations
184K-357K Annually
Senior level
184K-357K Annually
Senior level
Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
Design, build, and maintain infrastructure software for large-scale AI training and inference. Improve efficiency, resiliency, availability, observability, and reliability across distributed AI systems. Develop APIs and tools, analyze failures from application through hardware levels, define reliability metrics, and support NVIDIA AI platforms. The role requires expertise in distributed systems, monitoring, debugging, software engineering practices, and AI infrastructure, with preferred experience in large clusters, RDMA stacks, telemetry, and deep learning frameworks.
Top Skills: CC++Continuous IntegrationElkInfiniband VerbsJaxLibfabricLokiNcclPrometheusPythonPyTorchRayRdmaTensorFlowUcx
Yesterday
Easy Apply
Remote or Hybrid
United States
Easy Apply
100K-166K Annually
Senior level
100K-166K Annually
Senior level
Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
Manage enterprise customer implementations of Samsara’s IoT platform, from deployment planning through adoption and value realization. Build launch roadmaps, track implementation and account health, train executives and end users, resolve challenges, and coordinate with Sales, Support, Sales Engineering, Product, and other teams. Mentor Customer Success and Support colleagues while managing multiple customer engagements and strengthening long-term customer relationships.
Top Skills: Enterprise SoftwareIotSaaS
Yesterday
Easy Apply
Remote or Hybrid
United States
Easy Apply
109K-129K Annually
Junior
109K-129K Annually
Junior
Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
Serve as a technical sales expert for Samsara’s IoT platform, supporting discovery, demonstrations, value assessments, proof-of-concepts, deployment planning, and customer integrations. Explain complex solutions to nontechnical stakeholders, advise installers, write API automation scripts, translate customer feedback to product teams, and support strategic sales cycles. The role is remote in the U.S. West region and requires travel up to 50%.
Top Skills: AIAutomation And Control SystemsBashBusiness Systems IntegrationCan BusCloud InfrastructureCloud SoftwareCloud-Connected HardwareComputer NetworkingComputer VisionElectrical SystemsElectronicsIotJavaScriptOpen ApiPto ApplicationsPython

What you need to know about the Austin Tech Scene

Austin has a diverse and thriving tech ecosystem thanks to home-grown companies like Dell and major campuses for IBM, AMD and Apple. The state’s flagship university, the University of Texas at Austin, is known for its engineering school, and the city is known for its annual South by Southwest tech and media conference. Austin’s tech scene spans many verticals, but it’s particularly known for hardware, including semiconductors, as well as AI, biotechnology and cloud computing. And its food and music scene, low taxes and favorable climate has made the city a destination for tech workers from across the country.

Key Facts About Austin Tech

  • Number of Tech Workers: 180,500; 13.7% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Dell, IBM, AMD, Apple, Alphabet
  • Key Industries: Artificial intelligence, hardware, cloud computing, software, healthtech
  • Funding Landscape: $4.5 billion in VC funding in 2024 (Pitchbook)
  • Notable Investors: Live Oak Ventures, Austin Ventures, Hinge Capital, Gigafund, KdT Ventures, Next Coast Ventures, Silverton Partners
  • Research Centers and Universities: University of Texas, Southwestern University, Texas State University, Center for Complex Quantum Systems, Oden Institute for Computational Engineering and Sciences, Texas Advanced Computing Center

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account