Find your next role
Discover roles across ambitious teams
17,770 jobs
- More jobs
Shield AI
Staff, Platform Engineer AI Compute
San Diego, California · On-site · Full time · Senior level
$200k–$300k / year
About the role
Build and maintain GPU compute platforms for machine-learning workloads. Improve reliability, efficiency, and usability of shared GPU resources for developers.
What you’ll do
• Build and maintain GPU workload software on Kubernetes. • Design quotas, scheduling, permissions for shared compute. • Automate deployment and upgrades with Terraform and Helm. • Investigate and fix scheduling, configuration, storage, networking issues. • Benchmark workloads and recommend GPU capacity expansion.
What you’ll bring
• Experience with production Kubernetes GPU platforms. • Strong software engineering in Python, Go or similar. • Experience automating infrastructure deployment. • Linux, networking and storage troubleshooting skills. • Work authorization for US employment.
Life at Shield AI
Offers bonus, benefits, equity, and a supportive equal opportunity workplace.
About Shield AI
Shield AI is a defense-tech company developing intelligent systems to protect service members and civilians worldwide.