Machine Learning Engineer
$180K–$200K + 0.8–2% equity
New York, NY (In-office, 5 days/week)
AI-powered fashion design tools
Join an early-stage startup reshaping how the fashion industry designs, visualizes, and produces apparel. This team is building a creative platform that merges AI image generation, collaborative design tooling, and supply chain workflows—all in one digital canvas. Their software helps top fashion brands move from idea to sample faster than ever before, reducing costs and accelerating time-to-market.
What You’ll Do
- Fine-tune and productionize diffusion models tailored for garment design.
- Own the entire ML stack from training to deployment, inference, and monitoring.
- Translate research prototypes into production-ready image generation tools.
- Collaborate with product and engineering teams to align model performance with UX.
- Drive innovation in image post-processing, upscaling, and model compression.
- Build and maintain robust APIs for ML inference at scale.
What You Bring
- Proven experience training and deploying generative vision models (e.g., diffusion).
- Fluency in Python and PyTorch, with a deep understanding of image modeling.
- Experience with cloud infrastructure for ML (GCP preferred), containerization (Docker), and orchestration (Kubernetes).
- Strong engineering fundamentals and an experimental mindset—comfortable building and shipping quickly.
- Bonus: experience working with multimedia models (image + video) or applied creativity in fashion, gaming, or art-tech contexts.
Tech Stack & Environment
- ML: Diffusion models, PyTorch, Transformers, custom vision architectures
- Infra: GCP, Docker, Kubernetes, model optimization pipelines
- Tools: Streamlit/Gradio for demoing, REST APIs for integration, basic observability stack
Why Join
- Founding team of designers, engineers, and operators with deep experience in fashion and SaaS.
- Real enterprise traction: already powering design teams at major brands.
- High-impact role with ownership and direct line to product outcomes.
- Collaborative, in-person team culture in NYC—creative energy and fast iteration.
At People In AI, we work exclusively with teams applying cutting-edge AI to high-value real-world use cases. We’re excited to represent this opportunity and support your candidacy with insight, transparency, and context every step of the way.