URBN is hiring a Staff Machine Learning Engineer to join the development of AI-powered visual experiences, focusing on building and operationalizing image and video generation systems. The role involves owning the engineering side of multi-model generative pipelines and collaborating with a cross-functional team to deliver significant business impact.
Responsibilities:
- Design, build, and optimize image and video generation pipelines (generation, inpainting, upscaling, style transfer, conditioning, post-processing) into production-ready, observable services with attention to cost, latency, and throughput
- Design and develop agentic workflows (ADK, A2A, LangGraph, or similar), MCP servers (FastMCP or similar), and microservices (FastAPI, GraphQL, or similar) to orchestrate and serve generative AI capabilities at scale
- Develop prompt management systems and structured prompting strategies for consistent visual output, learning on the job how to wrangle prompts for fashion-specific and virtual try-on use cases. Engineer consistency mechanisms and quality gates in partnership with data scientists who own evaluation methodology
- Integrate multimodal and vision-language models into production workflows for image understanding, automated tagging, captioning, and quality pre-screening
- Collaborate with Product Designers, Product Managers, Data Scientists, and other Engineers to translate brand and business needs into scalable generative AI solutions. Evaluate new technologies, models, and vendors through proof-of-concept studies
- Implement and maintain ML architecture, including pipelines and applications that enable training and inference of generative models in production. Champion best practices in full-stack algorithm engineering