Senior Staff Engineer, ML Ops (R4941)
- Compensation
- $233K–$350K / yr
- Location
- San Diego, California
- Work setup
- Onsite
- Seniority
- Staff
- Employment
- Part time
- Experience
- Not specified
- Category
- Devops
- Posted
Skills
Must have
- 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 operating GPU-accelerated infrastructure and distributed training systems.
- Strong understanding of Kubernetes, Linux, networking, security, storage, and distributed systems.
- Experience with GPU scheduling concepts and large-scale AI workloads.
- Experience packaging and deploying cloud-native infrastructure using Terraform and Helm.
- Strong software engineering skills in Python and Golang and modern cloud-native technologies.
- Experience collaborating closely with ML researchers to translate research workflows into scalable platform capabilities.
Nice to have
- 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 deploying and optimizing AI models for edge hardware.
- Experience designing infrastructure for classified, sovereign, or air-gapped environments.
- Experience with observability technologies such as OpenTelemetry, Prometheus, and Grafana.
- Experience contributing to or maintaining open-source infrastructure projects.
Full description
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 technologies. With offices and facilities across the U.S., Europe, the Middle East, and Asia-Pacific, Shield AI’s technology actively supports operations worldwide. For more information, visit . Follow Shield AI on LinkedIn, X, Instagram, and YouTube.
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, training, evaluating, and deploying next-generation AI systems.
The AI Factory serves two purposes. Internally, it powers autonomy development across Hivemind and other AI programs. Externally, it becomes the reference architecture deployed into customer environments, spanning commercial cloud, on-premise infrastructure, sovereign deployments, and fully air-gapped systems.
We are looking for a Senior Staff Engineer to help define and build this platform. You will partner closely with ML researchers, platform engineers, and autonomy teams to deliver an exceptional developer experience for training and deploying modern AI models.
Success in this role requires balancing researcher productivity, platform simplicity, operational excellence, and long-term maintainability. You will work hands-on across the stack, helping shape both the platform architecture and its implementation while staying closely aligned with the rapidly evolving AI ecosystem.
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 directly with ML researchers to understand evolving training workflows and ensure the platform supports state-of-the-art AI frameworks, foundation model development, reinforcement learning, distributed training, and emerging research workflows.
- Developer Experience: Design self-service AI development workflows that enable engineers to move seamlessly from local experimentation to large-scale distributed execution using familiar open source tools and frameworks.
- Distributed AI Infrastructure: Build the infrastructure required to support distributed training, simulation, inference, and reinforcement learning workloads. Evaluate and integrate orchestration, scheduling, and resource management technologies to maximize scalability and developer productivity.