Latest AI JobsBrowse Jobs
P

Member of Technical Staff - ML Infrastructure Engineer, Post-training

Preference-ModelSan FranciscoEngineering
Apply on company site
Salary
$200K – $350K
Job type
Full-time
Experience
Senior
Posted
11 Sept 2026
Status
Open · checked 7d ago

About this role

This role suits experienced software engineers who want to build the compute and data systems that power large language model post-training in San Francisco. Preference-Model is hiring a Member of Technical Staff for ML Infrastructure in San Francisco to scale the infrastructure for in-house reinforcement learning environments.

Role Overview

Preference-Model seeks a Senior ML Infrastructure Engineer to design systems that support post-training research on large language models. The position focuses on creating scalable compute, scheduling, and data pipelines to accelerate reproducible experiments.

Key Responsibilities

Design, build, and scale compute, scheduling, and data infrastructure supporting post-training research within in-house reinforcement learning environments
Develop core machine learning framework primitives and internal tooling to accelerate experimentation and reduce the time from concept to result
Construct evaluation, benchmarking, monitoring, logging, debugging, automated testing, and deployment systems to ensure reliability at scale
Collaborate with Research Engineers to transform experimental requirements into functional infrastructure specifications and respond rapidly to feedback

Requirements

Strong software engineering fundamentals with practical experience building production-grade large language model inference and training infrastructure from the ground up
Familiarity with training and inference internals such as transformers, distributed training, and libraries including vLLM, SGLang, and Megatron
Experience with reinforcement learning training frameworks like Slime, veRL, Ray Train, and SkyRL
Knowledge of distributed systems principles, cloud platforms such as AWS and GCP, container orchestration with Kubernetes, and high-throughput low-latency workloads
Background in data engineering tools and the construction of scalable data pipelines
Proficiency in core machine learning frameworks like PyTorch or JAX
Ability to balance production rigor with fast-paced research while communicating tradeoffs effectively

Skills and Tools

PyTorch
JAX
vLLM
SGLang
Megatron
Slime
veRL
Ray Train
SkyRL
AWS
GCP
Kubernetes

Salary and Benefits

Compensation range between $200K and $350K with equity and performance-based bonus opportunities
Health, vision, and dental benefits alongside a 401K match
Onsite daily lunch and weekly snack provisions
Visa sponsorship and relocation support

About the Company

Preference-Model is a superintelligence data research company backed by $16M in seed funding led by a16z. The organization builds reinforcement learning environments to train capable and aligned artificial intelligence models.

How to Apply

Interested candidates should apply on the official company careers page.

Skills and tech stack

Questions candidates ask

What is the salary range for this role?

The salary ranges from $200K to $350K, with offers including equity and a performance-based bonus.

Where is the position located?

The position is located in San Francisco.

Is visa sponsorship available?

Yes, visa sponsorship and relocation support are available.

What frameworks and tools are required?

Required tools and frameworks include PyTorch, JAX, vLLM, SGLang, Megatron, Slime, veRL, Ray Train, SkyRL, AWS, GCP, and Kubernetes.

What type of company is Preference-Model?

Preference-Model is a superintelligence data research company that builds reinforcement learning environments for training aligned artificial intelligence models.

Ready to apply?

The application is on the official Preference-Model careers page.

Apply on company site