Staff Machine Learning Engineer, SecureAI

Secure Every Identity, from AI to Human

Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence.

This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk.

The Okta for AI Agents Team is building the future of digital identity management. The way companies are using agents and allowing them access is undergoing a fundamental shift, enabling teams to move fast while staying secure.

Our North Star is clear: To get AI right, you have to get identity right. We are not just managing identities; we are building the industry's first Identity Security Fabric for the agentic era. As organizations rapidly deploy AI, these non-human entities are acting with access to critical tools, often bypassing traditional perimeters and creating a 'Shadow AI' crisis. Our mission is to move identity from a reactive gatekeeper to a unified control plane, transforming AI risk into business ROI.

We are tackling this by implementing a comprehensive four-pillar maturity model: Discover, Onboard, Protect, and Govern. We are driving innovation by defining the standards for Agentic Identity—including pioneer work with securing AI Agents and support for protocols like MCP. You will be helping us build the infrastructure that allows enterprises to scale AI safely, ensuring every access decision—whether made by a human or an automated agent—is authenticated, authorized, and audited at scale.

We are a diverse team of engineers, product managers, and designers who are bringing Okta’s expertise in identity to define the future of Agent Identity management, focusing on enablement while staying secure.

In this role, you will join the Agent Access Policies sub-team within Okta Secures AI as a Staff Machine Learning (ML) Engineer to advance Okta's authorization capabilities. By replacing static, rule-based systems with dynamic AI security mechanisms, you will deliver real-time threat inspection, agent intent evaluation, and behavioral analysis—ensuring autonomous AI agents operate safely within specified bounds.

Your core mission will be architecting and driving features for our unified control plane to establish the premier Identity Security Fabric for the agentic era. In doing so, you will help execute our guiding principle: "To get AI right, you have to get identity right."

• Implement intent-based enforcement to verify agent runtime requests match their intended purpose, requiring key capabilities.

• Apply LLM reasoning and prompt parsing to interpret prompts, tool payloads, and intent in real time.

• Integrate low-latency inference or semantic evaluation engines directly into the API gateway request path.

• Use embeddings, vector search, or zero-shot classification to score alignment between agent intent and executed actions.

• Design confidence-scored decision engines that feed semantic verification results into policy frameworks (e.g., Cedar).

• Establish evaluation benchmarks, prompt injection defenses, and guardrails to prevent bypasses or false positives.

• Architect scalable ML and Generative AI systems integrating retrieval, inference, and evaluation pipelines.

• Optimize prompting, context retrieval, and RAG workflows for accuracy, safety, and efficiency in Claude-based systems.

• Build automated evaluation pipelines to measure model quality, correctness, groundedness,