Machine Learning Engineer

$101,566 - $131,584/Yr

General Dynamics - Springfield, VA

Full-time - Mid Level

Springfield, VA

10,001+ employees

Transportation Equipment Manufacturing

About the position

As a Machine Learning Engineer at GDIT, you will play a crucial role in enhancing national safety and security by developing innovative machine learning solutions tailored to the unique needs of intelligence clients. Your work will involve rapid prototyping of multimodal deep learning solutions and data pipelines, focusing on improving analytic workflows and addressing key intelligence questions using advanced computer vision techniques.

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) for image retrieval and segmentation tasks.
• Conduct AI-assisted labeling, object detection, and visual question answering using geospatial datasets.
• Optimize convolutional neural networks (CNN) for object detection or segmentation tasks using satellite imagery.

Requirements

• Bachelor or Master's Degree in Computer Science, Artificial Intelligence, Machine Learning, Data Science, or equivalent experience.
• 5+ years of experience in machine learning and application development.
• Experience applying transfer learning and knowledge distillation methodologies.
• Experience building secure containerized Python applications with CI/CD pipelines.
• Experience using Python to query and retrieve imagery from S3 compliant APIs.
• Experience with deep learning frameworks such as PyTorch or TensorFlow.
• Experience with version control systems such as 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.
• Experience with Vision Transformers (ViT) such as DINO or DeiT.
• Experience communicating methodological choices and model results.
• Experience with verification and validation test benches.
• Experience with Explainable AI (XAI) techniques.
• Experience with Open Neural Net Exchange (ONNX).

Benefits

• 401K with company match
• Comprehensive health and wellness packages
• Professional growth opportunities including paid education and certifications
• Paid vacation and holidays