Software Engineer - ML Infrastructure
Gray Swan AI, Remote U.S. or Pittsburgh, PA
The role
We’re looking for a Software Engineer, AI Infrastructure to work at the intersection of machine learning, software engineering, and production infrastructure. You’ll partner closely with our ML team to build the systems, tools, and workflows that help move models from experimentation into reliable production environments.
You’ll improve the developer experience around model training and evaluation, build scalable infrastructure to support ML workflows, and help translate customer requirements and policy considerations into practical technical solutions. This role is ideal for a strong software engineer who enjoys solving complex infrastructure challenges while working closely with ML researchers and engineers to accelerate development and deliver real-world impact.
What you will do
- Build and maintain the infrastructure, services, and deployment workflows that enable ML models to move from experimentation into reliable production environments.