Senior Machine Learning Engineer (Diffusion Models & Generative AI)

arrow

/ $180000 - $200000 annum

INFO

Salary
SALARY:

$180000 - $200000

Location

LOCATION

Job Type
JOB TYPE

Permanent

Senior Machine Learning Engineer (Diffusion Models & Generative AI)

NEW YORK - Onsite, 5 Days/Week
Up to $200,000 + Benefits + H1B Transfer Available

Are you passionate about building cutting-edge AI systems that transform the online shopping experience? A fast-growing Series A e-commerce startup in New York is looking for a Senior ML Engineer to push the boundaries of diffusion models and generative AI for next-generation retail solutions.

Backed by leading investors and already working with top global retailers, this is your chance to join an ambitious company at a critical stage of growth. You'll work with a world-class engineering team applying multimodal AI and diffusion models to create immersive, personalized, and automated product experiences.


THE COMPANY

This AI-powered e-commerce startup is redefining how products are discovered, designed, and purchased online. By harnessing the power of diffusion models, computer vision, and multimodal learning, they're unlocking new ways to enrich catalogs, generate content, and personalize shopping.

With Series A funding secured, a growing team of 40+, and a strong customer pipeline, they're tackling challenges in:

  • Synthetic product image generation

  • Virtual try-ons and visualization

  • Automated catalog enrichment

  • AI-powered personalization and search

Their mission: use generative AI to supercharge e-commerce innovation and transform the way consumers shop.


THE ROLE

As a Senior ML Engineer, you'll take ownership of designing, training, and deploying diffusion-based generative models to power real-world e-commerce applications.

You will:

  • Research, develop, and fine-tune diffusion models for product imagery, augmentation, and personalization

  • Build scalable pipelines for training, evaluation, and inference with large datasets

  • Work with cross-functional teams to deploy AI-powered catalog and shopping features

  • Explore and adapt state-of-the-art generative AI methods (diffusion, multimodal transformers, vision-language fusion) into production systems

  • Stay on top of innovations in generative AI, foundation models, and synthetic data


YOUR SKILLS & EXPERIENCE

Must-haves:

  • MS or PhD in Computer Science, Machine Learning, or related field

  • 5+ years of experience building and deploying ML models in production

  • Strong expertise in diffusion models and/or generative modeling

  • Proficiency in Python, PyTorch/TensorFlow

  • Strong understanding of deep learning architectures, training optimization, and evaluation

Nice-to-haves:

  • Experience in e-commerce or retail product datasets

  • Exposure to multimodal systems (vision + text)

  • Knowledge of LLMs or RAG pipelines for search and recommendations


BENEFITS & CULTURE

  • Competitive salary up to $200K

  • Full health, dental, and vision coverage

  • H1B transfer sponsorship available

  • Work onsite in a collaborative NYC office environment

  • Daily catered lunches, team events, and a fast-paced, innovative culture


READY TO APPLY?

If you're excited about building the future of generative AI in e-commerce, we'd love to hear from you.

CONTACT

Luc Simpson-Kent

Senior recruitment Consultant

SIMILAR
JOB RESULTS

4k-Harnham_DA copy
CAN’T FIND THE RIGHT OPPORTUNITY?

STILL
LOOKING?

If you can’t see what you’re looking for right now, send us your CV anyway – we’re always getting fresh new roles through the door.