Drata
Senior Platform AI Engineer
Hybrid - San Francisco · Senior
Sponsorship not specified$192k-$260kDetected 56 days ago
TypeScriptPythonNode.jsVector DatabasesAWSCloud PlatformsTerraformCI/CDSite Reliability EngineeringAPI DevelopmentRESTMachine LearningLLMsRAGAgentic AILLMOpsAI OrchestrationA/B TestingComplianceDesign SystemsCollaboration
About the role
- This is not a traditional infrastructure role.
- You'll debug prompt templates alongside Terraform modules.
- When a model upgrade changes behavior across 15 workflows, you'll assess quality impact - not just confirm the containers are healthy.
Responsibilities
- Design and build MCP (Model Context Protocol) servers that expose Drata's platform to AI agents.
- You'll own the protocol layer that determines whether agents can reliably find and use the right tools - writing semantic parameter descriptions, contextual hints, and tool schemas that optimize for model comprehension, not just developer ergonomics.
- Build and operate the infrastructure for deploying multi-step agent workflows - state management across complex reasoning chains, tool routing and execution runtimes, and long-running agentic processes that persist over time.
- Own the orchestration layer that coordinates agent planning, tool calls, and human-in-the-loop patterns.
- Design systems that handle agent failure modes gracefully: retries on ambiguous tool outputs, fallback strategies when models produce unexpected results, and observability into multi-step execution traces.
- Own the operational side of our LLM workflows: model upgrades across production pipelines (assessing behavior changes, not just version bumps), prompt versioning and A/B testing, AI workflow deployment with custom container compatibility, and output quality monitoring.
- Manage token capacity planning - understanding model costs, context limits, batching strategies, and rate governance across workflows.
- Operate and evolve our production AI stack: vector storage and indexing (designing chunking strategies and metadata schemas for retrieval quality), document parsing pipelines, multi-region deployment, and cost optimization across LLM providers.
- Implement caching layers and token-aware request routing to manage spend as AI workloads scale.
- Build CI/CD patterns specific to AI workflows (reproducible deployments, SDK version compatibility, workflow rollback semantics).
Skills
- writing Terraform one day, debugging a prompt template the next, and designing an agent orchestration framework the day after.
Compensation
- A variety of factors are considered when determining someone's leveling and compensation-including a candidate's professional background and experience.
- This role will receive a competitive base salary, benefits, and stock, typically in the form of Restricted Stock Units (RSUs).
- $192,000 - $259,800.
Benefits
- We provide stock equity to ensure that as the company grows, you share directly in that success.
- Equity gives every employee a sense of ownership and the opportunity to celebrate our wins together-because your contributions don't just support our progress; they help drive our collective success.
- We want to support you in life's most important moments, so we offer a paid Parental Leave policy, after six months of employment.
- Employees also receive access to Kindbody fertility and family-building benefits and dedicated leave specialists who help guide you through the entire process.
- Drata offers a flexible vacation policy, paid holidays, and other perks to recharge.
Company info
- Hear the Voice of the Team https://drata.com/about/life-at-drata: Explore our "Life at Drata" page for employee testimonials on our collaborative and the growth opportunities available.
- Experience the Impact https://www.greatplacetowork.com/certified-company/7044563: See why we are consistently recognized on Fortune's Best Workplaces lists.
- LinkedIn https://www.linkedin.com/company/drata/posts/?feedView=all - follow us for company updates, employee stories, and career news.
- A comprehensive suite of financial benefits, including a 401(k) plan, company-paid life and disability insurance, tax-advantaged spending accounts, and a range of discounted voluntary offerings to help you customize and strengthen your overall financial position.
This listing is sourced directly from Drata's careers page and normalized into a canonical job model.