P

Member of Technical Staff - ML Capabilities

icon briefcase Job Type : Full Time

Number of Applicants

 : 

000+

Click to reveal the number of candidates who applied for this job.
icon loader
Apply Now
icon loader Apply Now

Let AI Supercharge Your Job Hunt!

JobCopilot scans 500,000+ company career sites daily to find jobs for you

Never miss an opportunity Save hours by auto-filling applications forms Land more interviews with tailored applications
happy man
thunder iconActivate JobCopilot

Job Description - Member of Technical Staff - ML Capabilities

About Us

Preference Model is building automated ML research engineering.

Existing frontier models are brittle when applied to real-world ML tasks. The present bottleneck is the lack of high-quality RL training environments. Our first step is to build RL environments that reflect real-world complexity, with diverse tasks and robust reward functions.

Our founding team has previous experience on Anthropic’s data team building data infrastructure, and datasets behind Claude. We are partnering with leading AI labs to push AI closer to achieving its transformative potential.

About the Role

We’re hiring experienced ML Engineers to design and build reinforcement learning environments to safely advance model capabilities specifically on machine learning research and engineering tasks to do the work of an MLE at a frontier lab.

This role blends research and engineering. It will require you to both develop novel approaches and realize them in code. Your work will include designing and implementing RL environments, conducting experiments and evaluations, delivering your work into production training runs, and collaborating with other researchers and engineers.

You will join our ML Capabilities org, a small, high-ownership team and contribute directly to the data layer that powers frontier LLM capability.

Note: This role is only for experienced ML Engineers. We have separate openings for New Grads, and for Interns.

What You Will Do:

  • Design and build RL environments and reward functions that produce clean, learnable signals for frontier models on ML research and engineering tasks.

  • Build deep expertise across the frontier of ML research, training, and inference infrastructure.

  • Collaborate with others to brainstorm and create new ideas and tools to improve the environment building process.

What We are Looking For (Qualifications):

  • You have strong ML fundamentals and broad research interests. You read many papers or tutorials, understand topics deeply and have the creativity to translate them into RLVR problems.

  • Proficiency in Python and systems programming and at least one of PyTorch or JAX

  • Problem solvers who take ownership and drives solutions end-to-end

  • Passion for staying current with the rapidly evolving ML infrastructure landscape

  • Ability to meet throughput expectations and respond quickly to feedback

About You

  • Expert knowledge in an active DL/ML research area, with publications or public code to show for it. Research experience (PhD, MS) is a big plus.

  • Deep understanding of transformer internals, training/inference of modern LLMs, experience with inference libraries (vLLM, SGLang, etc)

  • Strong expertise in kernel development (CUDA, Triton, Pallas)

  • You have built complex interactive RL environments

What We Offer:

  • Competitive cash and equity compensation (>90th percentile)

  • Ownership and autonomy in a fast moving startup environment

  • Opportunity to work with top machine learning 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.

Note: We utilize AI note-taking during our interview sessions to ensure we capture all answers and details accurately. Candidates are allowed to use AI note-takers as well, however, no other AI tools are permitted during any live interviews.

Original job Member of Technical Staff - ML Capabilities posted on GrabJobs ©. To flag any issues with this job please use the Report Job button on GrabJobs.
Apply Now
Share Job
Share Job

Auto-Apply to ML Engineer Jobs with your AI JobCopilot

thunder icon Auto-Apply with AI

Similar ML Engineer Jobs in the US

GrabJobs is the no1 job portal in the US, connecting you to thousands of jobs fast! Find the best jobs in the US, apply in 1 click and get a job today!

Mobile Apps

Copyright © 2026 Grabjobs Pte.Ltd. All Rights Reserved.