Databricks

Staff Security Software Engineer, AI Security

Databricks$130K — $180K *
US-AnywhereRemote in California, US
Information Technology
5 - 7 years of experience
Job Overview by Ladders

Qualifications

  • 7-10 years of experience in offensive security or AI/ML security research.
  • Expertise in at least two AI security domains, including generative AI and AI infrastructure.
  • Ability to design and execute adversarial attacks on production AI systems.
  • Deep understanding of AI/ML architecture and its trust boundaries.
  • Proficient in Python and familiar with another programming language (Go, Java, Scala, Rust).
  • Demonstrated experience in cross-team security improvements and influencing architectural decisions.

Responsibilities

  • Lead AI red team engagements on Databricks' production AI systems.
  • Design and execute adversarial attack scenarios, such as prompt injection and cross-tenant data leakage.
  • Develop proof-of-concept exploits for AI vulnerabilities and assess the overall exposure.
  • Conduct comprehensive security architecture reviews for complex AI features.
  • Collaborate with AI engineering teams to identify and manage security risks early.
  • Automate AI security testing and build protective guardrails and measures.
  • Mentor team members in adversarial techniques and contribute to security training resources.

Benefits

  • Comprehensive benefits and perks tailored to meet employee needs.
Full Job Description
RDQ426R108

This role is open to candidates in the US (any location)
The Role

As a Staff Security Software Engineer on the AI Security team, you are a senior technical leader who sets the standards for how Databricks secures its AI and ML capabilities. You combine deep offensive security expertise with practical knowledge of AI/ML systems to identify and drive resolution of the most significant security risks in Databricks' AI platform.

You lead AI red team engagements against production AI systems, conduct security architecture reviews for complex, multi-system AI features, and build the tooling and frameworks that scale the team's impact. You are a subject matter expert in at least two AI security domains and you operate with significant autonomy- driving cross-team remediation, setting technical standards, and mentoring teammates in both offensive techniques and secure AI design.

The Impact You Will Have
AI Red Team & Adversarial Testing
  • 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
AI Product Security & Architecture Reviews
  • 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
  • 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
AI Security Tooling & Automation
  • 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
  • Develop and maintain security guardrails and enforcement mechanisms: LLM-as-judge review, prompt delimiting, output validation, rate limiting, and audit logging
Cross-Team Remediation & Standards
  • Set technical standards for how AI security risks are assessed, prioritized, and remediated across the engineering organization
  • Drive cross-team remediation for significant AI security findings, defining fix requirements, validating patches, and ensuring regression coverage in CI/CD pipelines
  • Produce high-quality threat models, security advisories, and post-mortems that inform organizational risk decisions for AI products
Mentorship & Community
  • Mentor engineers on the AI Security team in adversarial ML techniques, AI threat modeling, and security tooling development
  • Contribute to internal knowledge assets, including training materials, design patterns, and threat model templates, that raise AI security fluency across the engineering organization
  • Represent Databricks in the external AI security community through publications, conference talks, or open-source contributions


What We Look For
  • 7-10 years of combined experience in offensive security, AI/ML security research, or product security engineering, with demonstrated leadership in securing complex systems
  • Subject matter expert in at least two of the following AI security domains:

- LLM and generative AI security (prompt injection, jailbreaking, training data extraction)

- AI agent and orchestration security (MCP, memory sharing, multi-agent systems)

- ML infrastructure and serving security (model serving multi-tenancy risks, training infrastructure security)

- AI data governance and privacy (fine-grained access control, data residency, inference data isolation)
  • Demonstrated ability to design and execute adversarial attacks against production AI systems
  • Deep understanding of AI/ML platform architecture- how models are trained, served, and integrated, and where the trust boundaries between components lie
  • Expert in at least one major cloud platform (AWS, Azure, GCP) and its AI/ML security model
  • Proficient in Python; able to read and analyze ML model code, training scripts, and API serving code; working knowledge of at least one additional language (Go, Java, Scala, Rust)
  • Track record of driving cross-team AI security improvements and influencing product architecture decisions
  • Experience building automated security tooling for AI systems
  • Strong communicator- translates AI security risks into actionable guidance for engineers, product managers, and leadership
  • Pragmatic approach to risk- distinguishes real-world exploitable AI risk from theoretical concerns
Nice to Have
  • Published research on AI/ML security topics or experience presenting at AI security venues (DEF CON AI Village, NeurIPS workshops, Black Hat)
  • Experience with OWASP Top 10 for LLMs, MITRE ATLAS, or similar AI security frameworks
  • Familiarity with MLflow, Unity Catalog, Delta Lake, or Databricks platform internals
  • OSCP or equivalent offensive security certification
  • Academic or research background in machine learning, adversarial ML, or AI safety


BenefitsAt Databricks, we strive to provide comprehensive benefits and perks that meet the needs of all of our employees. For specific details on the benefits offered in your region click here.

About Databricks

Databricks is a unified analytics platform that provides data engineering, collaborative data science, and machine learning capabilities. The company was founded in 2013 by the original creators of Apache Spark, a popular open-source big data processing engine. Databricks provides a cloud-based platform that allows data teams to collaborate and build data pipelines, run machine learning models, and perform advanced analytics. The company has raised over $1 billion in funding and is valued at $38 billion as of November 2021.
Learn more about Databricks
Size
2,000 employees
Industry
Founded
2013

Similar Jobs

More Jobs at Databricks

More Information Technology Jobs

Find similar Staff Security Software Engineer, AI Security jobs: