JobMesh

Member of Technical Staff - Pre-Training Infra

Reflection AI · San Francisco, California, US

Our Mission Reflection’s mission is to build open superintelligence and make it accessible to all . We’re developing open weight models for individuals, agen...

Job description

Our Mission Reflection’s mission is to build open superintelligence and make it accessible to all . We’re developing open weight models for individuals, agents, enterprises, and even nation states. Our team of AI researchers and company builders come from DeepMind, OpenAI, Google Brain, Meta, Character.AI, Anthropic and beyond. About the Role: 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 st...