AI/ML Engineer
CACI International1 day ago
Ashburn, VA, USAMid Level / Senior
H1B sponsor
Base Salary
$87k - $182k/yr
Responsibilities
- Train, fine-tune, evaluate, and optimize YOLO object-detection models using proprietary datasets.
- Process, clean, format, annotate, curate, and quality-check video and image datasets, including generating synthetic data.
- Develop Python inference pipelines that process video frames and export structured JSON or Protobuf predictions.
- Generate performance reports, confusion matrices, and precision-recall curves while benchmarking model accuracy, speed, and 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.
- Three to five years of software engineering or data science experience, including at least two years of hands-on computer vision model development and training.
- U.S. citizenship and ability to pass a CBP background investigation.
- Availability to work a hybrid schedule in Ashburn, Virginia.
- Proficiency with Python, NumPy, Pandas, Scikit-Learn, and Git.
- Hands-on experience with PyTorch or TensorFlow, YOLO, and OpenCV or FFmpeg.
- Experience with data curation, formatting, and annotation platforms such as CVAT or LabelImg.
- Preferred qualifications include Docker, GPU execution environments, ONNX, TensorRT, multi-object tracking libraries, Jira, and Confluence.
Benefits
- Hybrid work schedule in Ashburn, Virginia, with up to 10% local travel.
- Flexible time off, healthcare, wellness, financial and retirement benefits, 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.