Lead AI/ML Computer Vision Engineer - YOLO & Edge

CACI International; United States; Hybrid; USD 104,000 - 218,000; Full time

Job summary

CACI International seeks a Lead AI/ML Engineer (Computer Vision) to lead a small team on CBP-related work in Northern Virginia. You will design, train, and optimize vision models, with a focus on object detection using YOLO architectures and GPU-accelerated inference.

The role requires strong Python, PyTorch, OpenCV, and cloud/edge deployment experience. A hybrid schedule in Ashburn, VA is available, and candidates must meet stringent security requirements.

Qualifications

• Bachelor’s/Master’s/Ph.D. in CS/Data Science/EE with AI/ML focus.
• 7+ years of technical experience with 4+ years in DL/CV and object detection.
• U.S. Citizenship and ability to pass CBP background checks.
• Hybrid schedule in Ashburn, VA available.

Responsibilities

• Lead a 3-person AI/ML team, set sprint goals, standards, and metrics.
• Design and train custom YOLO architectures for CBP environments.
• Architect end-to-end CV pipeline from data ingest to secure sandbox.
• Define labeling guidelines, data curation, and handle class imbalance.
• Optimize FPS and mAP using GPU toolkits and hardware acceleration.
• Collaborate with engineers to integrate live CV inference into the app.
• Serve as the primary POC for the AI/ML pod in Agile/Scrum.

Skills

Python programming

Agile/Scrum

Leadership

Communication

Model evaluation

Education

Bachelor’s/Master’s/Ph.D. in CS/Data Science/EE or related AI/ML focus

Tools

PyTorch

OpenCV

YOLO

TensorRT

CUDA

TensorRT

ONNX

AWS SageMaker

NVIDIA Jetson

DeepSORT/ByteTrack

Job description

CACI International seeks a Lead AI/ML Engineer (Computer Vision) to lead a small team on CBP-related work in Northern Virginia. You will design, train, and optimize vision models, with a focus on object detection using YOLO architectures and GPU-accelerated inference.

The role requires strong Python, PyTorch, OpenCV, and cloud/edge deployment experience. A hybrid schedule in Ashburn, VA is available, and candidates must meet stringent security requirements.