Senior GPU-Accelerated ML Engineer (GNNs & PyTorch)

NVIDIA Gruppe; Austin (TX); On-site; USD 170,000 - 280,000; Full time

Job summary

NVIDIA is seeking a highly experienced Senior Software Engineer to join a team building large-scale machine learning solutions on NVIDIA GPU infrastructure. You will help advance PyTorch-based frameworks and support tools for data science workloads.

This role emphasizes C++ software design, GPU-accelerated training/inference, collaboration with product teams, and technical leadership across multiple groups.

Qualifications

• Bachelor's degree (or equivalent) plus 5+ years of relevant experience in large-scale machine learning, deep learning, and data science; or a Master's/PhD plus 3+ years of experience.
• 3+ years of experience with PyTorch.
• 2+ years of experience training enterprise-scale ML models across distributed infrastructure.
• 2+ years of experience designing and operating efficient training and inference workflows on GPU infrastructure, including profiling, scaling, orchestration, and resource utilization.
• Excellent C++ programming and software design skills.
• Proven experience developing, debugging, and optimizing high-performance applications, preferably with GPU acceleration using CUDA.
• Strong collaboration, communication, and documentation habits.

Responsibilities

• Develop accelerated, PyTorch-based solutions for large-scale models, focusing on efficient training and inference on GPU infrastructure.
• Support CUDA-X Libraries and integrations used in PyTorch-based workflows.
• Partner with developers, PMs, and scientists to develop innovative GNN models and GPU-accelerated implementations.
• Gather technical requirements from customers and Solutions Architects to guide priorities.
• Provide technical leadership and mentorship to engineers across the team.
• Identify opportunities to improve the codebase and reduce maintenance through re-architecture.
• Apply agentic coding tools to identify/fix bugs, implement features, and refactor code.
• Solve complex issues, explain solutions, and coordinate across teams to meet objectives.

Skills

PyTorch

C++

GPU acceleration

Distributed training

Code optimization

Documentation

Collaboration

Education

Bachelor's degree or equivalent

Master's degree or PhD in relevant field

Tools

CUDA

Job description

NVIDIA is seeking a highly experienced Senior Software Engineer to join a team building large-scale machine learning solutions on NVIDIA GPU infrastructure. You will help advance PyTorch-based frameworks and support tools for data science workloads.

This role emphasizes C++ software design, GPU-accelerated training/inference, collaboration with product teams, and technical leadership across multiple groups.