Staff Platform Engineer AI Compute

Shield AI is a U.S. defense-technology company developing mission-autonomy software and autonomous aircraft for military and allied operations.

Series F+Recently funded
San Diego, United States
About Shield AI

Founded in 2015, Shield AI builds Hivemind autonomy software and V-BAT and X-BAT aircraft for operations in contested, GPS- and communications-denied environments. Its current site also presents Aechelon synthetic-reality simulation and Vision Systems detection and tracking products.

View jobs by Shield AI

Skills

Candidate Availability

Required and preferred rules are kept separate and reflect the wording in the original posting.

About the Role

You will build and maintain Kubernetes-based GPU compute platforms. You will automate deployments, troubleshoot workload and infrastructure issues, improve scheduling and access controls, create diagnostics, and use workload benchmarks to guide capacity and hardware decisions.

Requirements

  • Experience building and supporting production Kubernetes platforms with GPU or distributed-computing workloads
  • Software engineering skills in Python, Go, or a comparable language
  • Experience automating infrastructure deployment, configuration, and upgrades
  • Linux, networking, and storage troubleshooting skills
  • Experience with shared-resource scheduling, access controls, or workload isolation
  • Ability to work with machine-learning engineers and translate recurring problems into platform capabilities

Responsibilities

  • Build and maintain software and configuration for GPU workloads on Kubernetes
  • Design quotas, scheduling priorities, permissions, and workload isolation
  • Automate GPU service deployment and upgrades using infrastructure as code, Terraform, and Helm
  • Troubleshoot scheduling, configuration, storage, and networking issues
  • Build diagnostics for queue times, GPU utilization, memory requirements, and training throughput
  • Benchmark workloads and recommend GPU capacity and hardware
  • Deliver repeatable installations across cloud, on-premises, and air-gapped environments

Benefits

  • Bonus
  • Benefits
  • Equity
Staff Platform Engineer AI Compute at Shield AI | JobStash