Lead AI/ML Engineer

Attain Talent

3d•$180,000 - $230,000•Remote

About The Position

Attain Talent is seeking a Lead AI/ML Engineer to lead the design, evaluation, and implementation of cutting-edge artificial intelligence and machine learning solutions supporting high-impact federal initiatives. This role is ideal for a senior technical leader who thrives at the intersection of AI, cloud engineering, and innovation, helping drive enterprise adoption of emerging technologies while mentoring engineering teams and partnering directly with federal clients.

Requirements

Bachelor's degree in Computer Science, Data Science, Engineering, Mathematics, Information Systems, or a related technical field (or equivalent experience).
7–10 years of professional experience in software engineering or cloud engineering experience.
At least 3 years of experience deploying production enterprise applications in AWS.
At least 2 years of 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 and services including Amazon Bedrock, SageMaker, Hugging Face, Weights & Biases, or similar technologies.
Strong data analysis and manipulation experience using pandas.
Experience visualizing and presenting experiment results using tools such as matplotlib.
Experience working within AWS cloud environments.
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.
Applicants must be U.S. Citizens.
Ability to obtain a Public Trust clearance.

Nice To Haves

Experience leading or managing engineering teams.
Experience mentoring software engineers or machine learning engineers.
Experience with Large Language Models (LLMs), Generative AI, Retrieval-Augmented Generation (RAG), or AI agents.
Experience deploying AI solutions using MLOps best practices.
Experience with containerization technologies such as Docker and Kubernetes.
Experience building CI/CD pipelines for AI/ML workloads.
Familiarity with DevSecOps and secure software development practices.
Experience supporting federal, highly regulated, or mission-critical environments.

Responsibilities

Lead technical evaluations of emerging AI/ML technologies through prototypes, proof of concepts, and experimentation.
Analyze and communicate the benefits, tradeoffs, and implementation risks of new technologies to technical and business stakeholders.
Design, develop, and deploy production-ready AI/ML applications within AWS cloud environments.
Support AI/ML application development across development, operations, and security disciplines in cloud-native environments.
Drive enterprise adoption of modern AI technologies and engineering best practices.
Build, train, evaluate, and optimize machine learning models for production use cases.
Collaborate directly with clients and cross-functional engineering teams to deliver mission-critical solutions.
Provide technical leadership by mentoring engineers and supporting the professional growth of junior team members.
Present technical findings, experiment results, and architectural recommendations to internal leadership and federal customers.