Secure Every Identity, from AI and 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.

Secure Every

Identity, from AI and 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.

About Okta

Secures AI Team 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 rapid while staying secure. 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 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. 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.

Design confidence-scored decision engines that feed semantic verification results into policy frameworks (e.g., Architect scalable ML and Generative AI systems integrating retrieval, inference, and evaluation pipelines. Build automated evaluation pipelines to measure model quality, correctness, groundedness, and safety in production. Implement schema validation, structured output enforcement, and guardrails to ensure reliable, compliant AI outputs. 8+ years of software development experience, with strong programming expertise in Python (and familiarity with Go or Typescript a plus). ~ Hands-on experience with applied machine learning, from feature engineering to training and fine-tuning models. ~ Hands-on experience with modern Generative AI platforms (AWS Bedrock, OpenAI, Anthropic, etc.). ~ Hands-on experience with LiteLLM, LangGraph, LangChain, LlamaIndex, MCP, or other related AI agent frameworks. ~ Familiarity with ML frameworks (FastAPI, PyTorch, TensorFlow, Spark ML) and workflow orchestration tools (Airflow, etc.). ~ Experience defining evaluation metrics, pipelines, and feedback loops for ML/GenAI systems. ~ Proven ability to collaborate with product and engineering teams to drive greenfield initiatives forward,



navigate unknowns, and iterate quickly and frequently. ~ Experience building tools or infrastructure for AI/ML applications, with a deep understanding of the developer lifecycle in an AI-native world. ~ Bachelor's or Master's degree in Computer Science or related field ~ Experience integrating AI-driven systems with identity, authentication, or security products. ~ Exposure to ethical AI, model risk, or compliance frameworks. ~ Familiarity with evaluation datasets, synthetic data generation, or LLM-as-a-judge methods.

In addition, Okta offers equity (where applicable), bonus, and benefits, including health, dental, and vision insurance, RRSP with a match, healthcare spending, telemedicine, and paid leave (including PTO and parental leave) in accordance with our applicable plans and policies. The annual base salary range for this position for candidates located in Canada is between:: $168,000 CAD - $231,000 CAD Driving Social Impact Developing Talent and Fostering Connection + Community Our global community, spanning over 20 offices worldwide, is united by a drive to innovate. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, ancestry, marital status, age, physical or mental disability, or status as a protected veteran.

We also consider for employment qualified applicants with arrest and convictions records, consistent with applicable laws.

Notice for New York City Applicants & Employees: Okta may use Automated Employment Decision Tools (AEDT), as defined by New York City Local Law 144, that use artificial intelligence, machine learning, or other automated processes to assist in our recruitment and hiring process. In accordance with NYC Local Law 144, if you are an applicant or employee residing in New York City, pleaseclick here to view our full NYC AEDT Notice. #

📌 Staff Machine Learning Engineer, SecureAI (Toronto)
🏢 Okta
📍 Toronto