Member of Technical Staff, Model Training (infra & platform), Remote- Full TIme
Core
Architect, build, and scale shared infrastructure for distributed ML training, inference, and agent execution to support research velocity and model performance.
Role type
Senior IC distributed systems engineer (ML infrastructure)
Builds
Distributed training pipelines, GPU inference engines, simulation environments, and agent execution frameworks
Domain
AI/ML infrastructure, distributed systems, high-performance computing
Deliverable
infrastructure
Required skills
Distributed systems architecture, GPU performance optimization, Python, C++/Rust/Go, system concurrency, fault tolerance, API design, technical leadership
Preferred skills
PyTorch Distributed, DeepSpeed, Ray, Megatron-LM, vLLM, TensorRT-LLM, CUDA, Kubernetes, robotics infrastructure
Technologies
PyTorch, DeepSpeed, Ray, Megatron-LM, vLLM, TensorRT-LLM, CUDA, NCCL, Kubernetes
Responsibilities
Architect distributed training and execution pipelines; optimize GPU utilization and memory management; build experiment tracking and versioning systems; define technical standards and platform interfaces; lead engineering initiatives and mentor teams
Seniority
Senior, hands-on IC
