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
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
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
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
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
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.