Skip to content
RunwayRunwayUnited States

Member of Technical Staff, Research Engineer (Datasets)

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.

270k – 370k/yr
Remote4+ YOEData Engineering

About the role

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

Skills

PyTorchJAXRayKubernetesMachine LearningGenerative ModelsMultimodal DatasetsSynthetic Data GenerationDistributed ComputeData Pipelines
Reddit

Senior Staff Software Engineer, Indexing & Retrieval Platform

RedditUnited States

Leads development and scaling of ML indexing and retrieval infrastructure for Reddit's recommendation systems, integrating lexical/vector search and GenAI. Requires 10+ years in software engineering with 3+ in technical leadership, expertise in distributed systems, stream processing, and cloud-native architectures.

279k – 391k/yr
Remote10+ YOEData Engineering
Scale AI

Staff Software Engineer, Data Platform

Scale AISan Francisco, CA +2

Leads architecture and development of large-scale data platforms for AI, including storage, streaming, caching, and indexing. Requires 8+ years experience with databases, streaming tools, Kubernetes, and distributed systems.

248k – 311k/yr
Hybrid8+ YOEData Engineering
Airbnb

Senior Staff Data Engineer

AirbnbUnited States

Lead multi-year vision and architecture for data governance and quality at scale. Define best practices, tooling, and culture while coaching senior engineers and influencing executive strategy on compliance and data integrity.

248k – 310k/yr
Remote12+ YOEData Engineering
Discord

Staff Data Engineer, Ads

DiscordUnited States

Discord is seeking a Staff Data Engineer to lead technical vision and strategy for ads data infrastructure. This role involves building and maintaining sophisticated data pipelines, datasets, and analytical tools, and mentoring other engineers.

248k – 279k/yr
Remote7+ YOEData Engineering
Discord

Staff Data Engineer, Analytics

DiscordSan Francisco, CA

Leads technical strategy for analytics data infrastructure, owns metric definitions, builds curated datasets and data models using SQL/Python, implements quality monitoring, and collaborates cross-functionally. Requires 7+ years in analytics/data engineering.

248k – 279k/yr
On-site7+ YOEData Engineering