Product Manager, AI/ML & Foundation Models (R4991)

Posted 7 Hours Ago
2 Locations
In-Office
190K-290K Annually
Senior level
Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software
Our mission is to protect service members and civilians with intelligent systems.
The Role
Lead product vision and roadmap for the Hivemind AI Platform to enable training, evaluation, governance, and safe edge deployment of foundation and domain models. Drive MLOps, data/simulation pipelines, synthetic data generation, model governance, and integration with edge hardware. Partner cross-functionally to productionize research, ensure accreditation and safety, and support continuous learning and sovereign AI factory capabilities for mission-critical autonomy.
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:
 
The Product Manager will drive the strategy and execution of Shield AI’s next-generation autonomy intelligence stack—enabling customers and internal teams to train, evaluate, and deploy foundation and domain models that power resilient autonomy at the edge. This PM owns the product vision and roadmap for the Hivemind AI Platform (Forge, training pipelines, data infrastructure, evaluation, and deployment toolchains), ensuring we can manufacture, govern, and field advanced world models, robotics foundation models, and vision-language-action systems safely and at scale. 
 
This role sits at the intersection of AI/ML, autonomy, model lifecycle, infrastructure, and product strategy. The PM partners closely with engineering, AI research, Hivemind Solutions, and field teams to deliver the tooling that enables sovereign autonomy, AI Factories at the edge, and continuous learning—capabilities that are central to Shield AI’s strategic direction. 
 
This is a high-impact role for an experienced product leader excited to define how foundation models are trained, validated, governed, and deployed across thousands of autonomous systems in highly contested environments.

What you'll do:

  • AI Model Development & Training Platform
  • Own the roadmap for foundation model training workflows, including dataset ingestion, curation, labeling, synthetic data generation, domain model training, and distillation pipelines.
  • Define requirements for world models, robotics models, and VLA-based training, evaluation, and specialization.
  • Lead the evolution of MLOps capabilities in Forge, including data lineage, experiment tracking, model versioning, and scalable evaluation suites.
  • Data, Simulation & Synthetic Data Factory
  • Define product requirements for synthetic data generation, simulation-integrated data flywheels, and automated scenario generation.
  • Partner with Digital Twin, Simulation, and autonomy teams to convert natural-language mission inputs into data needs, training procedures, and model variants.
  • Safe Deployment & Model Governance
  • Lead the development of model governance and auditability tooling, including model cards, dataset rights, lineage tracking, safety gates, and compliance evidence.
  • Build guardrails and workflows to safely deploy models onto edge hardware in disconnected, GPS- or comms-denied environments.
  • Partner with Safety, Certification, Cyber, and Engineering teams to ensure traceability and evaluation pipelines meet operational and accreditation requirements.
  • Edge Deployment & AI Factory Integration
  • Partner with Pilot, EdgeOS, and hardware teams to integrate foundation-model-based perception and reasoning into autonomy behaviors.
  • Define requirements for distillation, quantization, and inference tooling as part of the “three-computer” development and deployment model.
  • Ensure closed-loop workflows between cloud model training and edge-native execution.
  • Cross-Functional Leadership
  • Collaborate with Engineering, Research, Product, Customer Engagement, and Solutions teams to ensure model outputs meet mission and platform constraints.
  • Translate advanced AI capabilities into intuitive workflows that platform OEMs and partner nations can use to build sovereign AI factories.
  • Sequence foundational capabilities that unblock autonomy, simulation, and customer-facing product teams.
  • User & Customer Impact
  • Develop deep empathy for ML engineers, autonomy developers, and Solutions engineers who rely on the platform.
  • Capture operational data gaps, mission-driven model needs, and domain-specific specialization requirements.
  • Lead demos and onboarding for model-development capabilities across internal and external teams.

Required qualifications:

  • 7+ years of experience in product management or highly technical ML/AI product roles.
  • 2+ years of experience in a hands-on software development role.
  • Strong engineering background (Computer Science, Electrical Engineering, Robotics, or related field).
  • Deep understanding of foundation models, robotics models, multimodal models, MLOps, and training infrastructure.
  • Experience managing complex products spanning data pipelines, cloud training clusters, model governance, and edge deployments.
  • Proven success partnering with research teams to transition ML innovations into stable, production-grade workflows.
  • Familiarity with simulation-based data generation and large-scale data management.
  • Excellent communicator with strong cross-functional leadership skills.

Preferred qualifications:

  • Experience working on autonomy, robotics, embedded AI, or mission-critical systems.
  • Hands-on familiarity with GPU infrastructure, distributed training, or data lakehouse architectures.
  • Experience supporting defense, dual-use, or safety-critical AI systems.
  • Background designing or operating AI Factory–style pipelines (data → training → evaluation → distillation → edge deployment).
  • Advanced degree in engineering, ML/AI, robotics, or a related field.

#LI-DM2
#LE

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

  • 7+ years of experience in product management or highly technical ML/AI product roles.
  • 2+ years of experience in a hands-on software development role.
  • Strong engineering background (Computer Science, Electrical Engineering, Robotics, or related field).
  • Deep understanding of foundation models, robotics models, multimodal models, MLOps, and training infrastructure.
  • Experience managing complex products spanning data pipelines, cloud training clusters, model governance, and edge deployments.
  • Proven success partnering with research teams to transition ML innovations into stable, production-grade workflows.
  • Familiarity with simulation-based data generation and large-scale data management.
  • Excellent communicator with strong cross-functional leadership skills.
  • Experience working on autonomy, robotics, embedded AI, or mission-critical systems.
  • Hands-on familiarity with GPU infrastructure, distributed training, or data lakehouse architectures.
  • Experience supporting defense, dual-use, or safety-critical AI systems.
  • Background designing or operating AI Factory-style pipelines (data -> training -> evaluation -> distillation -> edge deployment).
  • Advanced degree in engineering, ML/AI, robotics, or a related field.

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: Port Melbourne, Victoria
Year Founded: 2015

What We Do

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.

Why Work With Us

What makes Shield AI special is our people. We unlock the power of autonomy, and in the face of overwhelming odds and challenges, we find ways to win and make a difference for our customers. We bring together software, AI, and aerospace engineering disciplines to deploy the most intelligent aviation capabilities in the world.

Gallery

Gallery

Shield AI Offices

Hybrid Workspace

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

Typical time on-site: 3 days a week
HQMelbourne, Victoria
HQSan Diego, CA
Abu Dhabi, AE
Arlington, VA
Boston, US
Frisco, US
Kyiv, UA
Melbourne, AU
Oslo, NO
Prosper, TX
Taipei City, TW
Taipei City, TW
Waltham, MA
Washington, US
Learn more

Similar Jobs

Shield AI Logo Shield AI

Senior Staff Verification and Validation Engineer (R4835)

Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software
In-Office
2 Locations
170K-240K Annually

Shield AI Logo Shield AI

Senior Engineer

Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software
In-Office
2 Locations
95K-200K Annually

Shield AI Logo Shield AI

Staff Software Engineer

Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software
In-Office
3 Locations
188K-281K Annually

Shield AI Logo Shield AI

Marketing Manager

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

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account