About the job
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. You will own the training and serving infrastructure that puts generative models in front of artists working on Prime Video's animated and live action productions. You'll work daily with Applied Scientists, animators, filmmakers, and storytellers, building the systems that turn new model architectures into production-grade tools artists can rely on under tight deadlines.
Responsibilities
Build and operate ML training and serving infrastructure for the CreativeFlux platform
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
Build automated evaluation pipelines and quality metrics that catch model regressions before they reach Artists
Own operational health of model endpoints on EKS, including monitoring, alerting, on-call response, capacity planning, and post-incident analysis
Qualifications
Minimum
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
Preferred
5+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
Bachelor's degree in computer science or equivalent