Summary
✨ AI‑Generated
A senior ML infrastructure engineer is needed to build scalable systems powering frontier post-training research for large language models. You will design robust infrastructure and research environments that support reinforcement learning, complex real-world tasks, and high-quality reward mechanisms.
Highlights
Build foundational infrastructure for frontier machine-learning research, enabling scalable post-training of large language models and sophisticated reinforcement-learning environments.
Description
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
Frontier research moves only as fast as its infrastructure permits.
Building solid infrastructure is foundational to our mission of pushing self-directed learning as far as it can go.
We are looking for Senior ML Infrastructure Engineers to build the infrastructure and systems that power the frontier of post-training on large language models.
This role involves building scalable infrastructure to enable high-throughput systems and shape how our research is run, bringing us closer to models that can train themselves on what they aren't yet good at.
What You Will Do
Design, build, and scale the compute, scheduling, and data infrastructure that powers post-training research on our in-house RL environmentsDevelop and maintain core ML framework primitives and internal tooling that researchers rely on daily, accelerating reproducible experimentation and reducing time from idea to resultBuild evaluation and benchmarking infrastructure, monitoring, logging, and debugging tooling, and automated testing and deployment systems, so failures are caught early and infrastructure stays reliable as it scalesPartner directly with Research Engineers to translate research needs into infrastructure requirements, and ship fast in response to their feedback
What We are Looking For
Strong software engineering fundamentals and hands-on experience building production-grade LLM inference and training infrastructure (ideally from the ground up)Experience building LLM training/inference internals such as transformers, distributed training, and working on inference libraries like vLLM, SGLang, MegatronExperience working on RL training frameworks like Slime, veRL, Ray Train, SkyRLSignificant experience and understanding of distributed systems principles, and have hands-on experience with cloud platforms (AWS, GCP) and container orchestration (Kubernetes), building systems for high-throughput, low-latency workloadsHave experience with data engineering tools and building robust, scalable data pipelinesProficiency in core ML frameworks such as PyTorch or JAXCan balance production rigor with the pace of fast-moving research, and communicate infrastructure tradeoffs clearly to researchers who aren't infra specialists
What we offer:
Competitive cash and equity compensation (>90th percentile)Ownership and autonomy in a fast moving startup environmentOpportunity to work alongside senior and staff engineers from frontier labs and infrastructure companies, plus top ML engineersHealth, vision, dental, benefits401K matchLunch provided everyday onsiteWeekly snack ordersVisa 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.
Compensation Range: $200K - $350K
