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 a collaborative environment, ensuring that your expertise helps protect the country from various threats.

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 and automating builds using CI/CD pipelines.
• Proficiency in Python for querying and retrieving imagery from S3 compliant APIs and performing image preprocessing.
• Experience with deep learning frameworks such as PyTorch or TensorFlow for optimizing CNNs for object detection or segmentation tasks.
• Familiarity 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.
• Familiarity 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
• Short and long-term disability benefits
• Life and accidental death and dismemberment insurance
• Critical illness and business travel insurance