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 job

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