Senior ML Engineer - RL Environments for Frontier LLMs

Preference Model; Seattle (WA); On-site; USD 200,000 - 350,000; Full time

Benefits offered by this job

Health, vision, dental

401K match

Lunch onsite

Weekly snacks

Visa sponsorship

Relocation support

Job summary

Preference Model is hiring experienced Machine Learning Engineers to design and build reinforcement learning environments for advancing frontier models in ML research and engineering. You will own and implement RL environments, reward functions, with end-to-end experimentation and production integration.

This role blends research and engineering; you will stay current with the latest ML research, contribute novel approaches, and collaborate with researchers and engineers.

Qualifications

• 5+ years of experience in machine learning or research, focused on LLMs and transformer models.
• Strong ML fundamentals, broad research interests, ability to translate papers into RL/VR problems.
• Proficiency in Python and systems programming; at least one of PyTorch or JAX.

Responsibilities

• Design and build RL environments and reward functions for frontier models.
• Develop expertise across ML research, training, and inference infrastructure.
• Collaborate to brainstorm and create tools to improve environment building processes.

Skills

5+ years ML/research

LLMs/transformers

Python

PyTorch

JAX

Systems programming

CUDA

Triton

Pallas

Education

PhD or MS in ML/CS

Tools

Python

PyTorch

JAX

CUDA

Triton

Pallas

Job description

Preference Model is hiring experienced Machine Learning Engineers to design and build reinforcement learning environments for advancing frontier models in ML research and engineering. You will own and implement RL environments, reward functions, with end-to-end experimentation and production integration.

This role blends research and engineering; you will stay current with the latest ML research, contribute novel approaches, and collaborate with researchers and engineers.