Research Engineer, Pretraining Scaling

Job summary

Responsibilities

Anthropic's ML Performance and Scaling team trains our production pretrained models, work that directly shapes the company's future and our mission to build safe, beneficial AI systems.

• Own critical aspects of our production pretraining pipeline, including model operations, performance optimization, observability, and reliability
• Debug and resolve complex issues across the full stack—from hardware errors and networking to training dynamics and evaluation infrastructure
• Design and run experiments to improve training efficiency, reduce step time, increase uptime, and enhance model performance
• Respond to on-call incidents during model launches, diagnosing problems quickly and coordinating solutions across teams
• Build and maintain production logging, monitoring dashboards, and evaluation infrastructure

Tech stack

• JAX
• PyTorch