# Member of Technical Staff, Research Engineer (Datasets)

**Company:** [Runway](https://hotfix.jobs/companies/runway)
**Location:** Remote
**Role:** Data Engineering
**Salary:** $270k – $370k/yr
**Experience:** 4+ years
**Skills:** PyTorch, JAX, Ray, Kubernetes, Machine Learning, Generative Models, Multimodal Datasets, Synthetic Data Generation, Distributed Compute, Data Pipelines
**Posted:** 2024-10-10

> Research Engineer owning datasets for training world simulation AI models, designing multimodal datasets, running experiments, and building data pipelines to enhance model capabilities across tasks like robotics and creative tools. Requires 4+ years in ML with experience in generative models and frameworks like PyTorch or JAX.

## Job Description

## What you'll do
- Design multimodal, multitask datasets that teach world models new capabilities — deciding what data to collect, generate, or curate and measuring its effect on model behavior
- Run controlled training experiments to understand how data composition drives model performance across tasks and domains
- Build and operate large-scale pipelines for synthetic data generation, filtering, and quality control
- Define evaluations and benchmarks that measure whether our models are actually improving at the things that matter
- Partner with product and creative teams to translate target behaviors and capabilities into concrete data strategies

## What you'll need
- 4+ years of experience in machine learning, bonus points for data-centric approaches
- Experience with large multimodal datasets and generative models (video, image, or multimodal)
- Deep intuition for how data composition and quality translate to model capabilities
- Comfort working across the full research stack: data analysis, dataset creation, model training, evaluation, and back again
- Proficiency with at least one ML framework (e.g. **PyTorch**, **JAX**) and distributed compute tools (e.g. **Ray**, **Kubernetes**)
- Excitement about building AI that simulates the world

## Similar roles

- [Senior Staff Software Engineer, Indexing & Retrieval Platform](https://hotfix.jobs/jobs/a96bb2ac-4943-47cb-9031-edea86dca526) - Reddit - Remote - $279k – $391k/yr
- [Staff Software Engineer, Data Platform](https://hotfix.jobs/jobs/dbec41b1-0c1c-464e-a4b5-30355952f644) - Scale AI - San Francisco, CA - $248k – $311k/yr
- [Senior Staff Data Engineer](https://hotfix.jobs/jobs/bc9c9ab5-6cc2-46fa-b946-9a0573bc3c46) - Airbnb - Remote - $248k – $310k/yr
- [Staff Data Engineer, Ads](https://hotfix.jobs/jobs/0b6db8e1-bd66-4bea-8c7f-4bf874a31460) - Discord - Remote - $248k – $279k/yr
- [Staff Data Engineer, Analytics](https://hotfix.jobs/jobs/f88dd0d9-357b-4cf7-9be7-8944ed1959e1) - Discord - San Francisco, CA - $248k – $279k/yr

**Apply:** https://hotfix.jobs/jobs/200e0ab6-1de1-439e-82a7-a4740c019e27
**Canonical:** https://hotfix.jobs/jobs/200e0ab6-1de1-439e-82a7-a4740c019e27