Earth Is Our Runway
Shield AI Logo

Shield AI

Staff Engineer, Hivemind Reasoning (R6292)

Posted 59 Minutes Ago
Be an Early Applicant
In-Office
San Mateo, CA, USA
240K-360K Annually
Senior level
In-Office
San Mateo, CA, USA
240K-360K Annually
Senior level
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.
The summary above was generated by AI
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. 

Similar Jobs at Shield AI

An Hour Ago
In-Office
120K-190K Annually
Senior level
120K-190K Annually
Senior level
Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software • Defense Technology
Develop application security policies and integrate security scanning tools into CI/CD pipelines. Review findings, partner with developers to remediate vulnerabilities, perform threat modeling, and validate secure-by-default architectures and coding standards. The role also involves software analysis, secure coding, cryptography, DevSecOps practices, and collaboration across engineering teams.
Top Skills: CC++Ci/CdCryptographyCve ScanningDependency MappingDynamic AnalysisGitlabGoLinuxPythonStatic AnalysisThreat Modeling
An Hour Ago
In-Office
San Mateo, CA, USA
290K-430K Annually
Senior level
290K-430K Annually
Senior level
Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software • Defense Technology
Leads a team developing reasoning and cognition models for unmanned autonomous systems across multiple domains. Defines model roadmaps, training methods, datasets, evaluations, and deployment strategies for edge environments. Oversees research, model adaptation, simulation and real-world testing, multi-agent coordination, and performance optimization. Recruits and develops AI/ML engineers while communicating technical tradeoffs and delivering production-ready models to partner teams.
Top Skills: AmdJaxNvidia JetsonPyTorchQualcommTensorFlowVlasVlms
3 Hours Ago
In-Office
180K-270K Annually
Senior level
180K-270K Annually
Senior level
Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software • Defense Technology
Leads planning, execution, and delivery of complex software releases and autonomous-systems initiatives for the Hivemind Enterprise platform. Coordinates engineering, product, design, sponsors, and customer stakeholders; manages schedules, budgets, resources, risks, and technical tradeoffs; improves program-management processes; and ensures high-quality, on-time delivery. The role requires strong software-development knowledge, technical decision-making, communication, leadership, and experience managing integrated hardware/software programs.
Top Skills: Artificial IntelligenceAutonomous SystemsEnterprise SoftwareHivemind EnterpriseIntegrated Hardware/Software SystemsJIRAMachine LearningRobotics

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account