Staff Machine Learning Engineer, Generative AI (Auth0)

Okta
Toronto
Workplace: HybridFull timeCAD 168,000 - 231,000 annuallyFunction: Data Science & Machine LearningExperience: 7+ yearsSkills: ["Collaboration","Mentorship","Technical decision-making","Iterative problem solving","Cross-functional leadership"]

Build and accelerate production-ready GenAI and ML systems for identity and authentication experiences. Architect scalable, observable LLM/GenAI pipelines spanning retrieval, inference, and evaluation, including RAG workflows and guardrails for accuracy, safety, and compliance. Develop scalable infrastructure and platform services, partnering across Product, Security, and Platform Engineering, while mentoring engineers and driving fast iteration toward reliable, enterprise-grade AI.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Okta
Okta
2 days ago

Staff Machine Learning Engineer, Generative AI (Auth0)

✓ Verified Job

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

Source: Company careers pageValidated by: Fursa AI
Last checked: 2 hours agoStatus: Live
Reposted: similar role first listed 11 months ago

Job Summary

Build and accelerate production-ready GenAI and ML systems for identity and authentication experiences. Architect scalable, observable LLM/GenAI pipelines spanning retrieval, inference, and evaluation, including RAG workflows and guardrails for accuracy, safety, and compliance. Develop scalable infrastructure and platform services, partnering across Product, Security, and Platform Engineering, while mentoring engineers and driving fast iteration toward reliable, enterprise-grade AI.
Location: Toronto
Workplace: Hybrid
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key Responsibilities

  • •Architect, design, and deploy robust ML/GenAI systems with scalable LLMOps pipelines in production.
  • •Develop and iterate on structured prompting, context retrieval, and RAG workflows to improve accuracy, safety, and cost efficiency.
  • •Build automated evaluation pipelines to measure quality, correctness, groundedness, and safety in production.
  • •Implement schema validation, structured output enforcement, and other guardrails to keep outputs reliable, auditable, and compliant.
  • •Mentor and coach engineers while partnering with Product, Security, and Platform Engineering to deliver secure, high-quality AI-powered experiences.

Pay and Benefits

Salary: CAD 168,000 - 231,000 annually
Perks:Health InsuranceDentalVisionRrsp MatchTelemedicinePaid LeaveParental Leave

Key Requirements

  • •7+ years of software development experience with strong programming expertise in Python (Go or TypeScript a plus).
  • •Hands-on experience applying machine learning across feature engineering, training, and fine-tuning models.
  • •Experience with modern Generative AI platforms such as AWS Bedrock, OpenAI, and Anthropic.
  • •Deep understanding of RAG, embeddings, and knowledge-base workflows.
  • •Hands-on experience with AI agent tooling/frameworks (e.g., LiteLLM, LangGraph, LangChain, LlamaIndex, MCP).
Experience:7+ yearsGenerative AIMachine learningLLMOps
Skills:CollaborationMentorshipTechnical decision-makingIterative problem solvingCross-functional leadership
Languages:English
Tech Stack:PythonGoTypeScriptAWS BedrockOpenAIAnthropicClaudeLiteLLMLangGraphLangChainLlamaIndexMCPFastAPIPyTorchTensorFlowSpark MLAirflow

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