Autonomy for the world. Backed by a16z.
About the role
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.
What they're looking for
- 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
More about this role
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 www.shield.ai . Follow Shield AI on LinkedIn , X , Instagram , and YouTube .
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...
Browse similar: AI jobs · AI startup jobs · Startup jobs · San Francisco Bay Area