Software Engineer 5 – Model Runtime, AI Platform

Netflix
USA - Remote2026-04-08onsite

About the job

The Model Runtime team owns the systems that train, align, and serve Netflix's most critical ML models. We are a small, highly autonomous team with outsized impact — the infrastructure we build directly shapes what Netflix can do with AI.

Responsibilities

Build alignment and post-training infrastructure — Design infrastructure for reinforcement learning (GRPO, DPO, PPO), reward modeling, and preference optimization so Netflix can train recommendation models directly against what members actually value.

Enable next-generation GenAI workloads — Create infrastructure for multimodal and diffusion models, including distributed training, disaggregated serving, real-time, near-real-time and batch inference, and asynchronous GPU pipelines.

Scale distributed training — Engineer fault-tolerant training systems using FSDP, tensor/pipeline/context parallelism, and mixed-precision strategies across clusters of hundreds of GPUs.

Optimize across the full stack — Profile and tune from PyTorch operators down to GPU kernels, driving utilization improvements and building cost models that inform infrastructure strategy.

Evaluate emerging hardware and frameworks — Be the team's eyes on specialized accelerators, next-gen NVIDIA silicon, and the open-source ecosystem to keep Netflix at the efficiency frontier.

Qualifications

Minimum

Experience in ML systems engineering — building infrastructure for training, fine-tuning, or inference of pre-LLM and post-LLM era models at scale.

Strong systems programming skills with the ability to work across multiple layers of the stack, from high-level ML frameworks down to GPU kernels and memory management

Hands-on experience with PyTorch internals, large-scale distributed training and system-model codesign

Comfortable with ambiguity and working across multiple business and technical domains to execute on both 0-to-1 and 1-to-100 projects

Adopt and promote best practices in operations, including observability, logging, reporting, and on-call processes to ensure engineering excellence

Experience with cloud computing providers, preferably AWS

Excellent written and verbal communication skills

Strong communication skills; effective across distributed time zones and remote environments

Preferred

Deep experience with distributed training at scale (FSDP, parallelism strategies, checkpointing) or LLM post-training (SFT, RLHF, DPO/GRPO)

Inference optimization — vLLM, TensorRT, quantization, continuous batching, KV-cache management

GPU performance profiling and tuning (CUDA, NCCL, Nsight, PyTorch profiler)

Experience with multimodal or diffusion model architectures and generation pipelines

Track record building reusable ML libraries or contributing to open-source ML projects