Lead AI/ML Engineer

attaintalent

United States

Hybrid

USD 180,000 - 230,000

Full time

7 days ago

Be an early applicant

Get more replies from employers

Send a job-specific resume in minutes.

Benefits offered by this job

Remote Work

Hybrid roles specified in job post

Medical, Dental, and Vision

Life Insurance, Short/Long Term Disab.

401(k) with matching

Generous PTO

Education budget

Wellness budget

Bonus incentive programs

Job summary

Attain Talent is seeking a Lead AI/ML Engineer to drive design, evaluation, and deployment of AI/ML solutions for high-impact federal initiatives. You will mentor engineering teams, collaborate with federal clients, and shape technical strategy across cloud-based platforms.

You will lead prototypes, evaluate emerging technologies, and deliver production-ready AI/ML applications in AWS. Strong communication and leadership are essential for cross-functional success.

Qualifications

7–10 years of professional software/cloud engineering experience.
At least 3 years deploying production AWS solutions.
At least 2 years deploying production AI/ML apps in enterprises.
Strong proficiency in Python and ML frameworks (TF or PyTorch).
Experience with AI platforms such as SageMaker, Bedrock, HF, W&B.

Responsibilities

Lead evaluations of AI/ML technologies via prototypes and POCs.
Communicate benefits, tradeoffs, and risks to tech and business stakeholders.
Design, develop, and deploy production-grade AI/ML apps in AWS.
Support AI/ML development across dev, ops, and security in cloud-native setups.
Drive enterprise adoption of modern AI technologies and best practices.
Build, train, evaluate, and optimize ML models for production use cases.
Collaborate with clients and cross-functional teams to deliver mission-critical solutions.
Provide technical leadership by mentoring engineers and growing junior staff.
Present findings and recommendations to leadership and federal customers.

Skills

Python

Leadership

Communication

Problem-solving

Mentoring

Education

Bachelor's degree in Computer Science, Data Science, Engineering, Mathematics, Information Systems, or related field

Tools

AWS

TensorFlow

PyTorch

SageMaker

Bedrock

Hugging Face

Weights & Biases

Docker

Kubernetes

Job description

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.

You Will Get To

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.

Who You Are

A technical leader who enjoys solving complex engineering and machine learning challenges.

Passionate about applying AI to solve meaningful, real-world problems within the federal government.

Comfortable leading technical discussions with engineers, architects, and executive stakeholders.

Curious about emerging AI technologies and excited to evaluate new tools and platforms.

A collaborative mentor who enjoys helping teams grow while driving technical excellence.

An excellent communicator capable of translating complex technical concepts into actionable recommendations.

Qualifications

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.

Nice to Have

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.

Client Requirements

Applicants must be U.S. Citizens.
Ability to obtain a Public Trust clearance.

Salary

We are committed to offering a competitive salary for this position, with an estimated range of $180,000 to $230,000 annually. Please note that this range is intended to provide a general idea of what to expect. The final offer may vary based on experience, skills, and other factors.

Attain Partners is committed to fair and equitable compensation practices. The individual base salary for this position is unique to each candidate and will be commensurate with experience, education, and skills. In addition to base salary, this role is eligible for an annual discretionary bonus.

Full Time Employee Benefits