AI/ML Engineer
CACI International1 day ago
Ashburn, VA, USAMid Level
H1B sponsor
Base Salary
$87k - $182k/yr
Responsibilities
- Train, fine-tune, and evaluate YOLO object-detection models using proprietary datasets and varied hyperparameters, augmentations, and model sizes.
- Process, clean, format, annotate, and curate production video and image datasets, including synthetic data generation and annotation-quality verification.
- Develop Python inference pipelines that process video frames and export structured object predictions in JSON or Protobuf formats.
- Generate performance reports, confusion matrices, and precision-recall curves using metrics such as mAP, FPS, and inference latency.
- Collaborate with the Senior AI/ML Lead, video engineers, and developers to deliver production-ready code in Agile/Scrum sprints.
Requirements
- Bachelor of Science degree in Computer Science, Data Science, Software Engineering, or a related discipline.
- At least 5 years of software engineering or data science experience, including 2 or more years building and training computer vision models.
- Experience programming in Python and using NumPy, Pandas, and Scikit-Learn.
- Hands-on experience with PyTorch or TensorFlow for neural-network training.
- Experience implementing YOLO for object detection, segmentation, or classification.
- Experience with image and video processing using OpenCV or FFmpeg.
- Experience with data curation, formatting, and annotation platforms such as CVAT or LabelImg.
- Experience using Git and collaborating on shared codebases.
- Strong analytical skills focused on benchmarking model performance.
- Willingness to work onsite two to three days per week and pass a CBP background investigation; U.S. citizenship is required.
- Preferred experience includes Docker, GPU execution environments, ONNX, TensorRT, multi-object tracking libraries, Jira, and Confluence.
Benefits
- Full-time regular employment with up to 10% local travel.
- Onsite work required two to three days per week in Northern Virginia.
- Flexible time off, healthcare, wellness, financial, retirement, family support, continuing education, and other time-off benefits.
- Learning and development resources and opportunities for career growth.
Categories
About CACI International
CACI International is a public government contractor that delivers enterprise IT, cybersecurity, electronic warfare, space, and intelligence solutions to U.S. defense, intelligence, and federal civilian customers. It provides systems engineering, software and data services, and mission support across C4ISR and business systems, primarily through long-term government contracts. Founded in 1962 and headquartered in Reston, Virginia, CACI trades on the NYSE under the ticker CACI.