Senior ML Platform Engineer — Automate RL Environments

Preference Model

San Francisco (CA)

On-site

USD 180,000 - 260,000

Full time

12 days ago

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Benefits offered by this job

Competitive cash and equity
Ownership and autonomy
Health, vision, dental
401K match
Lunch onsite
Visa sponsorship & relocation

Job summary

Preference Model is seeking a senior ML Platform Engineer in San Francisco to own and scale the environments that power our RL research stack. You will lead integrations, packaging, and automated validation to ensure labs receive reliable environments for training capable, aligned superintelligences.

You will build and maintain Karotte, our open-source framework, and manage the platform that runs environments on CPU, GPU, and TPU while delivering them to partner labs with minimal hand work.

Qualifications

  • 5+ years of experience building and owning ML Platform Engineering, ideally AI related ML Platform Engineering.
  • Strong software engineering fundamentals and a track record of building and operating production infrastructure or developer platforms that other engineers depend on.
  • Hands-on experience with containers and orchestration (Docker, Kubernetes, or similar) and at least one major cloud platform (GCP, AWS, or similar).
  • A working understanding of how to run untrusted code safely: isolation, resource limits, and what happens when a process fills the disk or refuses to exit.
  • You've owned something tedious end to end, such as a migration, a release process, or a delivery pipeline, and automated away as much of it as you could.
  • Extensive hands-on use of coding agents (Claude Code, Cursor, Codex, or similar), and an intuition for where models cut corners.
  • Proficiency in Python, including writing libraries and APIs that other people build on.

Responsibilities

  • Own the integrations that deliver our environments to partner labs: build container images, match each lab's submission format, and automate the checks that confirm requirements are met before anything ships.
  • Build and maintain Karotte, our open-source environment framework, from API design through releases and migrations that keep existing environments working.
  • Run and scale the infrastructure that builds, runs, and grades environments, including containerized rollouts and grading across CPU, GPU, and TPU, VM-based sandboxing, and the resource limits and isolation that stop a model from gaming its own grader.
  • Build the internal tools environment builders use every day (CLI, docs, data catalog, access controls, and run monitoring) so they spend their time on environment design instead of infrastructure.
  • Partner with environment builders and post-training researchers to turn recurring pain points into platform features, and ship quickly in response to their feedback.

Skills

Python
ML Platform Engineering
Containers
Kubernetes
Cloud Platforms (GCP / AWS)

Tools

Docker
Kubernetes
GCP
AWS

Job description

Preference Model is seeking a senior ML Platform Engineer in San Francisco to own and scale the environments that power our RL research stack. You will lead integrations, packaging, and automated validation to ensure labs receive reliable environments for training capable, aligned superintelligences.

You will build and maintain Karotte, our open-source framework, and manage the platform that runs environments on CPU, GPU, and TPU while delivering them to partner labs with minimal hand work.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Senior ML Infra Engineer: Scale High-Throughput AI Research
Senior ML Infra Engineer: Scale High-Throughput AI Research

Preference Model • San Francisco (CA)

On-site
USD 180,000 - 260,000
Founding ML Engineer — RL Environments for Hardware AI
Founding ML Engineer — RL Environments for Hardware AI

Rnb Consultancy • San Francisco (CA)

On-site
USD 150,000 - 250,000
Equity
Founding role
Member of Technical Staff - ML Platform Engineer
Member of Technical Staff - ML Platform Engineer

Preference Model • San Francisco (CA)

On-site
USD 180,000 - 260,000
Competitive cash and equity
Ownership and autonomy
Health, vision, dental
+3
Staff RL Environments Engineer
Staff RL Environments Engineer

Scale AI, Inc. • San Francisco (CA)

On-site
USD 252,000 - 315,000
Health coverage
Dental coverage
Vision coverage
+4
Staff Software Engineer, RL Environments & Platform
Staff Software Engineer, RL Environments & Platform

Scale AI • New York (NY)

On-site
USD 252,000 - 315,000
Health, dental, vision coverage
Retirement benefits
Learning and development stipend
+2
Staff Software Engineer — RL Environment Platform
Staff Software Engineer — RL Environment Platform

Anthropic • New York (NY)

Hybrid
USD 405,000 - 605,000
Senior ML Platform Systems Engineer
Senior ML Platform Systems Engineer

Cursor • Palo Alto (CA)

On-site
USD 180,000 - 240,000
Staff ML Systems Engineer — Enterprise RL & Environments
Staff ML Systems Engineer — Enterprise RL & Environments

Ambral Labs • United States

On-site
USD 165,000 - 325,000
Equity ownership
Equinox membership
Free meals, coffee, and snacks
+2
Staff RL Environments Engineer: End-to-End Systems
Staff RL Environments Engineer: End-to-End Systems

Scaleai • San Francisco (CA), New York (NY)

On-site
USD 252,000 - 315,000
Health insurance
Dental & vision coverage
:Learning stipend
ML Engineer - RL Environments & Simulation Infrastructure
ML Engineer - RL Environments & Simulation Infrastructure

DeepRec.ai • San Francisco (CA)

On-site
USD 150,000 - 220,000
Equity