Hiring: Member of Technical Staff - Pre-Training Infra Company: Reflection Location: San Francisco, CA Job Posted Time: 2026-09-12 13:40:12 Target Skills & Keywords : Machine Learning About the job Experience: •Strong experience working with modern distributed training frameworks such as Megatron, DeepSpeed, or similar large-scale training systems. •Operational familiarity with large-scale model parallelism strategies (data, tensor, pipeline, or expert parallelism). •Operational familiarity with GPU communication libraries such as NCCL and performance tuning for distributed workloads. •Strong debugging skills across GPU compute, distributed training systems, and large-scale ML pipelines Required Skills: •Build and scale distributed training systems that power frontier model pre-training. •Work closely with research teams to design and operate large-scale training runs for foundation models. •Develop infrastructure that enables efficient training across thousands of GPUs using modern distributed training frameworks. •Optimize training throughput, stability, and efficiency for large model training workloads. •Collaborate directly with pre-training researchers to translate experimental ideas into scalable, production-ready training systems. •Improve performance of distributed training workloads through optimization of communication, memory usage, and GPU utilization. •Build and maintain training pipelines that support large-scale datasets, checkpointing, and experiment iteration. •Debug and resolve performance bottlenecks across distributed training stacks including model parallelism, GPU communication, and training runtime systems. Compensation: •Top-tier compensation: Salary and equity structured to recognize and retain our talent globally Interested candidates, please apply directly through the job posting on company's career page or try via AI auto apply on this platform. Don't miss this opportunity to join a forward-thinking team!