Senior Staff ML Engineer — Vision & Autonomy Systems

Shield AI; Boston (MA); On-site; USD 233,760 - 350,640; Full time

Benefits offered by this job

Bonus

Equity

Benefits

Job summary

Shield AI is seeking a highly experienced machine learning leader to design, train, and maintain cutting-edge vision and vision-language models for autonomous systems. You will build scalable data pipelines, conduct supervised fine-tuning, and deploy models to embedded hardware using TensorRT and ONNX, driving performance and reliability across real-world missions.

The role requires deep expertise in 3D vision, proficiency with PyTorch and TensorFlow, and the ability to translate research into

Qualifications

• Minimum 10 years of related experience with a Bachelor’s degree, or 9 years with a Master’s, or 7 years with a PhD.
• 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 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

Responsibilities

• Model Development – Design, train, fine-tune, and maintain state-of-the-art vision, vision-language, and vision-language-action models.
• Data Pipelines & Model Training – Build scalable data pipelines, supervised fine-tuning workflows, and evaluation loops.
• Model Deployment & Optimization – Deploy and optimize ML models for embedded hardware using TensorRT and ONNX.
• Perception & Autonomy Applications – Apply ML techniques to perception and autonomy problems across autonomous systems.
• Research-to-Production – Translate ML research into production-ready capabilities balancing performance and efficiency.
• Cross-functional Collaboration – Partner with perception, autonomy, platform, and software teams.
• Model Evaluation & Validation – Develop benchmarks and evaluation frameworks to guide improvements.
• Continuous Improvement – Improve training infrastructure and deployment workflows to accelerate delivery.

Skills

Machine Learning fundamentals

PyTorch

TensorFlow

Python

C++

Analytical skills

Clearance (SECRET)

Education

Bachelor’s degree or higher in a related field

Tools

TensorRT

ONNX

Job description

Shield AI is seeking a highly experienced machine learning leader to design, train, and maintain cutting-edge vision and vision-language models for autonomous systems. You will build scalable data pipelines, conduct supervised fine-tuning, and deploy models to embedded hardware using TensorRT and ONNX, driving performance and reliability across real-world missions.

The role requires deep expertise in 3D vision, proficiency with PyTorch and TensorFlow, and the ability to translate research into