True Talent Solutions Partners

Member of Technical Staff, ML Systems

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

Open. First seen 29 October 2026.

Apply on the company's site →

About the role

About the Role

Join an AI and machine learning team building high-performance training and inference systems for image, video, and world-model workloads. You will improve performance across the stack, from GPU kernels to distributed training and serving.

What You'll Do

  • Optimize training and inference performance for 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 approaches.
  • Build benchmarks and regression tests to preserve performance gains in production.
  • Co-design kernels, runtimes, and models with hardware in mind.

What We're Looking For

  • At least 2 years of experience in deep learning training or inference systems, GPU performance, or distributed systems.
  • Hands-on ML systems or GPU performance experience in a recent role, including work on inference or training performance.
  • Experience authoring core features in an inference or training framework, such as vLLM, SGLang, TensorRT-LLM, or Megatron.
  • Direct experience writing kernels for NVIDIA GPUs using CUDA, CUTLASS, Triton, or PTX.
  • Strong machine learning and computer science fundamentals, with a degree in computer science or a related quantitative field.
  • Experience optimizing diffusion, video, image, or other multimodal workloads is valuable.

Compensation & Benefits

Base salary is $180,000 to $260,000 annually. 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 106 open postings.

Standard for data science / ML roles

LLM · 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 107 open in Client Roles. 107 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 (57 of 107), TypeScript (43 of 107), LLM (37 of 107), React (37 of 107), Machine Learning (26 of 107), JavaScript (23 of 107), AWS (22 of 107)

OTHER ROLES ON THIS TEAM

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

Other openings for this role

2 postings for this role in Menlo Park since 27 October 2026, 1 still open. None has been taken down and posted again.

  • 29 October 2026 – open (this posting)
  • 27 October 2026 – closed 29 October 2026

Similar jobs

More jobs like this

SKILLSMachine Learning 19,746LLM 18,567Deep Learning 2,893CUDA 1,018AI and machine learning 8,792
PLACESMenlo Park 241United States 72,304
COMPANYJobs at True Talent Solutions Partners 107

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