Latest AI JobsBrowse Jobs
T

Member of Technical Staff - ML Infra

trajectorySan FranciscoMember of Technical Staff
Apply on company site
Salary
Not listed
Job type
Full-time
Experience
Senior
Posted
6 Oct 2026
Status
Open · checked 9d ago

About this role

This position suits engineers with deep expertise in training infrastructure, inference systems, or GPU kernels who want to build foundational platforms for AI systems in San Francisco. The Member of Technical Staff role at Trajectory focuses on creating infrastructure for artificial intelligence systems designed to self-improve across training, inference, and kernels.

Role Overview

Trajectory seeks a Member of Technical Staff for ML Infrastructure located in San Francisco. The professional in this position will construct systems that enable AI to enhance training, inference, and kernels while targeting monthly improvements across GPU scale, model size, throughput, memory efficiency, caching, and latency. Initial ownership will align with individual strengths in training, inference, or kernels.

Key Responsibilities

Construct and refine distributed training and reinforcement learning infrastructure, encompassing rollout execution, GPU scheduling, checkpointing, and recovery.
Enhance serving operations for production agents and training rollouts by managing batching, scheduling, and KV-cache utilization.
Create and refine GPU kernels and runtimes to boost memory utilization and execution efficiency while maintaining numerical correctness.
Establish reproducible benchmarks, observability tools, and automated research workflows to test and validate performance enhancements.
Collaborate alongside researchers to transform new algorithms into dependable systems spanning training, inference, and kernels.

Requirements

Strong foundational knowledge in distributed systems, networking, storage, and failure recovery alongside practical experience shipping demanding systems.
Deep specialization in training infrastructure, inference systems, or GPU kernels backed by demonstrable systems or measurable optimizations.
Proficiency in Python alongside relevant specialization languages such as C++, CUDA, or Triton.
Comprehension of framework internals like PyTorch or JAX paired with profiling and debugging abilities.
Clear communication skills and ownership to collaborate with researchers and deliver measurable performance gains.

Nice to Have

Practical background with training and reinforcement learning stacks including Miles, SkyRL, Prime Intellect's verifiers, or comparable platforms.
Familiarity with vLLM, SGLang, collective communication, or machine learning compilers depending on the area of specialization.

Skills and Tools

Python
C++
CUDA
Triton
PyTorch
JAX
vLLM
SGLang

About the Company

Trajectory operates as a research and product lab that builds a platform for continual learning. The organization provides a scalable approach enabling companies to capture product usage signals and continually post-train large-scale agentic models. The research and product teams bring experience from organizations such as DeepMind, OpenAI, Meta Superintelligence, Figma, Apple, Stripe, and Windsurf, and the company has secured funding from Sequoia, Conviction, Jeff Dean, and Fei Fei Li.

How to Apply

Interested candidates should visit the official company careers page to submit an application.

Skills and tech stack

PythonC++CUDATritonPyTorchJAXvLLMSGLangDistributed Systems

Questions candidates ask

Where is the role located?

The position is based in San Francisco.

What core technical areas does the role cover?

The role spans training, inference, and kernels, with initial ownership shaped around individual strengths.

What programming languages and tools are required?

Strong Python skills are required, along with languages relevant to the applicant's specialty such as C++, CUDA, or Triton, plus understanding of PyTorch or JAX internals.

What background does the company's team have?

The research team comes from DeepMind, OpenAI, and Meta Superintelligence, while the product team includes experience from Figma, Apple, Stripe, and Windsurf.

Ready to apply?

The application is on the official trajectory careers page.

Apply on company site