Startups · AI

Member of Technical Staff - Low Level & Kernels Capabilities

Preference Model Labs · San Francisco · On-site

← All jobs
About Preference Model Labs

Preference Model is building the next generation of training data to power the future of AI. Backed by a16z and South Park Commons.

About the role

We’re hiring experienced Machine Learning Engineers for our Low Level / Kernels Capabilities team. The Kernels team builds reinforcement learning (RL) environments at the lowest layers of the stack. Think GPU and accelerator kernels, vector ISAs, codec and crypto primitives, FPGA work, and more. These are the domains where frontier models are weakest, niche paradigms, hardware underrepresented in training data, and open benchmarks that show models lagging.

What they're looking for

  • Strong low-level/systems engineering: fluent in C / C++ / CUDA (or an equivalent kernel language), comfortable dropping to assembly when it matters
  • Strong, engineering-quality Python across your prior work, writing production code, automation and deployment scripts, data analysis and plotting (not notebook-only)
  • Hardware-aware coding: you write with the silicon in mind, considering memory hierarchy, occupancy, data movement, parallelism, latency vs throughput etc
  • Kernel development experience: you write kernels and optimize them iteratively against a profiler
  • An adversarial mindset: you turn fuzzy goals into robust, ungameable scoring, and you ask "how would a model cheat this?"
  • Hands-on work with LLMs
More about this role

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.

We’re hiring experienced Machine Learning Engineers for our Low Level / Kernels Capabilities team. The Kernels team builds reinforcement learning (RL) environments at the lowest layers of the stack. Think GPU and accelerator kernels, vector ISAs, codec and crypto primitives, FPGA work, and more. These are the domains where frontier models are weakest, niche paradigms, hardware underrepresented in training data, and open benchmarks that show models lagging.

This role blends research and engineering . It will require you to both develop novel approaches and realize them in code. You will own environments end-to-end : choose...

Read the full posting on Preference Model Labs's site ↗

Engineering

Build your edge while you search

Free tools for founders and investors, plus VC Unfiltered, our take on startups, venture and the people who build them.