ML infrastructure engineer

The Runway
New York, NY, United States
9 days ago
Apply on startup.jobs
Prepare application

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Experience level
Expert
Experience required
5 years minimum
Working hours
Regular working hours
Job source

Tech stack

Application Programming Interfaces (APIs) Command-Line Interface Cloud Engineering Java GUIs Python (Programming Language) Cloud Services Azure Machine Learning Management of Software Versions Graphics Processing Unit (GPU) Pytorch Large Language Models Model Validation
+4 more
Kubernetes Data Management Machine Learning Operations Data Pipelines

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.

Apply on startup.jobs
Prepare application

Good distractions

Talks and stories from around this role — technically off-topic, practically not.

1:15 min

Deploying local container pods to Kubernetes clusters

Stevan Le Meur Stevan Le Meur ¡ World Congress 2024

5:02 min

Developing terminal user interfaces natively within Java scripts

Jens Knipper Jens Knipper ¡ Europe 2026 Virtual

2:35 min

Preventing remote code execution in PyTorch models

Balåzs Kiss ¡ World Congress 2023

2:08 min

Applying large language models to infrastructure tasks

Alfonso Sandoval Rosas Alfonso Sandoval Rosas ¡ Europe 2026 Virtual

33 sec

Supporting NVIDIA GB200 GPUs on Kubernetes

Kevin Klues Kevin Klues ¡ World Congress 2025

1:34 min

Pivoting careers into specialized platform engineering roles

Xavier Portilla Edo ¡ LIVE

Videos

See all

Related articles

See all