3 months ago
Base Salary
$192k - $260k/yr
Responsibilities
- Design and build MCP servers and API schemas optimized for AI-agent tool discovery, comprehension, token budgets, and multi-tenant workspace isolation.
- Build and operate infrastructure for multi-step agent workflows, including state management, tool routing, execution runtimes, persistent processes, human-in-the-loop patterns, retries, fallbacks, and execution tracing.
- Manage LLM workflow operations including model upgrades, prompt versioning, A/B testing, container compatibility, output quality monitoring, token capacity planning, and rate governance.
- Design and operate production AI infrastructure including vector storage and indexing, chunking and metadata strategies, document parsing pipelines, multi-region deployment, caching, and token-aware routing.
- Make RAG architecture decisions involving embeddings, retrieval filtering, and coordinated data models.
- Build AI-specific CI/CD, deployment rollback, observability, token usage dashboards, response quality metrics, agent traces, and cost-per-workflow tracking.
- Provide documented platform primitives that enable product engineering teams to deliver AI features faster.
Requirements
- 7+ years of software engineering experience.
- 2+ years of experience building or operating AI/ML infrastructure in production.
- Strong Python experience; TypeScript and Node.js are nice-to-have.
- Experience with LLM APIs, vector databases, or AI orchestration platforms.
- Experience in several areas including AWS cloud infrastructure, ECS, S3, Bedrock, container orchestration, infrastructure-as-code, CI/CD pipeline design, API design, workflow orchestration engines, and distributed systems.
- Experience with some AI-specific tooling such as Claude, OpenAI, vLLM, SageMaker, embedding pipelines, prompt management platforms, or agent frameworks.
- Ability to understand and communicate technical tradeoffs involving AI infrastructure, reliability, and scalability.
Benefits
- Hybrid Bay Area schedule with in-office collaboration Tuesday through Thursday and flexible Mondays and Fridays.
- Stock equity, typically in the form of Restricted Stock Units (RSUs).
- Up to 100% employer-paid medical, dental, and vision premiums for employees and dependents.
- Wellness benefits and healthcare concierge services.
- 401(k), company-paid life and disability insurance, tax-advantaged spending accounts, and discounted voluntary benefits.
- Paid parental leave after six months, Kindbody fertility and family-building benefits, and dedicated leave specialists.
- Annual professional and personal development stipends and internal learning opportunities.
- Flexible vacation policy, paid holidays, and other time-off benefits.
Tech Stack
Categories
About Drata
Drata provides the trust network that enables businesses to operate, scale, and partner with confidence. Powered by AI and designed to operationalize trust, the Drata Agentic Trust Management Platform continuously interprets controls, risk, and assurance signals—reducing repetitive manual work while improving visibility into internal and third-party risk, enabling always-on audit readiness across compliance frameworks, and accelerating security reviews. Purpose-built for enterprise complexity, Drata unifies governance, risk, compliance, and assurance to deliver faster time-to-value, reduce operational overhead, and enable continuous trust for 8,000+ organizations worldwide.