ML infrastructure engineer
- Discuss this with your agent
- Open in Claude
- Open in ChatGPT
Prepare application
- Draft this with your agent
- Open in Claude
- Open in ChatGPT
Role details
Tech stack
+4 more
Job description
Weâre looking for an ML infrastructure engineer to own model evaluation at Runway, end to end. Every decision we make about a model - which checkpoint to keep training, what to ship to millions of users, which datamix and architecture shows the most promise - rests on evals. Today that work is spread across the organization. Youâll turn it into one platform.
Your job will be to design and build the systems that generate samples at scale, score them with automated metrics and human annotations, track results across checkpoints and releases, and put the answers in front of researchers in minutes rather than days. Youâll define what âbetterâ means operationally - how we measure it, how confident we are, and how a result becomes a ship/no-ship decision.
Youâll be embedded within research teams as a member of ML Platform, and youâll set the technical direction for evals across the company. This is an extremely high-leverage role: the quality of our models is bounded by how well we can measure them.
A peek at our technical stack
Our inference and training code is written in Python (PyTorch) and is cloud-native. We leverage multiple clouds and use a combination of Kubernetes native and home-grown tooling for efficient job orchestration.
Weâre big users of agentic development and operations. We have a robust internal platform and provide broad access to the latest models and harnesses.
Youâll have access to best-in-class models, agents, GPUs, storage and cloud services you need to drive a world class evaluation pipeline that helps us deliver frontier models.
What youâll do
- Own the evaluation platform end to end: the tooling and systems required to generate, annotate, review and adapt at frontier scale
- Define the CLIs, APIs, GUIs and storage layers required to make this process seamless, fast, sophisticated and collaborative
- Work directly with research teams on video, image, audio, agents, and robotics to understand what they need to measure and build it, then generalize the result into the platform
- Help set the standards for how we evaluate models at Runway: reproducibility, metric definitions, reporting formats, and when a result is trustworthy enough to act on
- Support broad adoption of the platform and its integration throughout Runwayâs research efforts across training, production model serving and more
- Contribute broadly as a member of the ML Platform team to tools and systems that help Runway train and serve frontier models
Requirements
- 5+ years of experience building ML infrastructure or data platforms in production environments, with at least some of that time spent on evaluation, experimentation, or benchmarking systems
- Strong Python and PyTorch, and hands-on experience running large batch GPU workloads on Kubernetes
- Experience designing data pipelines and storage for large volumes of media or model outputs, with attention to versioning and reproducibility
- Familiarity with experimental statistics: paired comparisons, confidence intervals, multiple-comparison pitfalls, inter-rater agreement
- Comfort building internal tools end to end, from the command line to the browser
- Ability to lead a broad technical area: gather requirements from many teams, set direction, make tradeoffs, and drive a roadmap without waiting to be told
- Familiarity with the full model development lifecycle: data, training, evaluation, serving
- Self-starter who can work embedded with research teams and move fast
- Strong systems thinking and pragmatic approach to production reliability
- Humility and open mindedness; at Runway we love to learn from one another
Nice to have
- Track record of building evaluation suites for generative models
- Hands-on work with LLM- or VLM-as-judge pipelines
- Experience with online experimentation platforms and connecting offline metrics to product outcomes
- Prior work evaluating agents or robotics policies
Benefits & conditions
Runway strives to recruit and retain exceptional talent from diverse backgrounds while ensuring pay equity for our team. Our salary ranges are based on competitive market rates for our size, stage and industry, and salary is just one part of the overall compensation package we provide.
About the company
We are building AI to simulate the world through merging art and science.
We believe that world models are at the frontier of progress in artificial intelligence. Language models alone wonât solve the worldâs hardest problems - robotics, disease, scientific discovery. Real progress requires models that experience the world and learn from their mistakes, the same way that humans do. And this kind of trial and error can be massively accelerated when done in simulation, rather than in the real world.
World models offer the most clear path to general-purpose simulation, changing how stories are told, how scientific progress is made and how the next frontiers of humanity are reached.
Our team consists of creative, open minded, caring and ambitious people who are determined to change the world. We aspire to continuously build impossible things and our ability to do so relies on building an incredible team. If you are driven to do the same, weâd love to hear from you., There are many factors that go into salary determinations, including relevant experience, skill level and qualifications assessed during the interview process, and maintaining internal equity with peers on the team. The range shared below is a general expectation for the function as posted, but we are also open to considering candidates who may be more or less experienced than outlined in the job description. In this case, we will communicate any updates in the expected salary range.
Apply for this position
This job is hosted externally. Click below to view the full posting and apply.
Prepare application
- Draft this with your agent
- Open in Claude
- Open in ChatGPT
Good distractions
Talks and stories from around this role â technically off-topic, practically not.
Moments
Explore playlistsVideos
See allRelated articles
See all
What Are Large Language Models?
I Gave a Video Editor More Autonomy Than a Trading Bot. On Purpose.
How to Become an AI Engineer
MLOps â Whatâs the deal behind it?