Skip to main content
OpenAI
OpenAI

Technical Lead Manager - Training Runtime, Data(set) Movement

RemoteUnited States only
Published
Role
Engineering Management
Experience
Lead
Employment
Full-time
Company size
Enterprise
$295k–$445k/yr
Check eligibility

Open to US only. Set where you work from to check your eligibility.

No BS summary

Hands-on technical lead manager for large-scale ML training data movement and dataset read infrastructure. Needs experience owning dataset/data loading/storage or distributed training infrastructure, with Python plus lower-level systems code. US-based telecommute role.

Core skills

Python

Optional skills

RustC++

What you'll do

  • Design and build a unified dataset read platform for multiple current and future training frameworks.
  • Define dataset APIs, storage-format expectations, registration/versioning, and migration paths for reproducible and maintainable data access.
  • Build reliability into the read path, including stateful iteration, caching, fast restart, recovery, and operational contracts.
  • Build terminal and web-based visualizers for inspecting text, multimodal, and reinforcement learning data late in the pipeline.
  • Write and review production code in core data loading, service, caching, and reliability paths.
  • Partner with teams working on training frameworks, reinforcement learning, multimodal models, storage, runtime, and cluster infrastructure.
  • Expand ownership over broader in-cluster data movement systems, including checkpoint loads/saves and snapshot transfers.

What they require

  • Built or owned dataset, data loading, storage, or distributed training infrastructure at large scale.
  • Care about API design, debugging ergonomics, performance, and bit-level correctness.
  • Understand failure modes of large distributed training jobs and how data systems can create or prevent them.
  • Experience with stateful iterators, checkpoint/restart semantics, caching, remote services, or high-throughput storage reads.
  • Comfortable working across Python and lower-level systems code.
  • Experience with multimodal, video, reinforcement learning, or pretraining data pipelines where small data bugs are expensive and hard to diagnose.
  • Can lead through code and technical judgment before a team exists, and later manage engineers while staying hands-on.
  • Focus on developer experience by reducing manual preprocessing scripts and cluster-specific bugs.

Benefits

  • Offers equity
  • Reasonable accommodations for applicants with disabilities

American artificial intelligence research organization

🇺🇸 United StatesAIEnterpriseopenai.com

What people say about this company

4.6/ 5

$295k–$445k/yr