AI Engineer jobs in United States

Explore 895 matching jobs.

Drata

Senior Applied AI Engineer

Drata

Posted: 10/20/2026
Python
Netflix

Software Engineer 5 – Model Development and Management, AI Platform

Netflix

Posted: 10/20/2026
AirflowAWSJavaPythonScalaC++
Collectors

Staff Applied AI Engineer

Collectors

Posted: 10/20/2026
HP France

Transformation Lead, AI & Automation

HP France

Posted: 10/20/2026
ETLPythonSQLTableau
Unify

Applied AI Engineer

Unify

Posted: 10/20/2026
PythonTypeScript
Monad Foundation

Business Development Lead, AI

Monad Foundation

Posted: 10/20/2026
Booz Allen Hamilton

Agentic AI Engineer

Booz Allen Hamilton

Posted: 10/20/2026
AWSAzureCloudDockerGrafanaKubernetesNeo4jReact
Hewlett Packard Enterprise

Principal AI Engineer

Hewlett Packard Enterprise

Posted: 10/20/2026
CloudJavaPythonPyTorchTensorflowC++
HP

Transformation Lead, AI & Automation

HP

Posted: 10/20/2026
ETLPythonSQLTableau
G2i Inc.

Staff Software Engineer, AI Evals

G2i Inc.

Posted: 10/20/2026
JavaScriptPythonRustTypeScriptGo
Suno

Senior/Staff Full-Stack AI Engineer

Suno

Posted: 10/20/2026
PythonReactTypeScript
Mysten Labs

Product Lead, AI

Mysten Labs

Posted: 10/20/2026
Simple Software Solutions Group, Inc

Senior AI Developer – Agent Builder, AI Focus

Simple Software Solutions Group, Inc

Posted: 10/20/2026
Cornelis Networks

Principal Embedded Firmware Engineer – AI/HPC Networking Silicon

Cornelis Networks

Posted: 10/20/2026
FlashPythonC++
Johnson Controls

Senior AI Engineer

Johnson Controls

Posted: 10/20/2026
AssemblyCloudCyber SecurityERPPythonScikit-LearnSQL
Hewlett Packard Enterprise

Principal AI Engineer

Hewlett Packard Enterprise

Posted: 10/20/2026
CloudJavaPythonPyTorchScikit-LearnTensorflowC++
Hewlett Packard Enterprise

Principal AI Engineer

Hewlett Packard Enterprise

Posted: 10/20/2026
CloudJavaPythonPyTorchScikit-LearnTensorflowC++
Capital One

AI Engineer 5, MLX, Agentic AI, Gen AI platform Services

Capital One

Posted: 10/20/2026
AWSCloudJavaOpen SourcePythonPyTorchScalaC++Go
Capital One

AI Engineer 4 – MLX, Agentic AI, Gen AI platform Services

Capital One

Posted: 10/20/2026
AWSCloudDistributed SystemsJavaOpen SourcePythonPyTorchScalaC+++1 more
DirectSupply

AI Engineer

DirectSupply

Posted: 10/20/2026
AWSCloudDockerPostgresPythonTerraformTypeScript
Drata

Senior Applied AI Engineer

Drata
Posted 10/20/2026full-timeSan Francisco • California • United StatesSenior💰 $157,000 - $252,000 per yearWebsite
Apply

Core Competencies

Role fit
Core Competencies

Use this summary to align your resume positioning with the role.

Demonstrates expertise in applied research and machine learning, particularly in information retrieval and natural language processing. Proficient in designing evaluation methodologies and building production AI/ML systems with a strong focus on compliance and relevance.

Highest-signal resume keywords
Applied Research ExperienceMachine Learning ProficiencyInformation Retrieval ExpertisePython Programming SkillsEvaluation Methodology Design

Tech Stack

Tools & technologies
Python

About the role

Key responsibilities & impact
  • Drive the quality and effectiveness of Drata's AI systems through rigorous experimentation, evaluation, and applied research
  • Own the science behind how Drata's AI products retrieve, reason, and respond
  • Design and evaluate information access and reasoning strategies across RAG, agents, and classic ML
  • Prototype GenAI workflows, including agentic systems, that map and reason over compliance objects
  • Explore ML and probabilistic approaches including classifiers, ranking models, graph/link prediction, calibration, and structured prediction
  • Build and maintain evaluation frameworks with golden datasets, automated quality metrics, and regression detection
  • Implement and tune ranking and reranking systems
  • Run experiments to validate hypotheses and quantify improvements before production rollout
  • Debug failure modes and build error taxonomies across retrieval, reasoning, and generation
  • Collaborate with AI and Software Engineers to hand off validated approaches for productionization
  • Stay current on applied research in RAG, agents, LLM evaluation, and relevance modeling and bring innovations into the product

Requirements

What you’ll need
  • 6+ years of experience in applied research, data science, or ML focused on NLP, information retrieval, or knowledge systems
  • 2+ years of hands-on experience building or contributing to production AI/ML systems
  • Strong foundation in information retrieval, including dense and sparse retrieval, embedding models, and search relevance
  • Experience with RAG systems, chunking strategies, vector databases, and retrieval optimization
  • Proficiency in evaluation methodology, including metrics design, golden dataset creation, A/B testing, and statistical significance
  • Strong Python skills and comfort with notebook-driven research workflows
  • Experience communicating research findings to engineering teams and translating insights into actionable improvements
  • Experience with compliance, legal, or document-heavy domains is a bonus
  • Publications or contributions in IR, NLP, or RAG evaluation are a bonus
  • Willingness and ability to commute to the 2nd Street office each week on Tuesday, Wednesday, and Thursday
  • Authorization to work in the country where the job is located
  • Must not require employer sponsorship to obtain or maintain work authorization

Benefits

Comp & perks
  • Stock equity
  • Up to 100% employer-paid medical, dental, and vision premiums for employees and dependents
  • Comprehensive wellness benefits
  • Healthcare concierge services
  • 401(k) plan
  • Company-paid life and disability insurance
  • Tax-advantaged spending accounts
  • Discounted voluntary offerings
  • Paid parental leave after six months of employment
  • Kindbody fertility and family-building benefits
  • Dedicated leave specialists
  • Annual professional and personal development stipends
  • Internal learning opportunities
  • Flexible vacation policy
  • Paid holidays

ATS Keywords

✓ Tailor your resume
Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills
Machine LearningNatural Language ProcessingInformation RetrievalEvaluation MethodologyStatistical SignificanceRanking ModelsGraph PredictionEmbedding ModelsA/B TestingRegression Detection
Soft Skills
Communication SkillsCollaboration
Tools & Technologies
Vector DatabasesRAG SystemsNotebook-Driven Research Workflows
Industry Keywords
ComplianceDocument-Heavy DomainsResearch PublicationsNLP Evaluation