Principal Engineer, AI Infrastructure (R4941)

Posted 24 Days Ago
Be an Early Applicant
San Francisco, CA, USA
In-Office
320K-490K Annually
Expert/Leader
Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software
Our mission is to protect service members and civilians with intelligent systems.
The Role
Lead design and operation of a centralized AI and data platform for autonomy: define compute strategy, build distributed training and simulation systems, manage multi-modal datasets and MLOps, enable model evaluation, optimization, and deployment to constrained edge and customer environments, and drive platform standardization and broad adoption across programs.
Summary Generated by Built In
Founded in 2015, 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 and V-BAT and X-BAT aircraft. 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. 

Job Description:

Shield AI builds autonomy systems for defense applications, including air, maritime, and space platforms operating in complex and contested environments.  

We are establishing a centralized AI and Data Platform organization responsible for the infrastructure that underpins autonomy development across Hivemind and other programs. This team owns the systems used to train models, run simulation, manage data, and deploy models to operational environments.  

We are seeking a Principal Engineer that will scale an initial architecture into a platform that supports multiple autonomy programs.  

Success in this role requires disciplined execution, delivering fast iteration for engineering teams while maintaining reliability, cost control, and architectural consistency as the system scales.  

The Principal Engineer is accountable for ensuring engineers can move efficiently from idea to trained model to deployed capability, and that infrastructure decisions reflect the realities of the domain, including simulation-driven development, continuously evolving multi-modal sensor data, and deployment to constrained and reliability-critical systems.  

This role spans the full lifecycle of autonomy development, training foundation models, running large-scale and multi-fidelity simulation, managing training data, evaluating models, and deploying optimized models to edge systems.  

A key part of this role is defining how these capabilities extend beyond internal use. This includes establishing how Shield AI delivers AI infrastructure in customer environments across on-premise, cloud, hybrid, and sovereign or nationally constrained environments.  

What you'll do:

  • Platform Ownership: Define and operate the core AI and data platform across training, simulation, data management, evaluation, and deployment. 
  • Compute Strategy and Infrastructure:Own where and how workloads run across on-premise, cloud, and hybrid environments. Drive capacity planning, utilization, and cost-per-compute decisions, including support for classified and air-gapped systems 
  • Training and Simulation Systems:Build infrastructure for distributed training (supervised learning, RL/MARL, foundation models) and large-scale, multi-fidelity simulation. Ensure training and simulation systems operate together without bottlenecks. 
  • Data Platform: Ingest and manage multi-modal sensor data (EO, IR, radar, EW, IMU). Establish dataset versioning, data lineage, feature storage, data cataloging, and classification-aware storage and access controls. 
  • MLOps, Evaluation, and Model Lifecycle:Establish a consistent workflow for experiment tracking, model registry, artifact provenance, and automated validation. Implement evaluation and V&V gates so models meet defined standards before deployment. 
  • Deployment and Operational Feedback:Own the pipeline from training to deployment, including model optimization (e.g., distillation, quantization, pruning), deployment to edge systems, monitoring, drift detection, and retraining triggers. 
  • Customer AI Infrastructure:Define how AI infrastructure is deployed in customer environments across on-premise, cloud, hybrid, and sovereign settings. Establish a consistent approach that avoids one-off solutions while adapting to operational constraints. 
  • Platform Standardization:Define common tools, interfaces, and workflows across teams. Reduce duplication while maintaining flexibility where needed. 
  • Cross-Team Partnership: Work directly with Hivemind and other autonomy teams to ensure the platform supports real workloads and evolves with program needs.

Key Outcomes:

  • Faster iteration from idea to trained model to evaluated result 
  • High utilization of compute resources with clear visibility into usage and cost 
  • Simulation capacity that supports large-scale training without bottlenecks 
  • Consistent end-to-end lifecycle: development, evaluation, deployment, monitoring, and retraining 
  • Repeatable data loop: telemetry, scenario extraction, retraining, and redeployment  
  • Reliable deployment of optimized models to edge systems 
  • Broad platform adoption across autonomy programs 
  • Repeatable approach for deploying AI infrastructure in customer environments  
  • Representative performance targets: 

  • Training iteration cycles measured in days, not weeks 
  • Sustained high utilization of GPU resources under production workloads  

Required qualifications:

  • Experience building and operating ML infrastructure at scale (100+ GPU clusters, distributed systems) 
  • Experience defining compute strategy, including on-premise vs cloud tradeoffs, capacity planning, and cost management 
  • Strong understanding of ML workloads, including foundation models, RL/MARL, simulation-based training, and fine-tuning 
  • Experience building data platforms with dataset versioning, lineage, and cataloging 
  • Ability to debug and resolve system issues when needed  

Preferred qualifications:

  • Experience in defense or classified environments (e.g., air-gapped systems, SCIFs) 
  • Experience with simulation-heavy ML systems (robotics, autonomy, or similar domains) 
  • Experience deploying and optimizing models for edge hardware 
  • Familiarity with HPC systems (schedulers, parallel storage, high-speed networking)  

Why Join Us

    You will define the infrastructure that supports the development and deployment of autonomy systems across Shield AI.  

    This role establishes the foundation for how models are trained, evaluated, and deployed, and directly impacts how quickly new capabilities are delivered into operational environments.  

    You will have ownership over systems and decisions that are often distributed across multiple teams at other organizations, with the opportunity to shape how AI infrastructure is built and used both internally and in customer environments.  

     

#LI-DM2
#LF

Full-time regular employee offer package:
Pay within range listed + Bonus + Benefits + Equity
 
Temporary employee offer package:
Pay within range listed above + temporary benefits package (applicable after 60 days of employment)
 
Salary compensation is influenced by a wide array of factors including but not limited to skill set, level of experience, licenses and certifications, and specific work location. All offers are contingent on a cleared background and possible reference check. Military fellows and part-time employees are not eligible for benefits. Please speak to your talent acquisition representative for more information.
 
###
 
Shield AI is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, marital status, disability, gender identity or Veteran status. If you have a disability or special need that requires accommodation, please let us know. 

Skills Required

  • Experience building and operating ML infrastructure at scale (100+ GPU clusters, distributed systems)
  • Experience defining compute strategy including on-premise vs cloud tradeoffs, capacity planning, and cost management
  • Strong understanding of ML workloads including foundation models, RL/MARL, simulation-based training, and fine-tuning
  • Experience building data platforms with dataset versioning, lineage, and cataloging
  • Ability to debug and resolve system issues when needed
  • Experience in defense or classified environments (e.g., air-gapped systems, SCIFs)
  • Experience with simulation-heavy ML systems (robotics, autonomy, or similar domains)
  • Experience deploying and optimizing models for edge hardware (distillation, quantization, pruning)
  • Familiarity with HPC systems (schedulers, parallel storage, high-speed networking)

What the Team is Saying

Dylan
Mo
Willy
Michael
Vibhav
Kirby
Ryan
Evan

Shield AI Compensation & Benefits Highlights

  • Healthcare Strength Company materials describe excellent medical, dental, and vision coverage alongside a mental‑health EAP. Site perks such as an onsite gym in DC and a gym discount in San Diego support a health‑focused offering.
  • Retirement Support Careers materials highlight a 401(k) with company match as part of the standard package. A Total Rewards overview emphasizes retirement features within a broader, transparent compensation view.
  • Career-Linked Recognition & Rewards Compensation for in‑demand technical and senior go‑to‑market roles is described as competitive, with visible engineering ranges and top‑end packages. This points to meaningful upside tied to role, level, and scarce skills.

Shield AI Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: San Diego, CA
Year Founded: 2015

What We Do

At Shield AI, you won't wait years to see your work reach the field. You'll build hardware and software that operates in the real world right now, in the hands of the people who depend on it. Hivemind, our AI pilot, has been flying since 2018. It has flown more than 30 platforms, including an F-16, and it now sits under a U.S. Air Force production contract for Collaborative Combat Aircraft. When you write code or shape a system here, you contribute to technology with a proven flight record and a clear production future. V-BAT flies intelligence, surveillance, and reconnaissance missions with an operational record that stretches from Ukraine to the Indo-Pacific. It delivers eyes where they matter most, in the most demanding conditions on earth. The teams behind it watch their work get tested where the stakes are real. X-BAT takes its first flight this year. It's an AI-piloted fighter that needs no runway, built to operate where traditional aircraft can't. Join now and you help shape a program at its earliest, most formative stage. That's the kind of ground-floor work that defines a career. Do the most impactful work of your life, on problems that matter. Autonomy at this level asks a lot of you. You'll take on problems in perception, planning, and control that few teams anywhere are equipped to solve. You'll work across disciplines, from aerospace and robotics to machine learning and systems engineering, alongside people who hold themselves to an exacting standard and expect the same from you. Our mission is clear: protect service members and civilians with intelligent systems. That purpose runs through every decision, every design review, and every deployment. It's why the work here carries a weight you can feel. Ready to join our mission? Explore our open roles and find where you fit.

Why Work With Us

Founded in 2015 by a former Navy SEAL, Shield AI builds AI pilots and uncrewed aircraft. Veterans aren't an afterthought here, they're at every level. It's why the work carries weight: AI pilots and uncrewed aircraft flying real missions, from Ukraine to the Indo-Pacific, protecting service members and civilians.

Gallery

Gallery
Gallery
Gallery

Shield AI Teams

Team
Aircraft
Team
Growth (Sales and Marketing)
Team
Hivemind
About our Teams

Shield AI Offices

Hybrid Workspace

Employees engage in a combination of remote and on-site work.

Typical time on-site: Flexible
HQSan Diego, CA
United Arab Emirates
Arlington, VA
Company Office Image
Frisco, Texas
Ukraine
London
Melbourne, Victoria
Taiwan, Province of China
Waltham, MA
Learn more

Similar Jobs

Shield AI Logo Shield AI

Senior Engineer

Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software
In-Office or Remote
3 Locations
160K-290K Annually

Shield AI Logo Shield AI

Mechanical Engineer I, Systems Integration (R5344)

Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software
In-Office or Remote
3 Locations
75K-110K Annually

Shield AI Logo Shield AI

Associate Facilities Coordinator (R5454)

Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software
In-Office
San Francisco, CA, USA

Shield AI Logo Shield AI

Staff Engineer

Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software
In-Office
San Diego, CA, USA
230K-350K Annually

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account