Member of Technical Staff - AI/ML Infrastructure Engineer, RL Post-training

Preference Model
San Francisco, CA
On-site

Who this role is best for

Best suited to candidates with strong software engineering and distributed systems experience working in AI/ML infrastructure and RL environments.

Best fit for

  • Candidates with experience in building LLM training and inference systems from the ground up
    — “building production-grade LLM inference and training infrastructure (ideally from the ground up)”
  • Individuals who have worked with RL training frameworks and can translate research needs into infrastructure solutions
    — “Partner directly with Research Engineers to translate research needs into infrastructure requirements”
  • Engineers with a background in distributed systems and cloud platforms, capable of handling high-throughput workloads
    — “experience with cloud platforms (AWS, GCP) and container orchestration (Kubernetes), building systems for high-throughput, low-latency workloads”

Things to consider

  • The role demands a balance between production-grade rigor and adapting to fast-moving research timelines
    — “Can balance production rigor with the pace of fast-moving research”
  • Candidates should be prepared to work in a startup environment with high autonomy and ownership
    — “Ownership and autonomy in a fast moving startup environment”

How to stand out

  • Highlight experience in building internal tooling that accelerates reproducible experimentation
    — “accelerating reproducible experimentation and reducing time from idea to result”
  • Emphasize hands-on contributions to distributed training and inference systems
    — “experience building LLM training/inference internals such as transformers, distributed training”
  • Showcase skills in developing evaluation and benchmarking systems for large-scale ML workloads
    — “Build evaluation and benchmarking infrastructure, monitoring, logging, and debugging tooling”
Pace · Fast PacedCollaboration · HighAutonomy · MediumDecision Impact · Team

Derived from job-description analysis by Serendipath's career intelligence engine.

What success looks like

  • build scalable infrastructure for high-throughput systems
  • accelerate reproducible experimentation
  • reduce time from idea to result
  • ship fast in response to researcher feedback
Typical background
experience building production-grade LLM inference and training infrastructurehands-on experience with cloud platforms and container orchestrationproficiency in core ML frameworks

Skills & requirements

Required

Software Engineering FundamentalsLLM Inference And Training InfrastructureDistributed Systems PrinciplesCloud Platforms (aws, GCP)Container Orchestration (kubernetes)Data Engineering ToolsCore ML Frameworks (pytorch, JAX)

Preferred

RL Training Frameworks (slime, Verl, Ray Train, Skyrl, Miles)

Stack & domain

PythonReactLLM Inference And Training InfrastructureTransformersDistributed TrainingInference Libraries Like Vllm, Sglang, MegatronRL Training Frameworks Like Slime, Verl, Ray Train, Skyrl, MilesDistributed Systems PrinciplesCloud Platforms (aws, GCP)Container Orchestration (kubernetes)Data Engineering ToolsBuilding Robust, Scalable Data PipelinesCore ML Frameworks Such As Pytorch Or JAXLeadershipCommunicationFinanceHealthcare

About the role

Original posting from Preference Model via Ashby

ABOUT US

Preference Model is a superintelligence data research company. We build RL environments for training capable, better-aligned superintelligences.

Almost every aspect of how a model behaves is shaped by the reward signals it's trained on. The biggest problem with AIs today is that they don't always do what we intend. Sometimes that's because they aren't capable enough, and sometimes it's because they aren't aligned. Both problems come down to the training objectives.

We're trying to address both of these issues in the most direct, highest leverage way available to us, which today means researching how to build RL environments better, not just for the AIs of today but for the AIs as they will be when they outsmart us.

Over the past year, we've built RL environments for several frontier labs, and we're backed by $16M in seed funding led by a16z. Our founding team has previous experience on Anthropic’s data team building data infrastructure, and datasets behind Claude.

ABOUT THE ROLE

Frontier research moves only as fast as its infrastructure permits. Building solid infrastructure is foundational to our mission of pushing self-directed learning as far as it can go.

We are looking for Senior ML Infrastructure Engineers to build the infrastructure and systems that power the frontier of post-training on large language models. This role involves building scalable infrastructure to enable high-throughput systems and shape how our research is run, bringing us closer to models that can train themselves on what they aren't yet good at.

WHAT YOU WILL DO

  • Design, build, and scale the compute, scheduling, and data infrastructure that powers post-training research on our in-house RL environments
  • Develop and maintain core ML framework primitives and internal tooling that researchers rely on daily, accelerating reproducible experimentation and reducing time from idea to result
  • Build evaluation and benchmarking infrastructure, monitoring, logging, and debugging tooling, and automated testing and deployment systems, so failures are caught early and infrastructure stays reliable as it scales
  • Partner directly with Research Engineers to translate research needs into infrastructure requirements, and ship fast in response to their feedback

WHAT WE ARE LOOKING FOR

  • Strong software engineering fundamentals and hands-on experience building production-grade LLM inference and training infrastructure (ideally from the ground up)
  • Experience building LLM training/inference internals such as transformers, distributed training, and working on inference libraries like vLLM, SGLang, Megatron
  • Experience working on RL training frameworks like Slime, veRL, Ray Train, SkyRL, Miles
  • Significant experience and understanding of distributed systems principles, and have hands-on experience with cloud platforms (AWS, GCP) and container orchestration (Kubernetes), building systems for high-throughput, low-latency workloads
  • Have experience with data engineering tools and building robust, scalable data pipelines
  • Proficiency in core ML frameworks such as PyTorch or JAX
  • Can balance production rigor with the pace of fast-moving research, and communicate infrastructure tradeoffs clearly to researchers who aren't infra specialists

WHAT WE OFFER:

  • Competitive cash and equity compensation (>90th percentile)
  • Ownership and autonomy in a fast moving startup environment
  • Opportunity to work alongside senior and staff engineers from frontier labs and infrastructure companies, plus top ML engineers
  • Health, vision, dental, benefits
  • 401K match
  • Lunch provided everyday onsite
  • Weekly snack orders
  • Visa sponsorship & relocation support available

We value diverse perspectives and experiences. If you're excited about this role but don't check every box, we still encourage you to apply.

Source: Preference Model careers (Ashby)

Similar roles