Staff Engineer, Hivemind Reasoning (R6292)

Posted 2 Days Ago
Be an Early Applicant
San Mateo, CA, USA
In-Office
240K-360K Annually
Senior level
Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software • Defense Technology
Our mission is to protect service members and civilians with intelligent systems.
The Role
Design, train, adapt, and evaluate reasoning and cognition models for unmanned autonomous systems. Develop multimodal, embodied, and traditional AI/ML capabilities that operate under edge constraints, incomplete information, intermittent communications, and large multi-agent deployments. Establish reproducible training workflows, benchmarks, simulation and real-world evaluations, and model feedback loops. Partner with perception, MLOps, simulation, autonomy, hardware, and inference teams to deliver reliable models and communicate technical tradeoffs and performance evidence.
Summary Generated by Built In
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, V-BAT and X-BAT aircraft, and Aechelon simulation and synthetic reality technologies. 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:

Help shape the future of physical AI in defense by building reasoning models for unmanned autonomous systems across aerial, maritime, land, and space domains. 

We’re looking for a deeply technical engineer who can turn emerging research into reliable autonomous capabilities and shares our conviction that AI can protect service members and civilians. As a Staff Engineer on the Hivemind Reasoning team, you’ll drive the design, training, adaptation, and evaluation of reasoning and cognition models intended for deployment on unmanned platforms. 

You’ll take hands-on ownership of complex technical problems, from defining model architectures and training objectives to establishing rigorous evidence of mission effectiveness. Your work will support individual vehicles and coordinated teams of hundreds of agents, while accounting for limited compute, intermittent communications, incomplete information, and other constraints of operating at the edge in contested environments. 

You’ll partner closely with teams in perception, MLOps, large-scale simulation, autonomy, and inference to deliver capabilities as part of Hivemind Enterprise.  

What you'll do:

  • Design, implement, train, adapt, and evaluate reasoning and cognition models for unmanned autonomous platforms. 
  • Make and validate tradeoffs among foundation-model approaches, including vision-language models (VLMs) and vision-language-action models (VLAs), specialized deep-learning models, and traditional AI/ML methods. 
  • Develop capabilities for interpreting context, reasoning over time, coordinating among agents, and adapting to incomplete or changing information. 
  • Build reproducible training and adaptation workflows using appropriate combinations of self-supervised learning, supervised fine-tuning, imitation learning, reinforcement learning, and distillation. 
  • Define data requirements, training objectives, benchmarks, and evaluation methods that support meaningful comparisons among model approaches. 
  • Establish rigorous evaluations in simulation and with real-world data, measuring generalization, robustness, uncertainty, and performance as team size and operating conditions change. 
  • Incorporate edge constraints into model architecture and training decisions in partnership with hardware, autonomy, and inference teams. 
  • Diagnose model failures and create feedback loops among data curation, training experiments, and re-evaluation. 
  • Deliver trained models, reproducible results, and clear performance evidence to partner teams. 
  • Communicate technical tradeoffs, experimental findings, and model limitations clearly to research, engineering, product, and organizational leadership. 

Required qualifications:

  • BS/MS in Computer Science, Electrical Engineering, Mechanical Engineering, Aerospace Engineering, and/or similar degree, or equivalent practical experience. Typically requires a minimum of 7 years of related experience with a Bachelor’s degree; or 6 years and a Master’s degree; or 4 years with a PhD; or equivalent work experience. 
  • A demonstrated track record of developing and deploying advanced ML models in production environments, with substantial hands-on contributions to model design, training, and evaluation. 
  • Deep expertise in foundation models, modern deep learning, model architectures, and the training ecosystem. 
  • Strong command of training and adaptation approaches such as self-supervised learning, supervised fine-tuning, imitation learning, reinforcement learning, and distillation. 
  • Experience defining datasets, training objectives, experimental workflows, benchmarks, and reproducible criteria for comparing models.
  • Fluency in traditional AI/ML techniques and an ability to combine them thoughtfully with learned models. 
  • Experience with training frameworks such as PyTorch, JAX, TensorFlow, etc.
  • Experience with multimodal, embodied, robotic, or autonomous-system models; familiarity with VLA models and world models is particularly relevant. 
  • Familiarity with traditional robotics algorithms, e.g. path planning, probabilistic modeling, optimization approaches, etc.
  • Familiarity with NVIDIA Jetson and other edge compute platforms (e.g. Qualcomm, AMD, etc.).

Preferred qualifications:

  • MS or PhD in Computer Science, Electrical Engineering, or similar degree. 
  • Expertise in optimising and deploying ML models to resource-constrained hardware platforms. 
  • Experience with simulation-to-real on autonomous systems. 
  • Experience with unmanned autonomous systems in defense applications.
  • Familiarity with large, distributed systems.
  • Familiarity with high-fidelity simulation, e.g. rigid body dynamics, sensor & actuator models, etc. 

#LI-DM2
#LD

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

  • Bachelor’s or master’s degree in Computer Science, Electrical Engineering, Mechanical Engineering, Aerospace Engineering, or a similar field, or equivalent practical experience.
  • Typically 7 years of related experience with a bachelor’s degree, 6 years with a master’s degree, or 4 years with a PhD, or equivalent work experience.
  • Track record of developing and deploying advanced machine learning models in production, with hands-on contributions to model design, training, and evaluation.
  • Deep expertise in foundation models, modern deep learning, model architectures, and the training ecosystem.
  • Strong command of self-supervised learning, supervised fine-tuning, imitation learning, reinforcement learning, and distillation.
  • Experience defining datasets, training objectives, experimental workflows, benchmarks, and reproducible model comparison criteria.
  • Fluency in traditional artificial intelligence and machine learning techniques and combining them with learned models.
  • Experience with training frameworks such as PyTorch, JAX, or TensorFlow.
  • Experience with multimodal, embodied, robotic, or autonomous-system models.
  • Familiarity with vision-language-action models and world models.
  • Familiarity with traditional robotics algorithms, including path planning, probabilistic modeling, and optimization.
  • Familiarity with NVIDIA Jetson and other edge compute platforms such as Qualcomm or AMD platforms.
  • Master’s or PhD in Computer Science, Electrical Engineering, or a similar field.
  • Expertise optimizing and deploying machine learning models to resource-constrained hardware.
  • Experience with simulation-to-real transfer on autonomous systems.
  • Experience with unmanned autonomous systems in defense applications.
  • Familiarity with large, distributed systems.
  • Familiarity with high-fidelity simulation, including rigid body dynamics and sensor and actuator models.

What the Team is Saying

Dylan
Mo
Willy
Michael
Vibhav
Kirby
Ryan
Evan

Shield AI Compensation & Benefits Highlights

  • Healthcare Strength — Company materials highlight comprehensive medical, dental, and vision coverage with mental-health support and an EAP. Third-party benefits summaries also portray health coverage as a strong component of the package.
  • Retirement Support — Offerings include a 401(k) with employer matching, cited on the careers page and independent benefits directories. Multiple sources indicate a match exists, though exact formulas have varied over time and warrant confirmation.
  • Equity Value & Accessibility — Company communications describe equity for all full-time hires and stock benefits as part of total rewards. Independent listings also note company equity/RSUs as a standard element.

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

Engineer I, Mechanical (R6302)

Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software • Defense Technology
In-Office
2 Locations
81K-140K Annually

Shield AI Logo Shield AI

Engineer II, Automated Test (R6288)

Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software • Defense Technology
In-Office
San Mateo, CA, USA
110K-160K Annually

Shield AI Logo Shield AI

Manager, Infrastructure (R6154)

Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software • Defense Technology
In-Office
San Mateo, CA, USA
160K-250K Annually

Shield AI Logo Shield AI

Principal Engineer

Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software • Defense Technology
In-Office
San Diego, CA, USA
220K-330K Annually

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account