Machine Learning Engineer

$97,258 - $131,584/Yr

General Dynamics - Saint Louis, MO

Full-time - Mid Level

Remote - Saint Louis, MO

10,001+ employees

Transportation Equipment Manufacturing

About the position

As a Machine Learning Engineer at GDIT, you will develop innovative solutions to complex problems, focusing on enhancing the end-user experience. Your role will involve rapid prototyping of multimodal deep learning solutions and data pipelines to support GeoAI capabilities, ultimately contributing to national safety and security. You will work with cutting-edge technologies in computer vision and AI to address key intelligence questions using various geospatial datasets.

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.
• Tailor solutions to meet the unique requirements of clients in the intelligence sector.

Requirements

• Bachelor or Master's Degree in Computer Science, Artificial Intelligence, Machine Learning, Data Science, or equivalent experience.
• 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.
• Proficient in 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 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
• Short and long-term disability benefits
• Life and accidental death insurance
• Critical illness and business travel insurance