Staff AI/ML Engineer — Prototyping, NLP & AWS

CACI; United States; On-site; USD 108,000 - 228,000; Full time

Benefits offered by this job

Healthcare

Retirement plan

Learning resources

Flexible time off

Job summary

CACI is seeking a Staff AI/ML Engineer to tackle challenging problems across object detection, NLP, time series, RL, and LLMs within a research-focused team.

You will prototype algorithms, analyze multi-domain data, review cutting-edge work, and contribute to training and deployment pipelines on AWS. Mentoring teammates and creating clear documentation are part of the role.

Active TS/SCI with adjudicated Polygraph is required. Experience leading interdisciplinary teams is preferred.

Qualifications

• B.S. in machine learning, computer science, mathematics, or related field.
• 8+ years of AI/ML development experience.
• Proficiency with Python and ML libraries such as PyTorch, Lightning, OpenCV, NumPy, Polars.
• Experience with supervised and unsupervised learning; deep learning architectures.
• Experience contributing on a team using version control (e.g. git, GitLab, Bitbucket).

Responsibilities

• Implement machine learning algorithms as part of a small interdisciplinary team
• Analyze large multi-domain datasets including images, text, time series, and graph data
• Review and apply cutting-edge research publications
• Create technical documentation and user guides
• Understand training and deployment pipelines including AWS services
• Provide training and mentoring to team members

Skills

Python

PyTorch

OpenCV

NumPy

Polars

Lightning

Supervised learning

Unsupervised learning

Deep learning

Git

GitLab

Bitbucket

Docker

Kubernetes

Podman

Rancher

LangChain

MLFlow

LangFuse

HuggingFace

CUDA

Education

B.S. in machine learning, computer science, mathematics, or related field

M.S. or PhD in relevant technical field

Tools

Git

GitLab

Bitbucket

Docker

Kubernetes

Podman

Rancher

LangChain

Arize Phoenix

LangFuse

MLFlow

HuggingFace

CUDA

CuPy

Numba

CuDF

Job description

CACI is seeking a Staff AI/ML Engineer to tackle challenging problems across object detection, NLP, time series, RL, and LLMs within a research-focused team.

You will prototype algorithms, analyze multi-domain data, review cutting-edge work, and contribute to training and deployment pipelines on AWS. Mentoring teammates and creating clear documentation are part of the role.

Active TS/SCI with adjudicated Polygraph is required. Experience leading interdisciplinary teams is preferred.