jobapplier

Senior Staff Software Engineer (Perception)

Shield AI

Skills

Required

  • machine learning
  • computer vision
  • TensorRT
  • 3D vision problems/algorithms
  • machine learning frameworks
  • PyTorch
  • TensorFlow
  • deploying models
  • training an
  • ML
  • C++
  • Python

Preferred

  • vision-language-action models
  • vision-language models
  • open-source
  • Build scalable data pipelines
  • SFT
  • embedded hardware
  • hardware-accelerated inference frameworks

Join the Hivemind Solutions Perception team as a Senior Staff Software Engineer. In this role, you will develop and deploy advanced machine learning models that enhance the perception and decision-making capabilities of autonomous systems. You will work closely with machine learning researchers, autonomy engineers, perception engineers, and platform teams to translate emerging AI capabilities into reliable, production-ready systems for defense customers. This position offers excellent benefits, including medical coverage, mental health support, paid parental leave, and more.

Track record of publications in leading computer vision and robotics conferences and journals (e.g., CVPR, ICCV/ECCV, RAL, ICRA)

Responsibilities

  • Design, train, fine-tune, and maintain state-of-the-art vision, vision-language, and vision-language-action models that improve perception and decision-making for autonomous systems.
  • Build scalable data pipelines, supervised fine-tuning (SFT) workflows, and evaluation loops that continuously improve model performance on mission-relevant tasks.
  • Deploy and optimize machine learning models for embedded hardware using technologies such as ONNX, TensorRT, and hardware-accelerated inference frameworks.

Requirements

  • Ability to obtain a SECRET clearance
  • Typically requires a minimum of 10 years of related experience with a Bachelor’s degree; or 9 years and a Master’s degree; or 7 years with a PhD; or equivalent work experience

Strong understanding of 3D vision problems/algorithms

  • Experience with machine learning frameworks such as PyTorch and TensorFlow
  • Demonstrated expertise in deploying models using TensorRT and ONNX
  • Experience training an deploying ML models for computer vision in a production setting
  • Strong analytical and problem-solving skills, with the ability to translate research into practical applications
  • Mastery of machine learning fundamentals
  • Proficiency in C++ and Python
  • Experience with developing autonomous systems for defense customers

Preferred

  • Experience with training/finetuning vision-language models, vision-language-action models, and/or world models
  • Contributions to open-source projects in machine learning or computer vision

Pay

USD 233,760 - 350,640

Every job at Shield AI