Staff/Principal Machine Learning Engineer, Identity Management

Okta; Toronto; On-site; USD 177,000 - 265,000; Full time

Job summary

A leading technology company is seeking a Staff/Principal Machine Learning Engineer in Toronto to spearhead the integration of AI/ML into identity management solutions. This role focuses on designing innovative solutions and collaborative technical leadership to deliver strategic business outcomes, enhancing Okta's position in the identity security landscape.

Benefits

Health, dental, and vision insurance

RRSP with a match

Paid leave including PTO and parental leave

Telemedicine options

Qualifications

• 5+ years of hands-on experience developing AI/ML solutions.
• Experience with LLM ecosystems like Hugging Face.
• Strong practical experience with cloud AI/ML services.

Responsibilities

• Lead design and deployment of AI/ML-driven features.
• Provide technical guidance on AI/ML technologies.
• Mentor engineers and support cross-functional collaboration.

Skills

AI/ML solution development

Prompt engineering

Machine Learning frameworks

Cloud AI services

Debugging and performance tuning

Education

Relevant graduate degree in Computer Science or related field

Tools

TensorFlow

PyTorch

Scikit-learn

AWS SageMaker

Job description

Staff/Principal Machine Learning Engineer, Identity Management

Staff/Principal Machine Learning Engineer, Identity Management

1 day ago Be among the first 25 applicants

Get AI-powered advice on this job and more exclusive features.

Get to know Okta

Okta is The World’s Identity Company. We free everyone to safely use any technology—anywhere, on any device or app. Our Workforce and Customer Identity Clouds enable secure yet flexible access, authentication, and automation that transforms how people move through the digital world, putting Identity at the heart of business security and growth.

Get to know Okta

Okta is The World’s Identity Company. We free everyone to safely use any technology—anywhere, on any device or app. Our Workforce and Customer Identity Clouds enable secure yet flexible access, authentication, and automation that transforms how people move through the digital world, putting Identity at the heart of business security and growth.

At Okta, we celebrate a variety of perspectives and experiences. We are not looking for someone who checks every single box - we’re looking for lifelong learners and people who can make us better with their unique experiences.

Join our team! We’re building a world where Identity belongs to you.

Staff/Principal Machine Learning Engineer, Identity Management

Location: Toronto, Canada

Job Description:

Okta seeks a visionary Staff/Principal Machine Learning Engineer to spearhead AI/ML integration into our identity management solutions. As the key AI/ML architect and strategist for our Identity Management pillar, you'll shape our product future (notably Okta Identity Governance), identify opportunities, design innovative solutions using cutting-edge techniques like LLMs, and guide engineering teams. You will solve complex identity challenges, deliver significant customer value, and define the AI strategy for one of Okta's fastest-growing product areas.

What You'll Do (Responsibilities):

• Lead and Architect: Drive the end-to-end design, development, and deployment of AI/ML-driven features within the Identity Management platform, ensuring measurable business outcomes and alignment with product strategy.
• Strategic Guidance: Serve as the primary technical advisor on AI/ML matters for the Identity Management pillar, evaluating new technologies (including LLMs, vector databases, ML frameworks) and defining best practices.
• Technical Leadership & Mentorship: Engage in deep technical design discussions, conduct code reviews, and mentor engineers across the pillar, uplifting the team's capabilities in applying AI/ML to identity problems.
• Cross-functional Collaboration: Work closely with Product Management, architects, and other engineering teams to integrate AI/ML capabilities seamlessly into the product roadmap and user experience.
• Optimization & Refinement: Apply optimization techniques to enhance model performance and accuracy; utilize enterprise data to iteratively refine AI models specific to identity use cases.
  
  

What You'll Bring (Qualifications):

• 5+ years of hands-on experience developing and deploying impactful AI/ML solutions (including deep learning) using frameworks like TensorFlow, PyTorch, or Scikit-learn.
• Experience with prompt engineering and leveraging LLM ecosystems (e.g., Hugging Face, LangChain, vector databases).
• Proven track record of designing, training, validating, deploying, and monitoring ML models that achieve measurable business outcomes.
• Strong practical experience with cloud AI/ML services (AWS SageMaker and AWS Bedrock preferred).
• Proficiency in debugging, performance tuning, and optimizing large-scale data processing and ML models.
• Demonstrated ability to collaborate effectively with product owners, tech leads, and cross-functional teams; experience mentoring engineers, building trust with senior leaders, and acting as a strategic advisor.
  
  

Nice to Haves

• Experience in Java and/or Java EE web applications
• Experience or strong understanding of cloud security, identity management (IAM), and/or authentication protocols is highly advantageous.
• Experience with MLOps practices, including containerization (Docker), orchestration (Kubernetes), and CI/CD pipelines tailored for AI/ML workflows.
  
  

Below is the annual salary range for candidates located in Canada. Your actual salary will depend on factors such as your skills, qualifications, and experience. 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. To learn more about our Total Rewards program, please visit: https://rewards.okta.com/can.

The annual base salary range for this position for candidates located in Canada is between:

$177,000—$265,000 USD

What you can look forward to as a Full-Time Okta employee!

• Amazing Benefits
• Making Social Impact
• Developing Talent and Fostering Connection + Community at Okta
  
  

Okta cultivates a dynamic work environment, providing the best tools, technology and benefits to empower our employees to work productively in a setting that best and uniquely suits their needs. Each organization is unique in the degree of flexibility and mobility in which they work so that all employees are enabled to be their most creative and successful versions of themselves, regardless of where they live. Find your place at Okta today! https://www.okta.com/company/careers/.

Some roles may require travel to one of our office locations for in-person onboarding.

Okta is an Equal Opportunity Employer. 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.

If reasonable accommodation is needed to complete any part of the job application, interview process, or onboarding please use this Form to request an accommodation.

Okta is committed to complying with applicable data privacy and security laws and regulations. For more information, please see our Privacy Policy at https://www.okta.com/privacy-policy/.

Seniority level

• Seniority level

  Not Applicable

Employment type

• Employment type

  Full-time

Job function

• Job function

  Education and Training

• Industries

  Computer and Network Security, Software Development, and IT Services and IT Consulting

Referrals increase your chances of interviewing at Okta by 2x

Get notified about new Headmaster jobs in Toronto, Ontario, Canada.

Principal & Dr. Daniel C. Andreae Chair in Transformational Healthcare Education

Principal Associate, User Research & VoC

Principal, Governance, Legal and Compliance Consulting Practice

Principal Program Lead, Real Estate Market Strategy & Strategic Insights

Principal, Total Fund Management - Global Macro Insights

Head of Canada Programs, Munich Re Specialty

Greater Toronto and Hamilton Area, Ontario, Canada 2 weeks ago

Principal Scheduling Professional, Rail and Transit

Assistant Head of Claims - Full Time, Hybrid (Anywhere in Canada)

Senior Principal Construction Research Scientist:

We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.