Staff Software Engineer - Computer Vision Deployment

Willow is hiring a Staff Software Engineer to build and scale the infrastructure behind its AI-driven warehouse intelligence platform. You'll own the full lifecycle of computer vision models, from training pipelines to optimized cloud deployment, and design GPU compute clusters and orchestration systems for low-latency inference in production. This role sits at the intersection of machine learning, distributed systems, and cloud infrastructure, collaborating with research scientists and product teams. The team works from the SF office three days a week.

• Develop and maintain distributed cloud GPU infrastructure for world model training and low-latency inference.
• Build end-to-end computer vision pipelines from data ingestion through model training, evaluation, and deployment.
• Deploy and optimize ML models (including VLMs and VLAs) using model serving platforms and inference optimization techniques.
• Design and operate orchestration systems for ML/data pipelines used by engineers and non-engineers.
• Establish monitoring, benchmarking, and evaluation frameworks for production reliability.

• B.S./M.S. in Computer Science, Robotics, or related field (or equivalent practical experience).
• 7+ years of professional software engineering experience, with at least 3 years in machine learning infrastructure.
• Track record of deploying computer vision models in production environments.