BestApply

Lead AI/ML Engineer

Attain Talent

  • United States, United States
  • Remote
  • $180,000 - $230,000 a year
  • Posted Aug 17, 2026
Sign up — let your agent apply Sign in

BestApply tailors your resume and applies for you.

PyTorchHugging FaceAWSMatplotlibWeights & BiasesTensorFlowPythonAmazon BedrockPandasSageMaker

Job description

About the role

The Lead AI/ML Engineer will spearhead the design, prototyping, and production deployment of AI/ML solutions on AWS for federal clients, drive technology adoption, mentor engineers, and communicate findings to stakeholders.

About the company

Attain Talent (part of Attain Partners) is a recruiting and talent solutions firm that places senior technical leaders on high‑impact federal initiatives, emphasizing fair compensation and an inclusive workplace.

Requirements

  • Bachelor’s degree in Computer Science, Data Science, Engineering, Mathematics, Information Systems, or related field (or equivalent experience).
  • 7–10 years of professional experience in software engineering or cloud engineering.
  • Minimum 3 years experience deploying production enterprise applications in AWS.
  • Minimum 2 years experience deploying production AI/ML applications in enterprise environments.
  • Strong proficiency in Python.
  • Experience with machine learning frameworks such as TensorFlow or PyTorch.
  • Experience with AI/ML platforms/services like Amazon Bedrock, SageMaker, Hugging Face, Weights & Biases, or similar.
  • Strong data analysis and manipulation skills using pandas.
  • Experience visualizing and presenting experiment results with tools such as matplotlib.
  • Ability to quickly learn new technologies and contribute across multiple projects.
  • Strong analytical, communication, and problem‑solving skills.
  • Comfortable presenting technical solutions to clients and executive stakeholders.
  • U.S. citizenship and ability to obtain a Public Trust clearance.
  • Experience leading or managing engineering teams (nice to have).
  • Experience mentoring software or machine learning engineers (nice to have).
  • Experience with Large Language Models, Generative AI, Retrieval‑Augmented Generation, or AI agents (nice to have).
  • Experience deploying AI solutions using MLOps best practices (nice to have).
  • Experience with containerization technologies such as Docker and Kubernetes (nice to have).
  • Experience building CI/CD pipelines for AI/ML workloads (nice to have).
  • Familiarity with DevSecOps and secure software development practices (nice to have).
  • Experience supporting federal, highly regulated, or mission‑critical environments (nice to have).