Senior AI Engineer, Perception & Autonomy

F7d8749a; Washington (District of Columbia); On-site; USD 180,000 - 240,000; Full time

Benefits offered by this job

Bonus

Benefits

Equity

Job summary

Shield AI seeks an experienced ML engineer to design, train, and deploy vision models for autonomous systems. You’ll build data pipelines, SFT workflows, and evaluation tools to turn cutting-edge AI research into production-ready perception capabilities for defense customers.

Applicants should have strong CV expertise, 3D vision skills, and experience with PyTorch/TensorFlow, TensorRT, and ONNX. Proficiency in C++ and Python is required, and obtaining a SECRET clearance is necessary.

Qualifications

• Mastery of machine learning fundamentals.
• Experience training and deploying ML models for computer vision in a production setting.
• Strong understanding of 3D vision problems/algorithms.
• Experience with ML frameworks such as PyTorch and TensorFlow.
• Demonstrated expertise in deploying models using TensorRT and ONNX.
• Proficiency in C++ and Python.
• Ability to obtain a SECRET clearance.

Responsibilities

• Design, train, fine-tune, and maintain state-of-the-art vision, vision-language, and vision-language-action models for autonomous systems.
• Build scalable data pipelines, supervised fine-tuning (SFT) workflows, and evaluation loops to improve model performance on mission-relevant tasks.
• Deploy and optimize ML models for embedded hardware using ONNX, TensorRT, and hardware-accelerated inference frameworks.
• Collaborate with perception, autonomy, platform, and software teams to translate research into production-ready capabilities.

Skills

ML fundamentals

Computer vision in production

3D vision

PyTorch

TensorFlow

C++

Python

TensorRT

ONNX

SECRET clearance

Education

Bachelor's degree

Master's degree

PhD

Tools

TensorRT

ONNX

Job description

Shield AI seeks an experienced ML engineer to design, train, and deploy vision models for autonomous systems. You’ll build data pipelines, SFT workflows, and evaluation tools to turn cutting-edge AI research into production-ready perception capabilities for defense customers.

Applicants should have strong CV expertise, 3D vision skills, and experience with PyTorch/TensorFlow, TensorRT, and ONNX. Proficiency in C++ and Python is required, and obtaining a SECRET clearance is necessary.