Staff Machine Learning Engineer, SecureAI

Okta
Toronto
Workplace: HybridFull timeCAD 168,000 - 231,000 annuallyFunction: Data Science & Machine LearningExperience: 8+ yearsEducation: bachelorsSkills: ["Collaboration","Drive for excellence","Iterative problem-solving","Ownership","Speed and urgency"]

Build and evolve AI agent authorization capabilities as part of the Agent Access Policies sub-team. Replace static rule-based enforcement with dynamic, intent-based ML/GenAI security mechanisms that inspect runtime requests, evaluate agent intent, and analyze behavior in real time. Architect low-latency ML systems and RAG workflows across an API gateway, implement confidence-scored decision engines (e.g., Cedar), and create evaluation benchmarks, guardrails, and automated quality/safety pipelines to prevent bypasses and false positives.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Okta
Okta
4 days ago

Staff Machine Learning Engineer, SecureAI

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 2 hours agoStatus: Live

Job Summary

Build and evolve AI agent authorization capabilities as part of the Agent Access Policies sub-team. Replace static rule-based enforcement with dynamic, intent-based ML/GenAI security mechanisms that inspect runtime requests, evaluate agent intent, and analyze behavior in real time. Architect low-latency ML systems and RAG workflows across an API gateway, implement confidence-scored decision engines (e.g., Cedar), and create evaluation benchmarks, guardrails, and automated quality/safety pipelines to prevent bypasses and false positives.
Location: Toronto
Workplace: Hybrid
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key Responsibilities

  • •Architect and drive features for a unified control plane that delivers intent-based enforcement for AI agent runtime requests.
  • •Apply LLM reasoning and prompt parsing to interpret prompts, tool payloads, and intent in real time.
  • •Integrate low-latency inference/semantic evaluation into the API gateway request path and design confidence-scored decision engines feeding policy frameworks.
  • •Build scalable ML/GenAI systems that integrate retrieval, inference, and evaluation pipelines optimized for RAG workflows in Claude-based systems.
  • •Create evaluation benchmarks, guardrails, and automated evaluation pipelines to measure quality, correctness, groundedness, and safety; implement schema validation and structured output enforcement.

Pay and Benefits

Salary: CAD 168,000 - 231,000 annually
Equity and Bonus:Equity
Perks:Health InsuranceDentalVisionPensionTelemedicinePaid LeaveParental Leave

Key Requirements

  • •8+ years of software development experience with strong programming expertise in Python (Go or TypeScript a plus).
  • •Hands-on experience with applied machine learning (feature engineering through training and fine-tuning).
  • •Hands-on experience with modern Generative AI platforms such as AWS Bedrock, OpenAI, and Anthropic.
  • •Deep understanding of retrieval-augmented generation (RAG), embeddings, and knowledge-base workflows.
  • •Bachelor’s or Master’s degree in Computer Science or a related field.
Experience:8+ yearsGenerative AIRAGMachine learning
Education:Bachelor's in Computer Science or related field
Skills:CollaborationDrive for excellenceIterative problem-solvingOwnershipSpeed and urgency
Languages:English
Tech Stack:PythonGoTypeScriptAWS BedrockOpenAIAnthropicRAGEmbeddingsVector searchZero-shot classificationClaudeCedarFastAPIPyTorchTensorFlowSpark MLAirflowLiteLLMLangGraphLangChain

Company Brief

Okta
Provides identity and access management cloud solutions that help organizations secure and manage user authentication, single sign-on, multi-factor authentication, and lifecycle management across applications and devices.
Industry: Cybersecurity
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: San Francisco, United States
Founded: 2009
WebsiteLinkedIn