Member Of Technical Staff - Research Software Engineer

Reflection AI

New York, United States
On-site
Distributed training infrastructure
Large-scale data pipelines
Reinforcement learning training
Bridge the gap between research and production by turning cutting-edge algorithms into scalable training systems

Job Summary

  • Bridge the gap between research and production by turning cutting-edge algorithms into scalable training systems.
  • Design and optimize the core infrastructure behind frontier AI models — from reinforcement learning training loops and distributed GPU training to massive-scale data pipelines.
  • Top-tier compensation: Salary and equity structured to recognize and retain the best talent globally.

Matching Summary

Bridge the gap between research and production by turning cutting-edge algorithms into scalable training systems.

Skills & Requirements

Must-have

  • Distributed training infrastructure
  • Large-scale data pipelines
  • Reinforcement learning training
  • Numerical stability and efficiency
  • PyTorch or JAX
  • High-performance computing

Nice-to-have

  • Implementing research papers
  • High-agency environments
  • Solving hard technical problems
  • Open weight models development

Key Requirements

  • Strong software engineering skills
  • Experience in Distributed Training & Inference or Data Infrastructure
  • Understanding of machine learning algorithms

Work Rights

Not specified

Tailored Resume

Cover Letter