Skip to content
World LabsWorld LabsSan Francisco, CA

Research Platform Engineer

Builds mission-critical training, data, and inference infrastructure for AI research on large world models. Optimizes performance, debugs distributed systems, and improves researcher velocity with 5+ years experience in ML systems and strong Python proficiency.

Salary not listed
On-site5+ YOEML Engineering

About the role

What You Will Do

  • Design and build training infrastructure, data infrastructure, and data processing and sourcing pipelines.
  • Productionize models for serving and own parts of the inference stack.
  • Build internal tools and services that increase engineering and research velocity.
  • Debug hard problems across training, inference, and performance — including distributed systems issues under real research workloads.
  • Optimize throughput, latency, GPU utilization, and system-level scaling.
  • Improve research iteration speed and developer experience — cut debugging time, raise reliability, and make it faster for researchers to ship experiments.
  • Raise the engineering bar across research and platform code alike.

Key Qualifications

  • 5+ years of experience building and shipping production systems, with demonstrated ownership of infrastructure used by other engineers or researchers.
  • Strong depth in at least one of: ML infrastructure, distributed training or inference systems, data systems, or research tooling.
  • Strong distributed systems foundations — concurrency, consistency tradeoffs, replication, failure modes, and scaling behavior under real workloads.
  • Strong performance optimization skills across at least one of: training throughput, inference latency, GPU utilization, or system-level scaling.
  • Strong proficiency in Python, with the ability to work in C++, CUDA, Rust, or Go as the work demands.
  • Experience working directly with ML researchers or research engineers, including productionizing research code.
  • A product engineer's instincts for iteration speed and developer experience — applied to the systems researchers use every day.
  • Strong judgment about what to build and what to leave alone, particularly when research requirements are ambiguous or shifting.
  • High-ownership mindset; you measure yourself by outcomes shipped, not by tickets closed.

Preferred Qualifications

  • Experience at an AI lab or ML-native company, working on systems used directly by researchers.
  • Experience with large-scale training or inference systems — GPU orchestration, distributed training, or high-throughput inference.
  • Experience with low-level performance optimization — profiling, kernel-level tuning, memory and bandwidth optimization, distributed communication primitives.
  • Experience building developer experience tooling for research — notebooks, experiment tracking, reproducibility infrastructure.
  • Experience in early-stage or high-growth environments where scope and priorities shift frequently.

Skills

PythonC++CUDARustGoDistributed SystemsML InfrastructureGpu UtilizationDistributed TrainingInference Systems

Similar roles

ML Engineering jobs
Cerebras Systems

CoDesign & NextGen Performance Engineer

Cerebras SystemsSunnyvale, CA

Characterize, analyze, and optimize performance of state-of-the-art AI models on Cerebras' wafer-scale hardware. Build performance models, optimize kernels and compilers, debug runtime behavior, and develop visualization tools to influence next-gen AI architecture.

Salary not listed
On-site3+ YOEML Engineering
OpenAI

Research Engineer, Privacy

OpenAISan Francisco, CA

Research Engineer on OpenAI's Privacy team designing and prototyping privacy-preserving ML algorithms like differential privacy and federated learning at scale. Requires hands-on PETs experience, fluency in PyTorch/JAX, and a track record implementing or publishing novel privacy work.

380k – 445k/yr
HybridML Engineering
Console

Research Engineer

ConsoleSan Francisco, CA

Research Engineer building self-improving AI agent systems at Console. Develop eval/optimization loops, fine-tune specialist models, and improve agent reasoning over enterprise context using production data to drive measurable gains in quality, latency, and reliability.

200k – 350k/yr
On-siteML Engineering
Notion

Software Engineer, AI Platform

NotionSan Francisco, CA +1

Build and scale the shared AI platform foundations at Notion, enabling fast and safe shipping of AI products. Requires experience with LLM/ML platforms, strong ownership, and comfort across backend, infrastructure, and product code.

180k – 201k/yr
Hybrid5+ YOEML Engineering
Liftoff

Machine Learning Engineer

LiftoffCalifornia

Machine Learning Engineer building statistical models, optimization systems, and experiments for mobile ad tech economics on the Revenue Engine team. Requires PhD in CS/ML/Economics and industry experience applying ML or economics at scale.

215k – 275k/yr
RemoteML Engineering