# Senior AI Engineer

**Company:** [Drata](https://hotfix.jobs/companies/drata)
**Location:** Remote
**Role:** ML Engineering
**Experience:** 7+ years
**Skills:** Python, TypeScript, LLMs, RAG, Agent Frameworks, Prompt Engineering, Vector Databases, Pinecone, Chroma, Faiss, Embeddings, Snowflake, Knowledge Graphs
**Posted:** 2025-12-31

> Designs, builds, and scales agentic AI systems using LLMs for compliance automation, including multi-step reasoning, RAG, and production deployment. Requires 7+ years software engineering with 2+ years ML/AI, Python proficiency, and cross-functional collaboration.

## Job Description

## Responsibilities

### Build Agentic & Intelligent AI Systems
- Design and implement LLM-powered systems capable of multi-step reasoning, evidence grounding, and decision support in high-trust domains.
- Develop agentic workflows that combine retrieval, tool use, structured reasoning, and human oversight.
- Create interactive AI experiences that allow users to engage naturally with complex compliance and risk data.

### Automated Reasoning Over Regulations & Evidence
- Build AI systems that reason over structured and unstructured data to support regulatory interpretation, control alignment, and evidence validation.
- Ensure AI outputs are traceable, explainable, and auditable, meeting the expectations of enterprise customers and auditors.

### Production-Grade AI Architecture
- Architect and deploy scalable LLM + retrieval + agent systems in production environments.
- Optimize for latency, cost, reliability, and evaluation in real-world enterprise workloads.
- Partner with platform, security, product teams, and other application development teams with diverse skill sets to operationalize AI safely and effectively.

### Responsible & Trustworthy AI
- Embed human-in-the-loop workflows, confidence thresholds, and safety guardrails into AI systems.
- Ensure privacy-preserving data handling, robust failure modes, and transparent behavior aligned with Drata’s trust principles.

## Requirements

- 7+ years of hands-on software engineering experience; 2+ years specifically in ML/AI engineering.
- Proficiency in **Python**; **TypeScript** experience is a plus, especially for production AI system integrations.
- Familiarity with vector databases (**Pinecone**, **Chroma**, **FAISS**, etc.) and RAG system design.
- Proven experience building and shipping LLM-based applications in production, including **embeddings**, **RAG**, **agent frameworks**, **prompt engineering**, and evaluation.
- Track record of taking AI systems from concept to production, designing scalable, maintainable solutions.
- Ability to decompose complex tasks into agentic workflows and reason over high-stakes, structured, and unstructured data.
- Experience working cross-functionally with product, compliance, security, engineering teams, and partnering with other application development teams with diverse skill sets.

## Nice-to-Haves
- Experience in compliance, security, risk, or audit domains.
- Familiarity with **Snowflake**-based analytics, **knowledge graphs**, or enterprise data platforms.
- Experience partnering with non-technical stakeholders such as compliance or GRC teams.

## Similar roles

- [Senior Machine Learning Engineer, Agent Oversight](https://hotfix.jobs/jobs/1f504a69-044f-47ab-872b-120a1706cd88) - Scale AI - San Francisco, CA - $216k – $270k/yr
- [Senior Software Engineer, Agent Oversight](https://hotfix.jobs/jobs/808ce719-6b16-4b6e-8887-8c8f8a780b75) - Scale AI - San Francisco, CA - $216k – $270k/yr
- [Senior Software Engineer, AI Infrastructure](https://hotfix.jobs/jobs/dd1c3420-6c0a-47ed-a1a7-de401bdf506c) - Ambient.ai - Redwood City, CA - $168k – $205k/yr
- [Senior Software Engineer](https://hotfix.jobs/jobs/423c60aa-ca4c-400d-babe-a50099433cbb) - SpotOn - Chicago, IL - $160k – $190k/yr
- [Senior AI Engineer, Enterprise](https://hotfix.jobs/jobs/87a30fd1-3fe6-47f3-a0a1-da444f3c00f4) - Snowflake - Menlo Park, CA - $156k – $224k/yr

**Apply:** https://hotfix.jobs/jobs/fb336df9-33b6-45fd-807e-342b1bfc64b4
**Canonical:** https://hotfix.jobs/jobs/fb336df9-33b6-45fd-807e-342b1bfc64b4