# Member of Technical Staff, ML Systems at True Talent Solutions Partners - Company: True Talent Solutions Partners - Status: Open - Workplace: On-site - Location: Menlo Park - Level: Staff - Discipline: AI & ML - Employment: Full-time - Salary: $180K–260K - Posted: 2026-11-13 - Skills: machine learning, GPU kernels - Apply: https://jobs.ashbyhq.com/true-talent-solutions-partners/32ab5ed9-fe0a-412e-9b43-4d9428c18fb0 ## Description ABOUT THE ROLE Join an AI team building efficient training and inference systems for image, video, and world-model workloads. You will improve performance across GPU kernels, distributed execution, and multi-node serving. WHAT YOU'LL DO - Optimize system and GPU performance for training and inference across image, video, and world-model workloads. - Profile and address 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 benchmarking and regression harnesses to preserve performance gains in production. - Collaborate on hardware-aware kernel, runtime, and model design. WHAT WE'RE LOOKING FOR - Two or more years of experience in deep learning training or inference systems, distributed systems, or related ML systems work. - At least one year of hands-on ML systems or GPU performance work in a current or recent role. - Experience authoring core features in a training or inference framework, rather than focusing only on deployment or integration. - Direct GPU kernel development experience with CUDA, CUTLASS, Triton, or PTX on NVIDIA GPUs. - Experience improving inference or training performance across GPU kernels, runtime systems, or distributed execution. - Strong machine learning and computer science fundamentals, with a degree More True Talent Solutions Partners roles: https://deviantjobs.com/companies/true-talent-solutions-partners