Senior Staff AI Engineer — Vision, VLM & Autonomy

Shield AI; Washington; On-site; USD 130,000 - 190,000; Full time

Benefits offered by this job

Excellent Medical Coverage

Mental Health Employee Assistance Plan

Paid Parental Leave

Pet Insurance

Flexible Work Hours

Onsite Gym (DC)

Gym Discount (San Diego)

Free Parking

Competitive Compensation

Stock Benefits

401K Services and Match

Job summary

Shield AI is seeking an experienced ML engineer to advance perception for autonomous systems. You will design, train, and deploy state-of-the-art vision, vision-language, and vision-language-action models, and build scalable data pipelines with robust evaluation and deployment workflows.

You will collaborate with ML researchers, autonomy engineers, and platform teams to deliver mission-ready perception capabilities for U.S.

Qualifications

• Strong knowledge of 3D vision problems and algorithms.
• Experience deploying CV models in production.
• Experience with defense customers or defense-grade systems.
• Proficiency in C++ and Python for systems engineering.

Responsibilities

• Design, train, fine-tune, and maintain vision, vision-language, and VLA models.
• Build scalable data pipelines and supervised fine-tuning workflows.
• Develop evaluation frameworks and production deployment infrastructure.
• Collaborate with researchers, autonomy engineers, and platform teams.
• Deploy and optimize ML models for embedded hardware.
• Translate research into production-ready capabilities.

Skills

PyTorch

TensorFlow

ONNX

TensorRT

C++

Python

Vision-language models

Vision-language-action models

Tools

PyTorch

TensorFlow

ONNX

TensorRT

Job description

Shield AI is seeking an experienced ML engineer to advance perception for autonomous systems. You will design, train, and deploy state-of-the-art vision, vision-language, and vision-language-action models, and build scalable data pipelines with robust evaluation and deployment workflows.

You will collaborate with ML researchers, autonomy engineers, and platform teams to deliver mission-ready perception capabilities for U.S.