Senior Staff ML Platform Engineer — AI Factory

F7d8749a; San Mateo (CA); On-site; USD 210,000 - 260,000; Full time

Job summary

Shield AI is seeking a Senior Staff Engineer to define and build our Kubernetes‑native AI Factory platform. You will partner with ML researchers, platform engineers, and autonomy teams to deliver a scalable, high‑quality developer experience for training and deploying modern AI models.

You will work hands‑on across the stack, shaping architecture and implementation while balancing productivity, reliability, and long‑term maintainability in a fast‑moving AI ecosystem.

Qualifications

• Experience building Kubernetes-native AI or MLOps platforms.
• Deep understanding of modern AI training frameworks (PyTorch, Transformers).
• Experience operating GPU‑accelerated infrastructure and distributed training systems.
• Strong knowledge of Kubernetes, Linux, networking, security, storage.
• Experience with Terraform and Helm for cloud‑native infra.

Responsibilities

• Design and implement AI Factory Reference Architecture on Kubernetes.
• Partner with ML researchers to support evolving training workflows.
• Develop self‑service AI development workflows for local to distributed execution.
• Build infrastructure for distributed training, simulation, and RL workloads.
• Design shared GPU infrastructure and optimize resource utilization.
• Develop IaC and lifecycle management for cloud, on‑prem, air‑gapped deployments.

Skills

Kubernetes-native AI

distributed ML workloads

Python

Golang

collaboration with researchers

Tools

Terraform

Helm

PyTorch

Hugging Face Transformers

Ray

Job description

Shield AI is seeking a Senior Staff Engineer to define and build our Kubernetes‑native AI Factory platform. You will partner with ML researchers, platform engineers, and autonomy teams to deliver a scalable, high‑quality developer experience for training and deploying modern AI models.

You will work hands‑on across the stack, shaping architecture and implementation while balancing productivity, reliability, and long‑term maintainability in a fast‑moving AI ecosystem.