Software Engineer - ML Infrastructure

Gray Swan builds AI safety and security evaluation and threat-detection systems. The Software Engineer - ML Infrastructure builds and maintains production infrastructure, tools, and workflows that move machine-learning models from experimentation into reliable deployment, collaborating with ML teams and addressing customer and policy requirements.

Responsibilities:

• Build and maintain the infrastructure, services, and deployment workflows that enable ML models to move from experimentation into reliable production environments
• Develop internal tools and automation that improve the developer experience for model training, experimentation, evaluation, and iteration
• Design and optimize systems supporting model evaluation pipelines, experiment orchestration, reproducibility, and performance monitoring
• Partner with ML engineers and researchers to identify infrastructure bottlenecks and build solutions that accelerate model development and deployment
• Support customer-specific technical requirements and policy-driven workflows, translating evolving needs into practical, scalable engineering solutions
• Improve the reliability, scalability, and observability of the systems powering our ML capabilities