Senior MLE, CreativeFlux, AI Studios
Prime Video & Amazon MGM Studios · Sunnyvale, CA · 1 mo ago
ConsultingFull-time
About the role
Join the AI Studios Engineering org within Prime Video and Amazon MGM Studios as a Machine Learning Engineer on CreativeFlux, the ML platform powering Nara, our AI-native content creation platform for professional animation and live action/VFX.
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.
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