Staff AI Engineer
At Drata, we’re rethinking how compliance gets done, and AI is a core part of that. We’re looking for a Staff AI Engineer to help shape how intelligent systems power trust-critical enterprise workflows. This isn’t a “build a feature and move on” role. You’ll help define how AI is architected across the company. You’ll make foundational decisions about what we build, how we measure quality, and where we invest. From early research to production deployment, you’ll own systems end-to-end, and your judgment will directly influence our technical direction. You’ll work at the intersection of LLMs, retrieval systems, agentic workflows, and real-world compliance challenges, partnering closely with product, platform, and security teams to build AI that’s powerful, reliable, and responsible.
Job Details
Responsibilities
- Design and own production AI systems end-to-end
- Evolve our AI stack over time
- Define how we measure success across the platform
- Mentor senior engineers
- Partner with product and compliance teams
Requirements
- 10+ years of software engineering experience
- 3+ years working directly on ML/AI systems
- Experience with RAG, embeddings, vector databases
- Strong Python skills
- Clear communication and cross-functional collaboration
Skills & Technologies
Related Opportunities
Discover more opportunities that match your interests and skills