5 months ago
Base Salary
$233k - $350k/yr
Responsibilities
- Lead the design and implementation of the Kubernetes-native AI Factory Reference Architecture.
- Build infrastructure for distributed training, simulation, inference, reinforcement learning, evaluation, and deployment.
- Create self-service developer workflows that support local experimentation through large-scale distributed execution.
- Design and optimize shared GPU infrastructure, including scheduling, storage, networking, observability, and reliability.
- Build dataset management, experiment tracking, artifact management, model versioning, evaluation, deployment, monitoring, and continuous improvement capabilities.
- Develop repeatable Infrastructure as Code and platform lifecycle solutions for cloud, on-premises, sovereign, customer-managed, and air-gapped environments.
- Evaluate AI infrastructure technologies and establish scalable, maintainable architectural patterns.
- Partner with ML researchers, autonomy teams, infrastructure engineers, and product teams.
Requirements
- Experience building Kubernetes-native AI or MLOps platforms for distributed machine learning workloads.
- Deep understanding of 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 and large-scale AI workloads.
- Experience deploying cloud-native infrastructure with Terraform and Helm.
- Strong software engineering skills in Python and Golang.
- Experience translating ML research workflows into scalable platform capabilities.
- Preferred: experience with Ray, KAI, Slurm, reinforcement learning, simulation-driven training, robotics, autonomy workloads, edge AI hardware, classified or air-gapped infrastructure, OpenTelemetry, Prometheus, Grafana, and open-source infrastructure projects.
Benefits
- Full-time regular employee offer package includes bonus, benefits, and equity in addition to pay within the listed range.
- Offers are contingent on a cleared background and possible reference check.
- Military fellows and part-time employees are not eligible for benefits.
- The employer provides disability or special-needs accommodations.
Categories
About Shield AI
Shield AI builds autonomous systems for military and national security customers, combining its Hivemind autonomy software with V-BAT and X-BAT unmanned aircraft and Aechelon simulation technologies. The privately held company sells hardware, software, and related services to U.S. and allied defense agencies. Founded in 2015 and headquartered in San Diego, it operates across the U.S., Europe, the Middle East, and Asia-Pacific, and its technology is used in operational deployments.