True Talent Solutions Partners

Member of Technical Staff, ML Systems

Menlo Park
Salary
$180,000 - $260,000
Level
Staff
Work type
On-site · Menlo Park
Visa
Sponsors visas

Open. First seen 8 November 2026.

Apply on the company's site →

About the role

About the Role

Join a technical team building efficient training and inference systems for image, video, and world-model workloads. You will improve performance across GPU kernels, distributed engines, and multi-node infrastructure, helping make demanding AI workloads faster and more efficient.

What You'll Do

  • Optimize system and GPU performance for training and inference across image, video, and world-model workloads.
  • Profile bottlenecks at the kernel, memory, system, and cluster levels using Nsight and related tools.
  • Implement low-level optimizations with CUDA and Triton.
  • Design and improve distributed training and inference engines for diffusion models.
  • Improve communication performance across GPUs and nodes using NCCL, RDMA, InfiniBand, RoCE, and disaggregated serving.
  • Build benchmarking and regression harnesses to sustain performance improvements in production.
  • Co-design kernels, runtimes, and models with hardware constraints in mind.

What We're Looking For

  • At least 2 years of experience with deep learning training or inference systems, distributed systems, or related ML systems work.
  • At least 1 year of hands-on ML systems or GPU performance experience in a full-time role.
  • Experience authoring core features in a training or inference framework, rather than only deploying or integrating one.
  • Directly written NVIDIA GPU kernels using CUDA, CUTLASS, Triton, or PTX.
  • Experience improving inference or training performance through kernels, runtimes, or distributed execution.
  • Strong computer science and machine learning fundamentals, with a degree in computer science or a related quantitative field.
  • Experience optimizing diffusion, video, image, or other multimodal workloads is relevant.

Compensation & Benefits

Annual base salary: $180,000 to $260,000 USD. Visa sponsorship is available.

Location

On-site in Menlo Park, California, United States.

What this role asks for

Particular to this role

CUDA · Deep Learning

Rare in True Talent Solutions Partners's other 129 open postings.

Standard for data science / ML roles

Machine Learning

Common in data science / ML postings at any company.

Compared with True Talent Solutions Partners's other open jobs.

The team: Client Roles

This role is one of 130 open in Client Roles. 130 were posted in the last 30 days, including 5 lead roles: the team is hiring the people who run it too.

WHAT THE TEAM WORKS WITH, ACROSS ITS POSTINGS

Python (68 of 130), LLM (45 of 130), TypeScript (39 of 130), React (37 of 130), Machine Learning (35 of 130), AWS (31 of 130), GCP (22 of 130)

OTHER ROLES ON THIS TEAM

From the team and dates on True Talent Solutions Partners's own job board.

Posting history

Reposted 9 times since 27 October 2026: a posting for this role in Menlo Park was taken down and the role was posted again.

Similar jobs

More jobs like this

SKILLSMachine Learning 20,214Deep Learning 2,969CUDA 1,041AI and machine learning 9,071
PLACESMenlo Park 242United States 75,765
COMPANYJobs at True Talent Solutions Partners 130

Is this your posting? Email info@careerholo.com to have it removed.