Machine Learning Engineer - Computer Vision

Overview

Arlington, VA

We are seeking a highly skilled and motivated Machine Learning Engineer specializing in Computer Vision to join our team. The ideal candidate will have a strong background in developing and deploying machine learning models focused on image and video processing. You will work closely with cross-functional teams to design, implement, and optimize vision-based AI solutions to address real-world challenges.

• Design, develop, and deploy computer vision models for tasks such as object detection, object tracking, video segmentation, and facial recognition.
• Optimize and fine-tune deep learning algorithms for real-time performance.
• Work closely with software engineers and product teams to identify opportunities for leveraging data.
• Collect, clean, and preprocess large datasets to prepare for model training and evaluation.
• Evaluate and optimize machine learning models for accuracy, performance, and scalability.
• Deploy models into production environments and monitor their performance to ensure reliability.
• Stay up-to-date with the latest advancements in computer vision and artificial intelligence.
• Collaborate with cross-functional teams to integrate machine learning solutions into business processes.
• Document processes, models, and implementations to ensure reproducibility and scalability.

• Bachelor's or Master’s degree in Computer Science, Artificial Intelligence, Data Science, or a related field.
• Experience in deep learning models, their training, and hyperparameter tuning using libraries such as Tensor Flow, PyTorch, and Transformers or other Huggingface tools.
• Experience with data manipulation tools such as Pandas, Num Py, and SQL.
• Strong programming skills in Python and C++.
• Experience in MLOps principles and model deployment and instrumentation on cloud platforms such as AWS, Azure, or Google Cloud for model deployment and knowledge with efficient serving tools such as ONNX, triton, and vllm.
• Proficiency in working with image and video data, including preprocessing and augmentation techniques.
• Strong understanding of machine learning algorithms, including supervised and unsupervised learning and deep learning.
• Strong communication skills and the ability to work collaboratively in a team environment.

• Familiarity with containerization and orchestration tools like Docker and Kubernetes.
• Experience with version control systems such as Git.
• Understanding software engineering best practices, including code review, testing, and documentation.
• Experience with Large Language Models (LLMs) is a great plus.
• Experience with data annotation tools and processes.
• Comprehensive health and wellness benefits.
• Professional development opportunities and continuous learning programs.
• Collaborative and inclusive work environment.