Drive the quality of Drata's AI systems through experimentation and applied research on RAG, agents, and information retrieval. Own evaluation frameworks and translate validated approaches into production with AI and Software Engineers.
145k – 196k/yr
Hybrid3+ YOEML Engineering
About the role
What you'll do
Design and evaluate information access + reasoning strategies across RAG, agents, and classic ML: chunking, embedding models, hybrid search, metadata filtering, semantic routing
Prototype GenAI workflows (including agentic systems) that map and reason over compliance objects (controls ↔ risks ↔ requirements ↔ evidence)
Explore ML + probabilistic approaches where GenAI is not the best fit: classifiers, ranking models, graph/link prediction, calibration, and structured prediction
Build and maintain evaluation frameworks: golden datasets, automated quality metrics, regression detection
Designs, develops, and optimizes GenAI inference engine for LLMs, focusing on performance, scalability, and efficiency across the full stack. Requires 3+ years in performance-critical systems, CUDA/GPU expertise, and ML inference knowledge.
142k – 205k/yr
On-site3+ YOEML Engineering
Software Engineer, AI Platforms
FigmaSan Francisco, CA +1
Build scalable AI platforms and infrastructure for Figma's design tools, including model training, agentic features, and APIs. Requires 5+ years software engineering experience with backend/infrastructure and 3+ years in AI or developer platforms.
149k – 350k/yr
Remote5+ YOEML Engineering
Software Engineer, Machine Learning
FigmaSan Francisco, CA +1
Build and productionize ML models for search, RAG, and generative AI features at Figma. Requires 5+ years software engineering with 3+ years in applied ML, Python proficiency, and experience with scalable data pipelines.
149k – 350k/yr
Remote5+ YOEML Engineering
Perception & Fusion Engineer
Applied IntuitionArlington, VA
Develop and integrate real-time sensor perception and fusion algorithms for autonomous vehicles across land, air, sea, and space domains. Requires MS/PhD or 5+ years experience with multi-modal sensors (EO/IR/radar), ML deployment, and DoD clearance eligibility.
150k – 220k/yr
On-site5+ YOEML Engineering
Applied ML Engineer
DeepgramCalifornia
Own the research-to-production pipeline at Deepgram, turning experimental speech ML models into reliable, scalable production services. Partner with researchers on robust workflows, automated release gates, inference optimization, and feedback loops across hybrid GPU infrastructure.