Member of Technical Staff, ML Systems
$180k - $230kFull-time
True Talent Solutions Partners
About the Role
Join an ML systems team building and improving the training and inference stack for image, video, and world-model workloads. You will focus on making these systems faster and more efficient, from GPU kernels through distributed execution and serving.
What You'll Do
Optimize training and inference performance for image, video, diffusion, and world-model workloads.
Profile bottlenecks across kernels, memory, systems, and clusters using Nsight and related tools.
Implement low-level GPU optimizations with CUDA and Triton.
Design and improve distributed training and inference engines for diffusion models.
Improve communication across GPUs and nodes using technologies such as NCCL, RDMA, InfiniBand, and RoCE.
Build benchmarks and regression tests to help performance improvements persist in production.
Co-design kernels, runtimes, and models with hardware performance in mind.
What We're Looking For
At least 2 years of experience in deep learning training or inference systems, distributed systems, or related ML systems work.
Hands-on experience writing GPU kernels or low-level optimizations with CUDA, CUTLASS, Triton, or PTX on NVIDIA GPUs.
Experience contributing core features to an inference or training framework, rather than focusing only on deployment or integration.
Experience with multi-GPU or multi-node communication technologies such as NCCL, RDMA, InfiniBand, or RoCE.
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.
Compensation & Benefits
Salary range: $180,000 to $230,000 annually. Visa sponsorship is available.
Location
On-site in Menlo Park, California, United States.
Vacancy posted 2 days ago
Similar jobs that could be interesting for youBased on the Member of Technical Staff, ML Systems in Menlo Park, CA vacancy
$180k - $230k
...Job Description Job Description Member of Technical Staff, ML Systems Company: TensorScale Location: Menlo Park, CA (in office 5 days per week) Compensation: $180,000 - $230,000 + competitive equity Employment Type: Full-time Visa Sponsorship: Visa...SuggestedFull timeH1bWork at officeVisa sponsorshipTransparent Search Group
Menlo Park, CA19 days ago- ...an adversary that never sleeps. What You'll Do Design Core Systems: Architect the services, pipelines, and storage engines... ...time streaming and large-scale storage. Infrastructure for AI/ML workloads (inference, model serving, data systems). Enterprise...SuggestedFull timeWork at officeFlexible hours
$180k - $230k
...AI infrastructure team building efficient training and inference systems for image, video, and world-model workloads. You will improve... ...from GPU kernels to distributed serving, helping make demanding ML workloads faster and more efficient. What You'll Do Optimize...SuggestedFull timeVisa sponsorship$230k - $350k
...Job Description Job Description Member of Technical Staff, Inference Systems Company: Photon Location: Palo Alto, CA (on-site 5 days per week) Compensation: $230,000 - $350,000 + competitive equity Employment Type: Full-time Visa Sponsorship: Visa...SuggestedFull timeH1bVisa sponsorshipTransparent Search Group
Palo Alto, CA19 days ago$180k - $280k
...($3T AUM) and led enterprise ML at Goldman Sachs and Amazon. We... ...those capabilities into real systems. This is a unique opportunity... ...used in investing and shape the technical direction of our agent platform... ...at Poesis As an early team member, you’ll help shape not just...SuggestedFull timeWork at officeVisa sponsorshipWork visaRelocation package3 days per week$200k - $280k
...Capital Group ($3T AUM) and led enterprise ML at Goldman Sachs and Amazon. We're... ...risk management. We’re looking for a Member of Technical Staff (Machine Learning) to build the models... ...Build rigorous, reproducible ML research systems for fast experimentation. Turn...Full timeWork at officeVisa sponsorshipWork visaRelocation package3 days per week$200k - $300k
...OPT, H1B). About this role As a Member of Technical Staff [Product/Agent] at NeoCognition, you'll... ..., design, and scalability of systems as usage grows Collaborate across research... ...product), backend serving layer (platform), ML/LM research (training, data, evals)....Full timeH1b$180k - $280k
...Capital Group ($3T AUM) and led enterprise ML at Goldman Sachs and Amazon. We're... ...We’re looking for an exceptional Member of Technical Staff (Quantitative Finance) to work alongside... ...engineering teams on the quantitative systems behind our investment process. You’ll...Full timeWork at officeVisa sponsorshipWork visaRelocation package3 days per week$180k
...Description Job Description SpaceXAI's mission is to create AI systems that can accurately understand the universe and aid humanity in... ...track record building or optimizing large-scale distributed ML systems (training/inference optimization, GPU utilization, multi...Temporary workSpaceXAI
Palo Alto, CAa month ago$180k
...Description Job Description SpaceXAI's mission is to create AI systems that can accurately understand the universe and aid humanity in... ...-training checkpoints. BASIC QUALIFICATIONS: Expertise in ML and large model scaling, with familiarity across all kinds of...Temporary workSpaceXAI
Palo Alto, CAa month ago$200k - $240k
...get made with only a partial view of how a system will behave. Our mission is to make... ...lives somewhere other than in your head. Technical direction. Work with the AI team on what... ..., or comparable orchestration running ML workloads in production Hybrid or multi...Full time- ...architecture patterns, maintain testing standards, and mentor team members. Skills and qualifications: Experience building... ...AI-native platform turns every camera and sensor into a smart system that enhances safety, efficiency, and awareness. Founded by...
Coram AI
Sunnyvale, CA1 day ago $150k
...Member of Technical Staff Location: Palo Alto, CA Company Stage of Funding: Seed Stage AI Startup ($27M Raised) Office Type: Onsite... ...executions. Build and optimize agentic workflows, orchestration systems, and production AI runtime infrastructure. Design APIs,...Work at officeVisa sponsorshipRelocation packageRecruiting from Scratch
Palo Alto, CA5 days ago$500 per month
...Role GC AI is hiring a Staff Product Manager to own... ...actually adopted. Technically deep. You can walk an engineering... .... You understand systems design, not just user... ...Familiarity with AI/ML products and how to build... ...Work Together Team members in the San Francisco Bay...Permanent employmentFull timeCurrently hiringWork at officeImmediate startHome officeRelocation packageShift work$180k
...Job Description Job Description SpaceXAI's mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly motivated, and focused on engineering excellence. This organization is for...Temporary workWorldwideSpaceXAI
Palo Alto, CAa month ago$280k - $350k
...for strong candidates willing to relocate. About this role As a Member of Technical Staff [Research] at NeoCognition, you'll be part of the core team advancing the frontier of LLM agents: systems that can reason, plan, and act reliably in the real world. You'll...Full timeRemote workRelocationVisa sponsorshipRelocation package$200k - $300k
...markets. In this founding-level engineering role, you will own projects from initial idea through production and help develop the systems that support market research, prediction, and trading. What You'll Do Take projects from ideation through implementation and...Full timeInternshipVisa sponsorship$200k - $300k
...About the Role Join a small, founding-stage team building AI systems for financial markets. You will work across the trading stack, taking ideas from research through production and helping the team move quickly from new approaches to live experiments. What You'll...Full timeWork experience placementInternshipVisa sponsorship$200k - $300k
...experience, including demonstrated progression in one role. Experience building or evaluating production AI or machine learning systems, especially agents or LLM-powered products. Strong Python skills. Familiarity with Django, React, and TypeScript. Experience...Full timeWork at officeClera
Palo Alto, CA2 days ago$180k
...Job Description Job Description SpaceXAI's mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly motivated, and focused on engineering excellence. This organization is for...Temporary workSpaceXAI
Palo Alto, CAa month ago$200k - $300k
...including a few years in one role with demonstrated progression. Experience building or evaluating production AI or machine learning systems, especially agents or LLM-powered products. Strong Python skills and familiarity with Django, React, and TypeScript....Full timeWork at office- ...obsessed with simplicity and correctness. Must have strong opinions about how systems should work, grounded in data and deep understanding of reality. Every engineer here is deeply technical and a relentless problem-solver. We think deeply, not frantically. How We...Full time
Whop
Palo Alto, CA5 days ago - ...services/APIs) Own projects full lifecycle Design discussions & technical scoping Implementation & testing Post-launch iteration... ...ideas to intuitive, performant, scalable UX ~️ Contribute to system architecture - ensure reliable, maintainable, scalable stack...Work at officeRemote workVisa sponsorshipMonday to Friday
ProductNow
Palo Alto, CAa month ago $180k
...Job Description Job Description SpaceXAI's mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly motivated, and focused on engineering excellence. This organization is for...Temporary workWorldwideSpaceXAI
Palo Alto, CAa month ago$180k
...Job Description Job Description SpaceXAI's mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly motivated, and focused on engineering excellence. This organization is for...Temporary workSpaceXAI
Palo Alto, CAa month ago$180k
...Job Description Job Description SpaceXAI's mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly motivated, and focused on engineering excellence. This organization is for...Temporary workSpaceXAI
Palo Alto, CAa month ago$180k
...Job Description Job Description SpaceXAI's mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly motivated, and focused on engineering excellence. This organization is for...Temporary workSpaceXAI
Palo Alto, CA11 days ago$115k - $165k
...Build Championship Teams) Partner with marketers, analysts, and technical leads to challenge assumptions and align on the right solution... .... Connect the marketing technology stack to surrounding systems via REST APIs, enabling data flows that standard connectors cannot...Temporary workWork at officeWorldwide- ...Responsibilities Maintain and iterate on embodied AI data collection systems, including multimodal sensor synchronization, data pipelines,... ...on-site data collection and algorithm deployment, document technical processes, and improve collection efficiency and model...Full timeFlexible hours
XDOF
San Mateo, CA3 days ago $180k - $220k
...and develop the pipelines and tools that make our product a product. This includes accelerating our development by developing the system that enables our code to go from development to deployment. You’ll also orchestrate how data flows throughout the deployed product....Full time
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Member of Technical Staff, ML Systems. Be the first to apply!
Related searches
- machine learning Menlo Park, CA
- artificial intelligence - machine learning intern Menlo Park, CA
- machine learning research scientist Menlo Park, CA
- machine learning scientist Menlo Park, CA
- junior IT service management analyst
- junior technical business analyst
- helpdesk administrator
- technical operations analyst
- product support technician
- mri tech aide