Senior Staff Engineer, ML Ops (R4941)
Shield AI · Engineering
- Location
- San Mateo, California
- Metro
- San Francisco Bay Area
- Workplace
- On-site
- Level
- Staff+
- Type
- Full-time
- Pay
- $280K–$420K/yr (published by employer)
- Posted
- 2 months ago
- Source
- Lever
Read the full posting & apply ↗
PythonGoKubernetesTerraformLinuxPyTorch
What the posting covers
About the company
Shield AI is a venture-backed defense-tech company with the mission of protecting service members and civilians with intelligent systems. Its products include Hivemind autonomy software, V-BAT and X-BAT aircraft, and Aechelon simulation and synthetic reality…
Job Description
Shield AI builds autonomy systems for defense applications, including air, maritime, and space platforms operating in complex and contested environments. We are building the AI Factory Reference Architecture, a Kubernetes-native platform for developing,…
What you'll do
AI Platform Development: Lead the design and implementation of the AI Factory Reference Architecture, delivering a Kubernetes-native platform for AI development, distributed training, simulation, evaluation, and deployment. AI Research Enablement: Partner…
What you'll do
Technology Leadership: Evaluate emerging AI infrastructure technologies and establish architectural patterns that balance scalability, performance, maintainability, and developer experience. Cross-Functional Collaboration: Work closely with AI researchers,…
Key Outcomes
Engineers move from idea to distributed training in hours instead of days. High GPU utilization through efficient scheduling with KAI on Kubernetes. Researchers use modern AI tooling without unnecessary platform friction. AI Workspaces become the standard…
Required qualifications
Experience building Kubernetes-native AI or MLOps platforms supporting distributed machine learning workloads. Deep understanding of modern AI training frameworks, including PyTorch, Hugging Face Transformers and distributed training techniques. Experience…
Preferred qualifications
Experience with Ray or other distributed AI orchestration frameworks. Experience with KAI, Slurm or other GPU scheduling technologies. Experience supporting reinforcement learning, simulation-driven training, robotics, or autonomy workloads. Experience…
Why Join Us
The AI Factory is the foundation for how Shield AI develops and deploys autonomy. As a Senior Staff Engineer, you will help build the reference architecture used internally by our engineering teams and delivered to customers building their own AI…
Excerpts only. The employer’s page has the complete posting.