AI/ML Engineer

CACI International • Ashburn, VA

2d • $86,600 - $181,800 • Hybrid

About The Position

CACI is currently looking for a Mid-Level AI/ML Computer Vision Engineer to join our BEAGLE Agile Solution Factory (ASF) Team supporting the Customs and Border Protection (CBP) client located in Northern Virginia! As a Mid-Level AI/ML Engineer on our Computer Vision team, you will work within a specialized 3-person pod dedicated to building custom object detection and classification models using the YOLO framework. Your mission will be to process live and recorded Full-Motion Video (FMV) feeds, enabling automated parsing of entities like roads, vehicles, people, and terrain features. You will play a hands-on role in bringing raw production video into our secure training environment, curating datasets, training models, and deploying high-performance inference pipelines.

Requirements

• Must be a U.S. Citizen with the ability to pass a CBP background investigation.
• College degree (B.S.) in Computer Science, Data Science, Software Engineering, or a related discipline.
• 3–5 years of software engineering or data science experience, with 2+ years of hands-on experience building/training Computer Vision models.
• Must be available to work a hybrid schedule in Ashburn, VA.
• Programming in Python and using core machine learning libraries (NumPy, Pandas, Scikit-Learn).
• Hands-on experience with PyTorch or TensorFlow for training neural networks.
• Experience implementing YOLO (e.g., YOLOv8/v9) for object detection, segmentation, or classification.
• Familiarity with image and video processing techniques using OpenCV or FFmpeg.
• Data curation, formatting, and experience using data annotation platforms (e.g., CVAT, LabelImg).
• Using Git for version control and collaborating on shared codebases.
• Strong analytical skills, with a focus on benchmarking model performance (mAP, FPS, inference latency).

Nice To Haves

• Familiarity with containerized development environments (Docker).
• Experience with GPU execution environments and model export formats (ONNX, TensorRT).
• Experience with multi-object tracking libraries.
• Familiarity with Agile tools like Jira and Confluence.

Responsibilities

• Train, fine-tune, and evaluate custom YOLO object-detection models on proprietary datasets.
• Run experiments with different hyperparameters, augmentations, and model sizes to find the optimal balance of speed and accuracy.
• Process, clean, and format raw video files recorded from production environments into high-quality datasets for training.
• Manage the data annotation lifecycle, labeling images and video frames, verifying annotation quality, and generating synthetic data where needed.
• Write Python scripts to parse video frames, feed them into inference engines, and export structured prediction data (JSON/Protobuf) representing detected objects.
• Generate performance reports, confusion matrices, and precision-recall curves to identify model blind spots and continuously improve detection capabilities.
• Work closely with the Senior AI/ML Lead, video engineers, and developers in an Agile/Scrum framework to deliver production-ready code in short sprints.

Benefits

• flexible time off benefit
• robust learning resources
• competitive compensation
• benefits and learning and development opportunities
• comprehensive benefits such as; healthcare, wellness, financial, retirement, family support, continuing education, and time off benefits.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level