5h ago
Member of Technical Staff - Research Software Engineer
New York
✨ $200k-$300k / yearest.
full-timeseniorai-ml
🛠 Tech Stack
💼 About This Role
You'll design and optimize core training infrastructure for frontier AI models, bridging research and production. Your impact includes building RL training loops, distributed GPU systems, and petabyte-scale data pipelines. You'll work on open superintelligence that is accessible to all.
🎯 What You'll Do
- Design and optimize large-scale training loops and data pipelines.
- Implement state-of-the-art techniques with numerical stability.
- Build internal tooling for launching and monitoring experiments.
- Diagnose bottlenecks across the training stack (GPU, communication, dataloader).
📋 Requirements
- Strong software engineering experience with ML systems.
- Deep experience in distributed training & inference or data infrastructure.
- Ability to implement research papers into production-grade code.
- Experience with performance optimization in distributed systems.
✨ Nice to Have
- Experience with NCCL, RDMA, GPU interconnects.
- Familiarity with FSDP / ZeRO and model sharding.
- Knowledge of Triton or custom kernel development.
🎁 Benefits & Perks
- 💰 Top-tier compensation: Salary and equity globally competitive.
- 🏥 Health & wellness: Comprehensive medical, dental, vision, life, disability.
- 👶 Life & family: Paid parental leave, family planning support.
- 🌴 Paid time off and relocation support.
- 🍽️ Daily meals and team off-sites.
0 0 0