ML Platform Engineer

About the Role

As an ML Platform Engineer at Stitch Fix, you will play a key role in building and maintaining the critical infrastructure that powers machine learning and AI across our organization. You will design, develop, and support scalable, resilient services and frameworks for ML model training and deployment, feature engineering and serving, candidate generation, AI agent deployment and observability, and other core platform capabilities.

In this role, you'll contribute to the day-to-day operations of the ML Platform team, ensuring the smooth functioning of existing systems while driving improvements. You’ll collaborate closely with full-stack data scientists, offering consultation and support to help them unlock the full potential of our platform.

With significant autonomy, you’ll have the opportunity to shape the future of ML and AI at Stitch Fix. Your ideas and expertise will drive improvements, codify best practices, and influence how we approach machine learning and AI systems at scale.

Responsibilities:

• Collaborate with cross-functional teams, including data scientists, engineers, and business partners, to solve complex distributed systems and business challenges at scale.
• Be part of a team with high visibility across the organization, driving impactful solutions that make a difference.
• Share your ideas and help guide the team’s investments toward high-value opportunities.
• Foster a culture of technical collaboration and contribute to the development of scalable, resilient systems.

About You

• You bring 3+ years of experience in software development, with a focus on data, ML, or AI infrastructure.
• You have a Bachelor's degree in an engineering or computer science field.
• You have a proven track record of building scalable, distributed production systems.
• Your coding and design skills are exceptional. While Python and Golang are our primary tools, your overall engineering expertise and problem-solving abilities matter most.
• You’re proficient in leveraging big data, web, and AI technologies such as Kafka, Postgres, Redis, FastAPI, Ray, AnyScale, LangGraph, LangSmith or similar tools.
• You have hands-on experience with AWS or other cloud-based providers.
• You excel at thinking globally, developing solutions that address multiple needs while prioritizing business impact.
• Your strong cross-functional communication skills enable you to simplify complex challenges and collaborate effectively with business partners to drive solutions forward.

Why you'll love working at Stitch Fix...

• We are a group of bright, kind people who are motivated by challenge. We value integrity, innovation and trust. You’ll bring these characteristics to life in everything you do at Stitch Fix.
• We cultivate a community of diverse perspectives— all voices are heard and valued.
• We are an innovative company and leverage our strengths in fashion and tech to disrupt the future of retail. 
• We win as a team, commit to our work, and celebrate grit together because we value strong relationships.
• We boldly create the future while keeping equity and sustainability at the center of all that we do. 
• We are the owners of our work and are energized by solving problems through a growth mindset lens. We think broadly and creatively through every situation to create meaningful impact.
• We offer comprehensive compensation packages and inclusive health and wellness benefits.