Staff Security Software Engineer, AI SecurityActive

Databricks
Apply now

The opportunity

The AI Security team at Databricks sits at the frontier of securing the AI/ML services in the Databricks platform. As we ship AI capabilities at the leading edge of the industry, including Agent Bricks, the Genie suite, AI Model Serving, MLflow, and Unity AI Gateway, the AI…

What you'll do

  • Lead AI red team engagements against Databricks' production AI systems,: including Foundation Model APIs, Genie and natural language query systems, Model Serving infrastructure, MCP-connected agents, and RAG pipelines

  • Design and execute adversarial attack scenarios: prompt injection, jailbreaking, memory poisoning, cross-tenant data leakage in multi-tenant serving, and sandbox bypasses

  • Develop proof-of-concept exploits for AI-specific vulnerability classes and: perform variant analysis to identify the full scope of exposure across the AI platform

  • Contribute to the evolution of the Databricks AI Security Framework (DASF),: maintaining and extending the risk taxonomy, control library, and testing methodology as AI capabilities evolve

  • Lead comprehensive security architecture reviews for complex AI features: threat modeling agentic workflows, RAG pipelines, multi-model serving chains, and MCP-based tool integrations

  • Partner directly with AI and ML engineering teams to identify security risks: early in the design process and define practical, scalable controls

What they're looking for

  • Assess and drive resolution of cross-cutting AI security risks: Unity Catalog permission enforcement in AI contexts, inference data isolation, model artifact integrity, fine-tuning pipeline security, and external model API governance via AI Gateway
  • Identify recurring security patterns across AI features; advocate for: class-level architectural fixes rather than feature-by-feature point solutions
  • Design and build automated AI security testing tooling, including adversarial: prompt libraries, agent behavior analysis frameworks, and continuous testing harnesses
  • Build AI-assisted automation that scales security reviews, threat modeling,: and vulnerability triage for AI features
Staff Security Software Engineer, AI Security at Databricks | Role Match