Senior Research Engineer, Model Training / Pretraining

Job Description

Senior Research Engineer

\n

Seattle, Washington | Hybrid

\n

\n

The position is a Senior Research Engineer on the model scaling team, reporting to a Director of AI Research. The work covers model training and the supporting infrastructure. Training runs use approximately 600 B300 and 1,000 H100 GPUs.

\n

\n

\n

Responsibilities

\n

\n

\n
• Build and maintain the training stack for distributed model training across GPU clusters.
\n
• Run scaling experiments covering model architecture, data, and optimisation.
\n
• Work in one specialist area: GPU kernel development (CUDA), Mixture-of-Experts architectures, or reinforcement-learning post-training (GRPO, PPO, RLVR).
\n
• Release models, code, and associated research publicly.
\n

\n

\n

Requirements

\n

\n

\n
• 6+ years of software engineering experience.
\n
• 4+ years of machine-learning infrastructure experience.
\n
• Python and PyTorch.
\n
• Experience training models from scratch (pretraining). Experience limited to retrieval-augmented generation or application-level fine-tuning does not meet this requirement.
\n
• Depth in at least one of: CUDA kernels, Mixture-of-Experts, or reinforcement learning (GRPO, PPO, RLVR).
\n

\n

\n

Also considered

\n

\n
• JAX.
\n
• Open-source contributions.
\n
• Published research.
\n
• Experience with GPU clusters at the scale described above.
\n

\n

\n

Interview process

\n

\n

\n
1. Hiring-manager screen — 25 minutes.
\n
2. Recruiter screen — 15 minutes.
\n
3. Machine-learning coding interviews — two sessions, 45 minutes each.
\n
4. Machine-learning system-design interview — 45 minutes.
\n
5. Technical deep-dive with the Director of AI Research — 60 minutes.
\n
6. Final interview — 30 minutes.
\n
7. Offer.
\n

\n

\n

\n
• Base salary: $147,000–$220,000 + strong bonus programmes in addition to base salary.
\n
• Hybrid in Seattle, Washington. On-site presence is required. Fully remote is not available.
\n
• Visa sponsorship is available.
\n

\n

\n

Application

\n

\n

This search is managed by Intelix.AI.

\n

Contact: Hasan Mohammad — View email address on ziprecruiter.com.