Skip to content
Shield AIShield AIWashington, DC

Staff Software Engineer, Perception

Lead development of state-of-the-art vision, VLM, and VLA models for autonomous perception systems. Own challenging ML problems, build data pipelines and deployment infrastructure for production autonomy, and mentor engineers while bridging research to real-world systems. Requires 7+ years experience, strong CV/ML expertise, and C++/Python proficiency.

195k – 293k/yr
On-site7+ YOEML Engineering

About the role

What You'll Do

  • 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.
  • Apply modern machine learning techniques to solve challenging perception and autonomy problems across aerial and other autonomous systems operating in complex, real-world environments.
  • Translate cutting-edge machine learning research into production-ready capabilities by balancing model performance, robustness, computational efficiency, and operational reliability.
  • Partner closely with perception, autonomy, platform, and software engineering teams to integrate machine learning capabilities into mission-ready autonomous systems.
  • Develop benchmarks, testing methodologies, and evaluation frameworks to measure model performance, identify failure modes, and guide future improvements.
  • Improve training infrastructure, developer tooling, deployment workflows, and model lifecycle management to accelerate experimentation and production delivery.

Required Qualifications

  • Typically requires a minimum of 7 years of related experience with a Bachelor’s degree; or 6 years and a Master’s degree; or 4 years with a PhD; or equivalent work experience.
  • Expertise in 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.

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).

Skills

PyTorchTensorFlowTensorRTonnxC++PythonComputer VisionMachine Learning3d visionvision-language models

Similar roles

ML Engineering jobs
Shield AI

Staff Software Engineer, Autonomous Pilot Integration

Shield AIWashington, DC +3

Lead technical delivery of first-of-a-kind autonomy solutions for new platforms in space, maritime, and contested logistics domains. Own end-to-end integration from simulation to live tests, influence architecture across teams, and provide on-site mission support in ambiguous, evolving environments.

195k – 293k/yrOn-site7+ YOEML Engineering
Square

Staff Machine Learning Engineer, Credit Products (Square Financial Services)

SquareSeattle, WA +1

Develops and deploys ML models for credit underwriting in regulated banking, owning full stack from data curation to production decisioning. Leads MLOps initiatives and improves policies using advanced techniques for underserved segments. Requires 8+ YOE in ML production and technical degree.

195k – 343k/yrHybrid8+ YOEML Engineering
Fivetran

Staff R&D Software Engineer, Fivetran AI

FivetranNew York, NY +2

Staff R&D Software Engineer building the governed context layer (Agents Schema and Context Builder) for trustworthy AI agents. Research AI techniques, prototype, ship full-stack production features, own reliability/SRE/QA, define technical direction across teams, and mentor others in a startup-like environment.

196k – 245k/yrHybrid8+ YOEML Engineering
Fivetran

Staff R&D Software Engineer, Fivetran AI

FivetranDenver, CO +3

Staff R&D Software Engineer building the governed context layer (Agents Schema and Context Builder) for trustworthy AI agents. Research AI techniques, prototype and ship production backend/frontend features, own reliability/SRE/QA, define technical direction across teams, and mentor others in a fast-moving startup-like environment.

196k – 245k/yrHybrid8+ YOEML Engineering
Okta

Staff Software Engineer, AI-Core

OktaSan Francisco, CA

Build and own core layers of a scalable Agentic AI platform for autonomous engineering agents at Okta, including agent identity/auth, knowledge/memory, observability, governance/safety, and orchestration. Requires 5+ years backend engineering experience plus AI/agent exposure; strong distributed systems and security knowledge.

194k – 267k/yrHybrid7+ YOEML Engineering