Member of Technical Staff - Pre-Training Infra

Reflection; New York (NY); Presencial; USD 130.000 - 180.000; Jornada completa

Ventajas ofrecidas por este puesto de trabajo

Top-tier compensation

Comprehensive medical, dental, and vision insurance

Fully paid parental leave

Paid time off

Daily meals provided

Descripción de la vacante

Reflection, based in New York, is seeking an experienced professional to build and scale distributed training systems for frontier model pre-training. You will work closely with research teams to design large-scale training runs and optimize training efficiency across thousands of GPUs. The ideal candidate has experience with modern distributed training frameworks, strong debugging skills, and familiarity with GPU communication libraries. The company offers top-tier compensation, comprehensive health benefits, and a supportive team environment.

Formación

• Experience building or operating distributed training systems for large machine learning models.
• Strong experience with modern distributed training frameworks like Megatron, DeepSpeed.
• Familiarity with large-scale model parallelism strategies.

Responsabilidades

• Build and scale distributed training systems for foundation models.
• Design and operate large-scale training runs with research teams.
• Optimize training throughput, stability, and efficiency.

Conocimientos

Distributed training systems

Distributed training frameworks (Megatron, DeepSpeed)

Optimization of GPU utilization

Familiarity with NCCL

Debugging skills in GPU compute

Descripción del empleo

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 stacks including model parallelism, GPU communication, and training runtime systems.
• Contribute to the development of systems that enable rapid experimentation and iteration on new training techniques.

Ideal Experience

• Experience building or operating distributed training systems for large machine learning models.
• Strong experience working with modern distributed training frameworks such as Megatron, DeepSpeed, or similar large-scale training systems.
• Familiarity with large-scale model parallelism strategies (data, tensor, pipeline, or expert parallelism).
• Experience optimizing training throughput and GPU utilization in large distributed environments.
• Familiarity with GPU communication libraries such as NCCL and performance tuning for distributed workloads.
• Experience working closely with ML researchers to productionize experimental training workflows.
• Strong debugging skills across GPU compute, distributed training systems, and large-scale ML pipelines.
• Experience working with large datasets and training pipelines used for foundation model pre-training.

What We Offer

We believe that to build superintelligence that is truly open, you need to start at the foundation. Joining Reflection means building from the ground up as part of a small talent-dense team. You will help define our future as a company, and help define the frontier of open foundational models. We want you to do the most impactful work of your career with the confidence that you and the people you care about most are supported.

• Top-tier compensation: Salary and equity structured to recognize and retain the best talent globally.
• Health & wellness: Comprehensive medical, dental, vision, life, and disability insurance.
• Life & family: Fully paid parental leave for all new parents, including adoptive and surrogate journeys. Financial support for family planning.
• Benefits & balance: paid time off when you need it, relocation support, and more perks that optimize your time.
• Opportunities to connect with teammates: lunch and dinner are provided daily. We have regular off-sites and team celebrations.