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.
- Design, modify, and train custom YOLO architectures for CBP full-motion video feeds.
- Architect pipelines to ingest, clean, and preprocess production-recorded video data in secure training and sandbox environments.
- Define annotation and labeling guidelines, curate datasets, and address class imbalance, weather variation, and low-contrast environments.
- Optimize model inference speed and accuracy using GPU hardware-acceleration toolkits.
- Integrate live computer vision inference pipelines with broader applications in collaboration with video-streaming engineers, software engineers, and system architects.
- Participate in delivery planning, testing, demonstrations, retrospectives, and related development activities.
- Work independently on complex technical tasks with limited direction.
Requirements
- Bachelor’s, master’s, or Ph.D. in computer science, data science, electrical engineering, or a related AI/ML-focused field.
- 7+ years of related technical experience, including 4+ years developing and deploying deep learning or computer vision models with an emphasis on object detection.
- Experience with PyTorch or TensorFlow and OpenCV.
- Advanced experience with YOLO models, including custom architectures, anchor-box design, loss-function modification, and hyperparameter tuning.
- Experience with NVIDIA CUDA, cuDNN, TensorRT, or ONNX for GPU-accelerated computing.
- Experience processing RTSP, HLS, and file-based video streams frame by frame.
- Expert-level Python programming and the ability to write clean, modular, optimized code.
- Experience with data curation, active learning, and annotation tools such as CVAT, Label Studio, or Roboflow.
- U.S. citizenship and ability to pass a CBP background investigation.
- 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 desired.
Benefits
- Flexible time off, learning resources, healthcare, wellness, financial and retirement benefits, family support, continuing education, and other time-off benefits.
- Full-time regular employment with up to 10% local travel.
- The 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.