Staff+ Software Engineer, Safeguards ML Infrastructure
San Francisco, CA
Posted 16d ago
About the job
Anthropic's mission is to create reliable, interpretable, and steerable AI systems that are safe and beneficial for users and society. We are seeking a Machine Learning Infrastructure Engineer to join our Safeguards organization. In this role, you will build and scale the critical infrastructure that powers our AI safety systems, working at the intersection of machine learning, large-scale distributed systems, and AI safety. You will develop the platforms and tools that enable our safeguards to operate reliably at scale and directly contribute to making AI systems more trustworthy and aligned with human values.
Responsibilities
- Design and build scalable ML infrastructure for real-time and batch classifier and safety evaluations.
- Build monitoring and observability tools for safety-critical applications.
- Collaborate with research teams to productionize safety research.
- Optimize inference latency and throughput for real-time safety evaluations.
- Implement automated testing, deployment, and rollback systems for ML models.
- Partner with Safeguards, Security, and Alignment teams to deliver infrastructure.
- Contribute to the development of internal tools and frameworks for safety research and deployment.
Requirements
- 5+ years of experience building production ML infrastructure, ideally in safety-critical domains.
- Proficient in Python and experienced with ML frameworks like PyTorch, TensorFlow, or JAX.
- Hands-on experience with cloud platforms (AWS, GCP) and container orchestration (Kubernetes).
- Understand distributed systems principles and built high-throughput, low-latency systems.
- Experience with data engineering tools and building robust data pipelines (e.g., Spark, Airflow, streaming systems).
- Results-oriented with a bias towards reliability and impact in safety-critical systems.
- Enjoy collaborating with researchers and translating research into production systems.
- Deep care for AI safety and societal impacts.
Benefits
- Annual compensation range: $320,000 - $405,000 USD
- Visa sponsorship available