Staff Machine Learning Engineer at Okta

5 hours ago

Winnipeg, Manitoba, Canada Okta Inc. Full-time
Transform identity security with Okta as a Staff Machine Learning Engineer focused on Generative AI in Toronto. This role combines ML architecture, AI system design, and cross-functional collaboration.

Join the innovative Generative AI team at Okta in Toronto, Ontario. As a Staff Machine Learning Engineer, you will architect and deploy advanced ML solutions, ensuring seamless integration with platform services. Your responsibilities will include improving ML systems for performance, reliability, and security, while you mentor team members and drive impactful initiatives.

Key Responsibilities: • Design scalable, observable ML and Generative AI systems • Develop context retrieval and RAG workflows for efficiency • Build automated evaluation pipelines to maintain model quality • Implement schema validation and compliance guardrails • Collaborate with product teams to create innovative AI experiences

Requirements: • 7+ years of experience in software development • Proficiency in Python; familiarity with Go or TypeScript desirable • Hands-on with Generative AI platforms like AWS Bedrock • Strong background in retrieval-augmented generation • Experience with ML frameworks and orchestration tools

Elevate your career by shaping the future of AI-powered identity solutions at Okta.