AI/ML Engineer (Computer Vision)
CACI International2 days ago
Base Salary
$105k - $231k/yr
Responsibilities
- Design, develop, and implement custom software solutions for AI/ML pilot projects and use cases.
- Collaborate on prototyping teams to deliver working software applications on short timelines.
- Analyze complex project problems and develop innovative technology, methodology, tool, and solution approaches.
- Participate in Agile delivery activities including planning, artifact creation, testing, demonstrations, and retrospectives.
- Design horizontally scalable and highly available systems.
- Work with data science teams to integrate applications with AI/ML, big data, and business intelligence solutions.
- Design, modify, and train custom YOLO architectures optimized for CBP full-motion video feeds.
- Architect pipelines to ingest, clean, and preprocess recorded production video in secure training and sandbox environments.
- Define annotation and labeling guidelines, curate datasets, and address class imbalance, weather variation, and low-contrast conditions.
- Optimize model inference speed and accuracy using GPU-acceleration toolkits.
- Integrate live computer vision inference pipelines with broader applications in collaboration with video-streaming engineers, software engineers, and system architects.
- Work independently on complex tasks with limited direction and management oversight.
Requirements
- Bachelor’s, master’s, or Ph.D. in computer science, data science, electrical engineering, or a related field with an AI/ML focus.
- At least 7 years of related technical experience, including at least 4 years developing and deploying deep learning or computer vision models with a strong object-detection focus.
- U.S. citizenship and ability to pass a CBP background investigation, including specified checks for felony convictions, illegal drug use, and misconduct.
- Experience with PyTorch, TensorFlow, and OpenCV.
- Advanced experience implementing and fine-tuning the YOLO family of models, including custom architectures and backbones.
- Experience training deep learning models from scratch, including custom anchor-box design, loss-function modification, and hyperparameter tuning.
- Experience with NVIDIA CUDA, cuDNN, TensorRT, or ONNX for GPU-accelerated computing.
- Experience processing video streams and files frame by frame, including RTSP and HLS sources.
- Expert-level Python programming and ability to write clean, modular, optimized code.
- Experience with data curation, active learning, and annotation tools such as CVAT, LabelStudio, or Roboflow.
- Preferred experience with multi-object tracking algorithms such as DeepSORT or ByteTrack.
- Preferred familiarity with AWS SageMaker and EC2 GPU instances.
- Preferred experience deploying models to NVIDIA Jetson or other edge and tactical hardware.
- Federal agency or Customs and Border Protection dataset experience is preferred.
Benefits
- Full-time regular employment with up to 10% local travel.
- Flexible time off and autonomy to manage time away from work.
- Healthcare, wellness, financial, retirement, family support, continuing education, and other time-off benefits.
- Learning resources and professional development opportunities.
- Position may be worked in more than one location.
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.