AI/ML Engineer

CDIT isseeking a Mid-Level AI/ML Computer Vision Engineer to support the BEAGLE AgileSolution Factory (ASF) team serving U.S. Customs and Border Protection (CBP) inNorthern Virginia.

You will bepart of a focused three-person computer vision pod building custom objectdetection and classification models with the YOLO framework. The team processeslive and recorded Full-Motion Video (FMV) feeds to automatically identifyroads, vehicles, people, and terrain features. This is a hands-on role: youwill bring raw production video into a secure training environment, curatedatasets, train and tune models, and deploy high-performance inferencepipelines.

Model Training & Experimentation: Train, fine-tune, and evaluate custom YOLO object detection models onproprietary datasets; experiment with hyperparameters, augmentations, and modelsizes to balance speed and accuracy.

Data Engineering & Ingestion: Process, clean, and format raw video recorded in productionenvironments into high-quality training datasets.

Dataset Annotation & Curation: Manage the annotation lifecycle, including labeling images and videoframes, verifying annotation quality, and generating synthetic data whenneeded.

Inference Pipeline Development: Write Python scripts that parse video frames, feed them into inferenceengines, and export structured prediction data (JSON/Protobuf) for detectedobjects.

Model Evaluation: Produce performance reports, confusion matrices, and precision-recallcurves to find model blind spots and drive continuous improvement.

Agile Delivery: Partner with the Senior AI/ML Lead, video engineers, and developers inan Agile/Scrum environment to deliver production-ready code in short sprints.

B.S. in Computer Science, Data Science,Software Engineering, or a related discipline.

5+ years of software engineering or datascience experience, including 2+ years hands-on building and training computervision models.

Python programming with core ML libraries(NumPy, Pandas, Scikit-Learn).

Hands-on experience training neural networkswith PyTorch or TensorFlow.

Experience implementing YOLO (e.g., YOLOv8/v9)for object detection, segmentation, or classification.

Image and video processing experience withOpenCV or FFmpeg.

Data curation and formatting, with experienceusing annotation platforms such as CVAT or LabelImg.

Git version control and collaboration onshared codebases.

Strong analytical skills with a focus onbenchmarking model performance (mAP, FPS, inference latency).

Willing to work on site 2–3 days per week inAshburn, VA.

U.S. Citizenship and ability to pass a CBPbackground investigation, which includes but is not limited to:

–3-year check for felony convictions

–1-year check for illegal drug use

–1-year check for misconduct such as theft orfraud

Containerized development environments(Docker).

GPU execution environments and model exportformats (ONNX, TensorRT).

Multi-object tracking libraries.

Agile tools such as Jira and Confluence.

CDIT placesskilled professionals on mission-focused government programs and supports themthroughout their engagement. This role offers direct, hands-on work on appliedcomputer vision for a federal border security mission, within a small,high-impact team.

CDIT is anEqual Opportunity Employer. All qualified applicants will receive considerationfor employment without regard to race, color, religion, sex, sexualorientation, gender identity, national origin, disability, or protected veteranstatus.