P

Preference Model

Reinforcement learning environments for frontier AI

Machine Learning Engineer New Grad - Machine Learning Capabilities

AI ResearchConfirmed live in the last 24 hours
Full-timeSan Francisco, CA, USA

About the Job

Requirements

  • Strong machine learning fundamentals and broad research interests, with the ability to understand research topics deeply and translate them into reinforcement learning with verifiable rewards problems.
  • Expert knowledge in an active deep learning or machine learning research area, demonstrated by publications or public code.
  • Deep understanding of transformer internals.
  • Proficiency in Python, NumPy, and systems programming.
  • Ability to take ownership and drive solutions end-to-end.
  • Ability to meet throughput expectations and respond quickly to feedback.

Responsibilities

  • Design and build reinforcement learning environments and reward schemes that produce clean, learnable signals for frontier models working on machine learning research and engineering tasks.
  • Build expertise across frontier machine learning research, training, and inference infrastructure.
  • Collaborate with others to brainstorm and create ideas and tools that improve the environment-building process.
  • Stay up to date with the latest research, develop novel approaches, and implement them in code.
  • Conduct experiments and evaluations, deliver work into production training runs, and collaborate with researchers and engineers.

Desired Qualifications

  • Research experience, including a PhD, is strongly preferred.
  • Experience with PyTorch or JAX is preferred.
  • Expertise in kernel development using CUDA, Triton, or Pallas, including optimizing non-trivial neural modules for specific hardware.
  • Research projects, coursework, or personal work involving reinforcement learning environments in any framework or at any scale.
  • Open-source contributions to machine learning infrastructure or reinforcement learning tooling.
  • Experience with a cloud platform such as AWS, GCP, or Azure, or infrastructure-as-code tools.

About the Company

Preference Model

Reinforcement learning environments for frontier AI

Preference Model builds reinforcement learning environments and harnesses to train frontier AI models, focusing on how reward signals shape model behavior. It designs robust RL harnesses and simulators that treat reward functions and graders as an attack surface, testing them across millions of training rollouts to harden against loopholes and gaming, with fail-safes for safety. Unlike others that only build models or tools, it targets the reward design and alignment bottlenecks, drawing talent from Anthropic, Stripe, Google DeepMind, and others, and backed by notable investors. Its goal is to broaden access to capable AI while preserving strong safeguards to keep frontier systems safe.

Company size
11-50
Company stage
Seed
Total funding
$16M
Headquarters
San Francisco, CA
Founded
2025

Recently Posted Jobs

Explore more opportunities at Preference Model

See more jobs