Machine Learning Platform Lead Engineer, Training and Inference

Paramount•Burbank, CA

17d•$130,200 - $195,300

About The Position

We are seeking a Senior Lead / Lead ML Platform Engineer to architect and own the technical direction for our Training and Inference infrastructure. This is a high-leverage role designed for an expert who understands the deep technical stack required to shift ML models from research to global production. You will be responsible for the "engine room" of the AMLG, ensuring that our MLEs can train massive models efficiently and serve them with sub-millisecond reliability. This role requires a unique blend of expertise in distributed systems and hardware acceleration. You will lead the adoption and optimization of AnyScale (Ray) for distributed training and manage a high-performance Kubernetes-based inference environment. You aren't just managing clusters; you are building a seamless, scalable platform that abstracts the complexity of GPUs and distributed compute for the entire organization. The ML Platform Lead is the force-multiplier for every other ML pod. In this role, you will directly shape: The Training Foundation: Establishing AnyScale/Ray as the standard for distributed compute, enabling MLEs to train models on petabytes of data without managing infrastructure. Inference at Scale: Architecting the serving layer that handles billions of requests per day, optimizing for both p99 latency and GPU utilization. Operational Excellence: Setting the organizational standards for how ML models are deployed, monitored, and scaled across the enterprise.

Requirements

6-8+ years of experience in ML Infrastructure, Platform Engineering, or high-scale Backend Engineering.
Extensive experience with Kubernetes (K8s) and serving frameworks for large-scale ML models.
Strong knowledge of GPU architecture, CUDA, and optimizing ML workloads for hardware acceleration.
Proven track record of owning the technical direction for a major domain and driving impact across multiple teams.

Nice To Haves

Experience with Infra-as-Code (Terraform/Pulumi) and building automated MLOps pipelines.
Deep expertise with Ray (AnyScale) or similar distributed compute frameworks.
Familiarity with ML observability tools (Prometheus, Grafana, Weights & Biases, or MLFlow).
Experience managing multi-cloud or hybrid-cloud ML environments.
Deep knowledge of Python and C++ for performance-critical systems.

Responsibilities

Own the long-term architectural direction for the Training and Inference domains, ensuring the platform scales 10x over a 1–3 year horizon.
Lead the implementation and optimization of Ray/AnyScale, providing a unified compute layer for batch processing, model training, and reinforcement learning.
Design and maintain K8s-based inference servers (e.g., Triton, TorchServe, or vLLM) optimized for GPU memory management and high throughput.
Navigate the trade-offs between different GPU instances (A100s, H100s, T4s), optimizing for cost, availability, and performance.
Solve high-leverage problems that affect multiple pods (e.g., Entry, Session, Presentation), establishing reusable patterns for CI/CD, model versioning, and canary deployments.
Define and enforce SLIs/SLOs for the platform, ensuring that infrastructure failures never interrupt the user-facing personalization experience.
Act as a technical mentor to senior engineers across the ML Platform and Applied ML pods, raising the bar for system design and operational rigor.