Senior Staff Software Engineer, Perception (R4985)

Shield AI; Boston (MA); On-site; USD 234,000 - 351,000; Full time

Benefits offered by this job

Bonus

Equity

Benefits

Job summary

Shield AI, a defense-tech company, seeks an experienced ML engineer to develop and deploy autonomous perception models for U.S. and international defense customers. You will build vision, VLM, and VLA models, collaborate across teams, and optimize deployments on embedded hardware.

You will design data pipelines, conduct supervised fine-tuning, and drive production readiness through robust evaluation and tooling, while advancing state-of-the-art AI systems.

Qualifications

• 10+ years of related experience with a Bachelor's degree.
• 9+ years with a Master's degree.
• 7+ years with a PhD.
• Strong understanding of ML fundamentals.
• Experience training and deploying ML models for computer vision in production.
• Proficiency in C++ and Python.

Responsibilities

• Design, train, fine-tune, and maintain state-of-the-art vision, vision-language, and vision-language-action models.
• Build scalable data pipelines, supervised fine-tuning (SFT) workflows, and evaluation loops.
• Deploy and optimize ML models for embedded hardware using ONNX, TensorRT, and related frameworks.
• Apply ML techniques to perception and autonomy problems across aerial and autonomous systems.
• Translate research into production-ready capabilities with balance between performance and efficiency.
• Collaborate with perception, autonomy, platform, and software teams to integrate ML capabilities.
• Develop benchmarks and evaluation frameworks to measure model performance and guide improvements.
• Improve training infrastructure, tooling, deployment workflows, and lifecycle management.

Skills

Machine learning fundamentals

Strong analytical skills

Ability to obtain SECRET clearance

Education

Bachelor’s degree; or 9 years with Master; or 7 years with PhD

Tools

PyTorch

TensorFlow

ONNX

TensorRT

C++

Python

Job description

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 .

The Hivemind Solutions Perception team develops the next generation of perception capabilities for autonomous systems by combining state-of-the-art machine learning with the proven foundations of computer vision. The team advances how autonomous platforms understand and interpret the world by developing vision, vision-language (VLM), and vision-language-action (VLA) models that tackle core perception challenges such as object understanding, scene interpretation, and mission-relevant environmental awareness. Working at the intersection of research and production, our engineers build the data pipelines, supervised fine-tuning (SFT) workflows, evaluation frameworks, and deployment infrastructure needed to transform cutting-edge AI research into reliable, mission-ready perception capabilities.

In this role, you'll develop and deploy advanced machine learning models that enable autonomous systems to better understand, reason about, and interact with their environment. You'll partner closely with machine learning researchers, autonomy engineers, perception engineers, and platform teams to translate emerging AI capabilities into reliable, production-ready systems for U.S. and international defense customers. This is an ideal opportunity for engineers who enjoy building state-of-the-art AI systems while solving the practical challenges of deploying them on operational autonomous platforms.

What You'll Do

• Model Development – 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.
• Data Pipelines & Model Training – Build scalable data pipelines, supervised fine-tuning (SFT) workflows, and evaluation loops that continuously improve model performance on mission-relevant tasks.
• Model Deployment & Optimization – Deploy and optimize machine learning models for embedded hardware using technologies such as ONNX, TensorRT, and hardware-accelerated inference frameworks.
• Perception & Autonomy Applications – Apply modern machine learning techniques to solve challenging perception and autonomy problems across aerial and other autonomous systems operating in complex, real-world environments.
• Research-to-Production – Translate cutting-edge machine learning research into production-ready capabilities by balancing model performance, robustness, computational efficiency, and operational reliability.
• Cross-functional Collaboration – Partner closely with perception, autonomy, platform, and software engineering teams to integrate machine learning capabilities into mission-ready autonomous systems.
• Model Evaluation & Validation – Develop benchmarks, testing methodologies, and evaluation frameworks to measure model performance, identify failure modes, and guide future improvements.
• Continuous Improvement – Improve training infrastructure, developer tooling, deployment workflows, and model lifecycle management to accelerate experimentation and production delivery.

Required Qualifications

• 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.
• Mastery of machine learning fundamentals.
• Experience training an deploying ML models for computer vision in a production setting.
• 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.
• Proficiency in C++ and Python.
• Strong analytical and problem-solving skills, with the ability to translate research into practical applications.
• Ability to obtain a SECRET clearance

Preferred Qualifications

• Experience with developing autonomous systems for defense customers.
• 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.
• Track record of publications in leading computer vision and robotics conferences and journals (e.g., CVPR, ICCV/ECCV, RAL, ICRA).

$233,760 - $350,640 a year

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.

We may use artificial intelligence (AI) tools to support parts of the hiring process, such as reviewing applications, analyzing resumes, or assessing responses and identifying potential inconsistencies or verification signals in application materials based on available information. These tools assist our recruitment team but do not replace human judgment. Final hiring decisions are ultimately made by humans. If you would like more information about how your data is processed, please contact us.