principal machine learning AI Infrastructure Engineer tech_leadership · Posted May 14, 2026
$320,000 – $490,000
USD per-year-salary

About this role

Shield AI is hiring a principal-level AI Infrastructure Engineer in the machine learning function based in San Francisco, CA. The posting calls out experience with Reinforcement Learning, Distributed Systems, Machine Learning, MLOps. Compensation is listed at $320,000–$490,000 per-year-salary.

Role
AI Infrastructure Engineer
Function
machine learning
Level
principal
Track
Tech leadership
Employment
Full-time
Location
San Francisco, CA
Department
Hivemind Platform Division
Posted
May 14, 2026

More roles at Shield AI

Engineer I, Electromechanical (R4951)
United States · staff
Agile Compliance
Engineer II, Applications (R4789)
Taipei, Taiwan · mid
API Development
Engineer II, Avionics Thermal Engineering (R4859)
United States · principal
Data Analytics
Engineer II, C++ Systems Integration (R4021)
Washington, DC · mid
Python TypeScript C++
Engineer II, Electrical Failure Analysis (R4593)
Dallas, TX · principal
Ray Security Embedded Systems
All Shield AI jobs →

Job description

from Shield AI careers
Founded in 2015, Shield AI is a venture-backed deep-tech company with the mission of protecting service members and civilians with intelligent systems. Its products include the V-BAT and X-BAT aircraft, Hivemind Enterprise, and the Hivemind Vision product lines. With offices and facilities across the U.S., Europe, the Middle East, and the 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.  

This is an excerpt. Read the full job description on Shield AI careers →
All machine learning jobs machine learning in San Francisco, CA Jobs in San Francisco, CA machine learning salaries machine learning career path
All Shield AI Jobs Browse machine learning roles principal positions