Machine Learning Engineer

$101,566 - $131,584/Yr

Unclassified - Springfield, VA

Full-time - Mid Level

Remote - Springfield, VA

10,001+ employees

About the position

As a Machine Learning Engineer at GDIT, you will be at the forefront of delivering innovative solutions to complex problems. Your role will involve tailoring cutting-edge machine learning solutions to meet the unique requirements of our clients, particularly in the intelligence sector. You will be responsible for rapidly prototyping containerized multimodal deep learning solutions and developing associated data pipelines that enhance GeoAI capabilities. This work is crucial for improving analytic workflows and addressing key intelligence questions. You will leverage your expertise in implementing State-of-the-Art (SOTA) Computer Vision (CV) and Vision Language Models (VLM) to conduct tasks such as image retrieval, segmentation, AI-assisted labeling, object detection, and visual question answering using diverse geospatial datasets, including satellite and aerial imagery, full-motion video (FMV), and ground photos. In this role, you will own the opportunity to serve as a critical component of our nation's safety and security. Your contributions will directly impact the protection of our country from various threats. You will be expected to apply your technical skills in transfer learning, knowledge distillation, and deep learning frameworks to optimize models for specific tasks, ensuring that the solutions you develop are both effective and efficient. Your work will not only focus on the technical aspects but also on the end-user experience, making it a priority to deliver solutions that are user-friendly and impactful. At GDIT, we prioritize the growth of our employees, and as a Machine Learning Engineer, you will have access to professional development opportunities, including paid education and certifications. We believe that our people are our differentiator, and we are committed to fostering an environment where you can thrive and advance your career while contributing to meaningful projects.

Responsibilities

• Rapidly prototype containerized multimodal deep learning solutions and associated data pipelines.
• Implement State-of-the-Art (SOTA) Computer Vision (CV) and Vision Language Models (VLM).
• Conduct image retrieval, segmentation tasks, AI-assisted labeling, object detection, and visual question answering using geospatial datasets.
• Apply transfer learning and knowledge distillation methodologies to fine-tune pre-trained models.
• Build secure containerized Python applications, including hardening and automating builds using CI/CD pipelines.
• Query and retrieve imagery from S3 compliant APIs and perform image preprocessing using libraries like Boto3 and NumPy.
• Optimize convolutional neural networks (CNN) for object detection or segmentation tasks using deep learning frameworks like PyTorch or TensorFlow.
• Utilize version control systems such as GitLab and leverage CUDA for GPU accelerated computing.

Requirements

• Bachelor or Master's Degree in Computer Science, Artificial Intelligence, Machine Learning, Data Science, or equivalent experience in lieu of degree.
• 5+ years of related experience in machine learning and application development.
• Demonstrated experience applying transfer learning and knowledge distillation methodologies.
• Experience building secure containerized Python applications with CI/CD pipelines.
• Proficiency in using Python for querying and retrieving imagery from S3 compliant APIs.
• Experience with deep learning frameworks such as PyTorch or TensorFlow.
• Familiarity with version control systems like GitLab.
• Experience leveraging CUDA for GPU accelerated computing.

Nice-to-haves

• Experience with the HuggingFace Transformers library and hub.
• Experience with OpenShift and container orchestration within Kubernetes.
• Familiarity with Vision Transformers (ViT) such as DINO or DeiT.
• Experience communicating methodological choices and model results.
• Experience with verification and validation test benches.
• Familiarity with Explainable AI (XAI) techniques.
• Experience with Open Neural Net Exchange (ONNX).

Benefits

• 401K with company match
• Comprehensive health and wellness packages
• Internal mobility team dedicated to helping you own your career
• Professional growth opportunities including paid education and certifications
• Paid vacation and holidays
• Short and long-term disability benefits
• Life, accidental death and dismemberment, personal accident, critical illness and business travel and accident insurance.