Staff Engineer, Reinforcement Learning (R3639)

Reposted 12 Days Ago
Be an Early Applicant
San Diego, CA
In-Office
182K-274K Annually
Senior level
Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software
Our mission is to protect service members and civilians with intelligent systems.
The Role
Develop and optimize reinforcement learning models and algorithms for autonomous systems, collaborating with teams and mentoring junior engineers.
Summary Generated by Built In
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 aircraft, Hivemind Enterprise, and the Hivemind Vision product lines. With nine 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:

Founded in 2015, Shield AI is a venture-backed defense technology company with the mission of protecting service members and civilians with intelligent, autonomous systems. Its products include Hivemind Enterprise—EdgeOS, Pilot, Commander, and Forge—as well as V-BAT and Sentient Vision Systems (wide-area motion imaging software). With offices in San Diego, Dallas, Washington, D.C., Abu Dhabi (UAE), Kyiv (Ukraine), and Melbourne (Australia), Shield AI’s technology actively supports U.S. and allied operations worldwide.  For more information, visit www.shield.ai. Follow Shield AI on LinkedIn, X and Instagram.    

The Hivemind Pilot team is an agile group of engineers focused on building a state-of-the-art Autonomy Software Development Kit (SDK) that enables resilient autonomy and intelligence for a wide range of platforms in diverse environments. The Behavior and Motion Planning team in Pilot develops and integrates algorithms that enable robots to make smart decisions and navigate safely. The team also rigorously tests these systems to ensure reliable performance in real-world environments. 

As a member of Behavior and Motion planning team, you will leverage your expertise in robotics and Reinforcement Learning (RL) to develop, deploy, and optimize models for autonomous systems that operate in complex, real-world environments. You will collaborate with cross-functional teams to deliver robust, scalable solutions that advance the state of Hivemind SDK. You will also contribute to developing technical requirements, test plans, and validating the performance of algorithms and models. 

What You'll Do:

  • Design, implement, and deploy reinforcement learning algorithms for a variety of platforms 
  • Collaborate with teams across the organization to integrate RL solutions that meet customer specifications 
  • Analyze and optimize performance of deployed RL models in dynamic environments  
  • Develop tools and infrastructure to support large-scale training, simulation, and evaluation  
  • Mentor and provide technical guidance to junior engineers  
  • Stay current with the latest advancements in RL, and apply them to solve challenging problems  
  • Contribute to the design and architecture of scalable, maintainable software systems 

Required Qualifications:

  • Master's degree in Computer Science, Robotics, or a related field and 5+ years of relevant professional experience or PhD with 4+ years of relevant experience. 
  • Familiarity with prototyping in Python is welcome, but this role demands professional C++ production deployment skills. Candidates whose primary experience is in Python are unlikely to find this position a good fit. 
  • Demonstrated experience deploying reinforcement learning algorithms in production environments  
  • Strong background deploying RL algorithms in production following full Software development lifecycle 
  • Ability to independently deploy high-reliability code suitable for real-world autonomous systems 
  • Experience with RL frameworks (e.g., TensorFlow (C++), libtorch, etc.) and RL training environments (e.g., OpenAI Gymnasium, Google DeepMind Control Suite, etc.) 
  • Solid understanding of software engineering best practices, including version control, testing, and CI/CD  
  • Familiarity with CUDA 
  • Excellent problem-solving and communication skills 

Preferred Qualifications:

  • Peer reviewed publications related to RL  
  • Strong background in Robotics or autonomous systems 
  • Experience with multi-agent RL or distributed RL systems  
  • Familiarity with simulation environments (e.g. Isaac Sim, MuJoCo)  
  • Experience with cloud-based training and deployment  
  • Experience working in aviation, or other safety-critical domains 


#LD
#LI-ED1 

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. 

Top Skills

C++
Cuda
Google Deepmind Control Suite
Libtorch
Openai Gymnasium
Python
TensorFlow
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
0 Employees
Year Founded: 2015

What We Do

Founded in 2015, Shield AI is a venture-backed defense technology company with the mission of protecting service members and civilians with intelligent systems. Its products include the V-BAT aircraft, Hivemind Enterprise, and the Hivemind Vision product lines. With offices in San Diego, Dallas, Washington, D.C., Boston, Abu Dhabi (UAE), Kyiv (Ukraine), and Melbourne (Australia), Shield AI’s technology actively supports U.S. and allied 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

Similar Jobs

EchoStar Logo EchoStar

Data Scientist

Aerospace • Cloud • Digital Media • Information Technology • Mobile • News + Entertainment • Retail
In-Office
Foster City, CA, USA
83K-137K Annually

ServiceNow Logo ServiceNow

Product Manager

Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Remote or Hybrid
San Diego, CA, USA
127K-215K Annually

ServiceNow Logo ServiceNow

Senior Software Engineer

Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Remote or Hybrid
San Diego, CA, USA
127K-239K Annually

ServiceNow Logo ServiceNow

Accessibility Specialist

Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Remote or Hybrid
Santa Clara, CA, USA
188K-328K Annually

Similar Companies Hiring

Credal.ai Thumbnail
Software • Security • Productivity • Machine Learning • Artificial Intelligence
Brooklyn, NY
Standard Template Labs Thumbnail
Software • Information Technology • Artificial Intelligence
New York, NY
10 Employees
PRIMA Thumbnail
Travel • Software • Marketing Tech • Hospitality • eCommerce
US
15 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account