Sr. MLE, Prime Video ML Platform
Core
Design and build scalable ML infrastructure and platform services to enable rapid model iteration, experimentation, and personalization for Prime Video's video discovery experiences.
Role type
Senior IC machine-learning platform engineer (technical lead)
Builds
ML platform services, model serving infrastructure, feature stores, and training pipelines
Domain
Streaming entertainment, personalization, distributed systems
Deliverable
production ML models | infrastructure
Required skills
ML framework expertise (JAX, PyTorch, vLLM, SGLang, Dynamo, TorchXLA, TensorRT), transformer architecture knowledge, training/inference lifecycle management, system design and scaling, mentorship
Preferred skills
Full SDLC experience, coding standards, code reviews, source control, testing, operations
Technologies
JAX, PyTorch, vLLM, SGLang, Dynamo, TorchXLA, TensorRT
Responsibilities
Build ML platform services with Tier-1 availability; evolve model serving and feature delivery infrastructure; design and scale training pipelines and real-time inference systems; lead strategic technical vision for model lifecycle management; partner with scientists and PMs to translate research into production capabilities
Seniority
Senior, hands-on IC with technical leadership