Profile cover
A

AI Image Generation Pipeline Engineer

Adrian Bergmann

Full-time · Job · Los_Angeles

About the role

Our creative platform generates 2 million images per day for e-commerce product listings. Each request needs to match brand guidelines, maintain product accuracy, and render in under 8 seconds. The generation models work — the infrastructure serving them at this scale is what needs senior engineering talent. You'll own the end-to-end generation pipeline: request routing across GPU clusters, dynamic model loading for client-specific fine-tuned checkpoints, image post-processing for format and resolution requirements, and the caching layer that serves repeat request patterns without burning GPU cycles. Scaling image generation isn't like scaling a web API. Model loading takes 15 seconds. GPU memory is finite. Batch sizes affect quality. You'll navigate these constraints while hitting latency SLAs that clients depend on for their product launch workflows.

Responsibilities

  • Architect the image generation serving pipeline for 2M+ daily requests
  • Design dynamic model loading for 200+ client-specific fine-tuned checkpoints
  • Build intelligent caching systems for repeat generation pattern detection
  • Optimize GPU cluster utilization through request routing and batch scheduling
  • Define and maintain latency SLAs for generation pipeline performance

Requirements

  • 6+ years in ML infrastructure with generative model serving experience
  • Deep Kubernetes and Docker expertise for GPU workload management
  • Strong PyTorch knowledge for model loading and inference optimization
  • AWS skills for scalable cloud GPU infrastructure
  • Experience with high-throughput ML serving systems at production scale

Benefits

  • Fully remote — US West Coast hours
  • Senior equity package
  • $3,500 annual learning budget
  • Working on one of the largest image generation platforms

Keep exploring related roles

Use these topic pages to discover more roles in the same hiring cluster.

Categories

AI & Machine Learning Engineers MLOps & AI Infrastructure

Skills

python pytorch docker kubernetes aws

Related remote AI jobs

Nearby roles that share similar skills, categories, or hiring intent.

Browse all jobs

Senior Product Security Engineer

RemoteAI company

Full-time Senior

AI Resident UK/EMEA

RemoteAI company

Full-time

Principal Engineer

RemoteAI company

Full-time Principal

Staff Application Security Engineer

RemoteAI company

Full-time Staff

Senior AI R&D Engineer – Model Training (f/m/d)

RemoteAI company

Full-time Senior

Senior AI Researcher- Pre-training (f/m/d)

RemoteAI company

Full-time Senior

Job Type

Full-time

Level

Senior

Language

English

Salary Range

$160k – $200k / year

AI Expertise

AI & Machine Learning Engineers MLOps & AI Infrastructure

Skills

python pytorch docker kubernetes aws fastapi

Ready to apply for this role?

Create a free talent account in under 2 minutes.

  • Apply to verified AI companies
  • Get AI-matched job recommendations
  • Message hiring managers directly
  • Build your public AI talent profile
Create free account & apply Log in

Interested in this role? Create your free profile to apply.

Apply — create free profile