Staff, Platform Engineer AI Compute (6034)
Summary
Platform engineer on Shield AI's AI Factory builds and runs a shared GPU compute platform on Kubernetes for ML training and deployment. Day to day: writing platform software, automating GPU services (Run:ai, Terraform, Helm), tuning scheduling/quotas, troubleshooting workloads, and guiding GPU capacity growth.
Job Description:
Machine-learning engineers need reliable access to GPUs to train and evaluate models. Shield AI’s AI Factory provides shared compute and development tools for these workflows, both for our engineers and for customer deployments.
We are looking for a Platform Engineer to make that compute platform easier to use, more reliable and more efficient. You will write software, improve how teams share GPUs, help developers troubleshoot workloads and use performance data to guide capacity growth. This is a hands-on software and platform engineering role.
What you'll do:
-
Build and maintain the software and configuration that let developers run GPU workloads on Kubernetes.
-
Design quotas, scheduling priorities, permissions and isolation so multiple teams can share compute predictably.
-
Automate deployment and upgrades of GPU services, including Run:ai, with infrastructure as code and tools such as Terraform and Helm.
-
Investigate why jobs wait, fail or run slowly, and fix recurring scheduling, configuration, storage and networking problems.
-
Build diagnostics that connect queue times, GPU utilization, memory requirements and training throughput to the developer’s experience.
-
Benchmark representative workloads and recommend when to add GPU capacity, how much to add and which hardware fits the workload.
-
Deliver repeatable installations and upgrades across cloud, on-premises and air-gapped environments.
Key Outcomes:
-
Developers can get the GPU resources they need through understandable access and resource policies.
-
Teams can diagnose slow or blocked training jobs and distinguish capacity shortages from configuration problems.
-
GPU expansion decisions reflect measured demand and workload performance.
-
Platform changes improve reliability without disrupting existing training workflows.
Required qualifications:
-
Experience building and supporting production Kubernetes platforms, including GPU or distributed computing workloads.
-
Strong software engineering skills in Python, Go or a comparable language.
-
Experience automating infrastructure deployment, configuration and upgrades.
-
Practical Linux, networking and storage troubleshooting skills.
-
Experience with shared resource scheduling, access controls or workload isolation.
-
Ability to work directly with machine-learning engineers, understand their workloads and turn recurring problems into maintained platform capabilities.
Preferred qualifications:
-
Experience with Run:ai or comparable GPU scheduling and resource-management systems.
-
Familiarity with distributed training and tools such as Ray or PyTorch.
-
Experience delivering on-premises or air-gapped platforms, or contributing to open-source infrastructure.
Why Join Us
The AI Factory is the foundation for how Shield AI develops and deploys autonomy.
As a Senior Staff Engineer, you will help build the reference architecture used internally by our engineering teams and delivered to customers building their own AI capabilities. Your work will directly impact how quickly researchers can iterate, how efficiently compute resources are utilized, and how AI systems move from experimentation to operational deployment.
This is an opportunity to work at the intersection of modern AI, distributed systems, Kubernetes, and defense, building a platform that enables the next generation of autonomy systems while leveraging the best of the open-source AI ecosystem.
Skills
As published by lever · 17 questions · 8 written answers
Basics
Which location are you applying for?, Resume/CV, Full name, Email, Phone, Current location, Current company, LinkedIn URL, Twitter URL, GitHub URL, Portfolio URL, Other website
Short answers (1)
- Please state your full legal name:
Pick from a list (8)
- Are you authorized to work in the United States?
- Will you require sponsorship for employment now or in the future?
- Shield AI provides technology, services, and support to the United States military and allied forces with their various missions. If selected for employment with Shield AI, your contributions will directly relate to and/or support these missions. Are there any reasons why you would not be able to fully contribute in support of the U.S. military and allied forces?
- How did you hear about us?
- Are you a transitioning service member?
- If yes, what partnership program are you working with? optional
- Are you local to or willing to relocate? optional
- What office(s) would you be willing to relocate to? (Select all that apply) optional
Written answers (8)
- If yes, what type of sponsorship? optional
- (Optional) If you selected other or would like to expand on the above, feel free to do so below: optional
- If "Industry Conference" or "Other," please provide more details: optional
- If other, please provide further details optional
- When are you available to start your fellowship? optional
- When are you available for full time work? optional
- What full time job(s) are you applying for?
- If you selected international, what location(s)? optional
