Senior MLE: Generative AI Platform Engineer

Amazon; Sunnyvale (CA); On-site; USD 193,000 - 262,000; Full time

Benefits offered by this job

Health insurance

RSUs

401(k) matching

Paid time off

Job summary

Amazon AI Studios is hiring a Machine Learning Engineer to own training and serving infrastructure for the CreativeFlux platform used by Prime Video animation and live action productions. You will deploy scalable ML workloads on EKS, integrate generative models into production pipelines, and collaborate with Applied Scientists and artists to move new architectures into production.

The role emphasizes hands‑on infrastructure, GPU optimization, and close collaboration with DCC tool integrations

Qualifications

• 5+ years of non‑internship professional software development experience.
• 5+ years of programming with at least one software programming language experience.
• 5+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience.
• Experience as a mentor, tech lead or leading an engineering team.

Responsibilities

• Build and operate ML training and serving infrastructure for the CreativeFlux platform. Strong hands‑on experience with EKS / Kubernetes is required.
• Design, deploy, and operate ML training and inference workloads on EKS, including GPU node groups, custom schedulers, autoscaling, resource quotas, and gang scheduling for multi‑GPU jobs
• Integrate generative ML models (image, video, audio) into production inference pipelines, partnering with Applied Scientists to take new architectures from research to scaled deployment
• Build training and fine‑tuning pipelines on EKS and SageMaker, including data preparation, distributed training, evaluation harnesses, and checkpointing strategies
• Improve GPU utilization, throughput, and latency on the inference fleet through batching, quantization, model compilation, serving framework tuning, and Kubernetes‑native scaling primitives (HPA, KEDA, custom controllers)
• Build automated evaluation pipelines and quality metrics that catch model regressions before they reach Artists, including human‑in‑the‑loop review where needed
• Contribute to integrations with Digital Content Creation (DCC) tools such as Adobe Animate, After Effects, Maya, and Storyboard Pro at the points where models are surfaced to creators
• Own operational health of model endpoints on EKS, including monitoring, alerting, on‑call response, capacity planning, and post‑incident analysis
• Write clean, well‑tested code and participate in design and code reviews
• Stay current with generative ML, distributed training, inference optimization, and GPU efficiency techniques, and apply them where they move team metrics

Skills

Software development

Programming

System design

Leadership / mentorship

Education

Bachelor's degree in computer science or equivalent

Tools

Kubernetes

EKS

SageMaker

Job description

Amazon AI Studios is hiring a Machine Learning Engineer to own training and serving infrastructure for the CreativeFlux platform used by Prime Video animation and live action productions. You will deploy scalable ML workloads on EKS, integrate generative models into production pipelines, and collaborate with Applied Scientists and artists to move new architectures into production.

The role emphasizes hands‑on infrastructure, GPU optimization, and close collaboration with DCC tool integrations