Staff Machine Learning Engineer, Generative AI (Auth0)
$154k - $230k per yearOkta
Staff Machine Learning Engineer, AIFS (Auth0)
Join to apply for the Staff Machine Learning Engineer, AIFS (Auth0) role at Okta .
The Team
The AI Frameworks and Services team is at the forefront of enabling AI-driven innovation across our organization. We build the foundational frameworks and essential services that empower engineering teams to integrate cutting-edge AI capabilities into our products, transforming how our users interact with them. We collaborate closely with the Machine Learning team and various product teams to ensure the seamless and secure delivery of AI-enhanced features that provide real value to our users.
The Opportunity
As a Staff Machine Learning Engineer on the Generative AI team, you will help shape and accelerate our Generative AI strategy by contributing across the stack of model development, infrastructure, and platform services. You’ll drive design and implementation of production-ready AI/ML systems at scale—ranging from LLM-powered features to reusable platform components that other teams across Okta can build on.
- Design and implement scalable infrastructure and platform services for large-scale Generative AI use cases.
- Drive technical decision making while balancing simplicity, flexibility, reliability, and performance.
- Architect, design, and deploy robust Machine Learning & GenAI systems, ensuring seamless integration with diverse platform services and scalable LLMOps pipelines in production.
- Lead initiatives to tune, optimize, and deploy agentic applications in production with a focus on performance, reliability, and security.
- Collaborate with product, platform engineering, and security teams to ensure solutions are production-ready, resilient, and aligned with enterprise needs.
What you will do
- Design and implement components across different layers of the stack using technologies such as Python, LangGraph, DSPy, and MCP.
- Collaborate with product teams to shape the future of Agentic AI and Auth for GenAI.
- Build reliable and scalable systems to run data processing pipelines, LLMOps infrastructure, and model inference systems.
- Utilize containerization technologies like Docker and orchestration tools like Kubernetes to manage and deploy these systems efficiently.
- Collaborate cross-functionally with product managers, researchers, and engineers to deliver secure, high-quality, and scalable AI/ML systems.
- Mentor and coach engineers, contributing to the growth of the team and the larger engineering community.
What you bring
- Strong programming expertise in Python (and familiarity with Go or Typescript a plus).
- Proven experience building and deploying AI/ML systems at scale, including LLMOps, CI/CD, and infrastructure-as-code.
- Hands-on experience with modern Generative AI platforms (AWS Bedrock, OpenAI, Anthropic, etc.).
- Deep understanding of retrieval-augmented generation (RAG), embeddings, and knowledge-base workflows.
- Familiarity with ML frameworks (FastAPI, PyTorch, TensorFlow, Spark ML) and workflow orchestration tools (Airflow, etc.).
- Experience with LiteLLM, LangGraph, LangChain, DSPy, LlamaIndex, MCP, or other related AI agent frameworks.
- Experience with containerized applications (Docker, Kubernetes) and cloud-native environments.
- 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.
Nice To Haves
- Experience with identity and auth products.
- Experience creating tools that integrate with AI-based coding platforms.
- Experience with packaging and distributing containerized applications using Docker and Kubernetes.
- Familiarity with identity/authentication products and how they intersect with AI.
- Background in building AI-native developer tools or SDKs.
Please note that we encourage candidates to apply even if you do not have experience with all of the criteria or technologies listed above; these are provided to give insight into the tech stack and general responsibilities for the role.
The annual base salary range for this position for candidates located in Canada is between: $154,000—$230,000 CAD. 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: can.
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 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 Personnel and Job Candidate Privacy Notice at legal/personnel-policy/.
Okta is The World’s Identity Company. We free everyone to safely use any technology, anywhere, on any device or app. Our flexible and neutral products, Okta Platform and Auth0 Platform, provide secure access, authentication, and automation, placing identity at the core of business security and growth.
Some roles may require travel to one of our office locations for in-person onboarding.
Referrals increase your chances of interviewing at Okta by 2x
Find your place at Okta today! company/careers/
Seniority level: Not Applicable
Employment type: Full-time
Job function: Engineering and Information Technology
Industries: Computer and Network Security, Software Development, and IT Services and IT Consulting
#J-18808-Ljbffr$168k - $231k per year
# Staff Machine Learning Engineer, Generative AI (Auth0)Toronto, Ontario, Canada**Secure Every Identity, from AI to 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...Suggested- ...responsibly propel the global lottery industry ever forward. Position Summary About the Role We are looking for a Staff Machine Learning Engineer to define and build the machine learning platform architecture for the organization. This team will create the enabling...SuggestedLocal areaRemote work
$170.4k per year
...precision. With 465 billion automated optimizations per second, the AI-powered StackAdapt Marketing Platform seamlessly connects... ...and marketing channels. We're looking to add Senior and Staff Machine Learning Engineers to our Data Science team! This team works on...SuggestedLocal areaRemote workWork from homeHome office$197k per year
...team builds the backend systems, deep learning models & user experience components behind... ...the executive leadership team. As a Staff Software Engineer, you’ll set technical direction for... ...power search. You’ll partner closely with machine learning, science, mobile, and other...SuggestedFull timeWork at officeRemote work$65 - $75 per hour
...Senior Data Scientist (Machine Learning & Generative AI) Bring your passion for data, machine learning, and GenAI to the insurance sector for our client. Help transform complex business problems into practical, innovative solutions while contributing to advanced analytics...SuggestedHourly payFull timeContract work- ...are curious, accountable, eager to learn, and supportive of their teammates—because... ...Project X Ltd. is seeking a Staff-level, hands-on software and ML engineering professional for an embedded, client... ...build production integrations between AI agents, enterprise APIs, agent orchestration...Full time
$120k - $250k per year
...application in its entirety. Layer 6 is the AI center of excellence for TD Bank Group. We develop and deploy industry-leading machine learning systems that impact the lives of over 27... ...with areas such as deep learning and generative AI, time series forecasting and responsible...Temporary workWork from home$142.4k - $190.1k per year
...continues to invest in the future of circular commerce and the next generation of conscious consumers. With the recent acquisition of Depop... ...intelligent, scalable experiences powered by the latest Machine Learning, AI, NLP, LLM/GenAI, and RAG technologies . We work with large-...Immediate start$160k - $220k per year
...Secure Every Identity, from AI to Human Identity is the key to unlocking the potential... .... If you are too, let's talk. The Auth0 Team Auth0 is an easy-to-implement... ...of defense for Auth0 customers. The Staff Software Engineer At Okta, we’re building the next generation...Full timeLocal areaWorldwide$235k - $270k per year
...Waabi, founded by AI visionary Raquel Urtasun, is the leader in Physical AI. With a world... ...to impact the world in a positive way. To learn more visit: Waabi is bringing its... ..., acting as the connective tissue between engineering, operations, and external partners. You will...InternshipLocal areaRemote work$118.8k - $148.5k per year
...manages over 1.7 million monthly human and AI interactions and serves as Lyft's... ...together ML, data, backend, and product engineers alongside data scientists and operations... ...partners to transform these systems. As a Machine Learning Engineer on the SCC team, you will fine-...Hourly payWork at officeFlexible hours3 days per week- ...human creativity. We build proprietary generative media models and AI native creative workflows, tackling... ...taste, and craft as much as research and engineering – shipping experiences that creatives... ...~ You’re familiar with modern deep learning such as transformers, diffusion, and...Full timeWork at office
- ...makes Stripe one of the best places to do machine learning. While being an integral part of almost... ...platforms and services that enable ML engineers and data scientists across Stripe to take... ...observability and monitoring, and agentic AI capabilities. We work closely with...Long term contractFlexible hours
- ...models to shipping in production. You can learn more about our team’s work from this recent talk . What you’ll do As a machine learning engineer, you will be responsible for analyzing... ...Engage with the latest developments in ML/AI and take calculated risks in...Full time
- ...Waabi, founded by AI visionary Raquel Urtasun, is the leader in Physical AI. With a world-class team, we... ...who want to impact the world in a positive way. To learn more visit: As a Research Engineer focused on Generative AI Deployment, you will own the path from research...Full time
$220k - $260k per year
...Select how often (in days) to receive an alert: Machine Learning Engineer, Associate Director Requisition ID: 49267 Business Unit: Fitch Group... ...21, 2026 Machine Learning Engineer, Associate Director – AI Innovation Teams Fitch Ratings is seeking a Machine...Long term contractImmediate start$400k per year
...Job Description: Position: Machine Learning Engineer Location: Remote / Hybrid Employment Type: Full-time Responsibilities Design, train... ...Requirements Bachelor’s or Master’s degree in Computer Science, AI, Data Science, or related discipline. Strong foundation in...Full timeRemote workRelocation$160k - $220k per year
...Secure Every Identity, from AI to Human Identity is the key to... ...The Developer Platform team at Auth0 (an Okta company) owns the surfaces... ...it. We're looking for an engineer who already builds with those skills... ...opportunity We're hiring a Staff Software Engineer to build the...Local areaWorldwideShift work$150k per year
...Get AI-powered advice on this job and more exclusive features. Base pay range... ...and collaborative team of researchers and engineers who thrive on pushing the boundaries of... ...skilled, detail-oriented, and passionate Machine Learning Engineer to join our enterprise team. In...Full time- ...It all started when engineer Fred Luddy wrote code that automated a tedious task for his... ...meaningful work. Today, ServiceNow is the AI control tower for business reinvention. Our... ...people. Job Description The Machine Learning Developer designs, builds, ships, and operates...Full timeWork at officeLocal areaImmediate startRemote workFlexible hours
$123k per year
...Job Requisition ID # 26WD Position Overview We are looking for an exceptional Machine Learning Engineer to design, build, operationalize, and scale production-grade AI/ML and Agentic AI systems at Autodesk For Go-to-market intelligence function. The mission of the team...Remote work$190.1k per year
...ON Full-time CA$142,400 - CA$190,100 / year eBay is hiring a Machine Learning Engineer for Depop, its peer-to-peer circular fashion marketplace. The team develops and productionizes machine learning and AI solutions using large-scale marketplace data. What you'll do...Full time$160k - $220k per year
...Secure Every Identity, from AI to Human Identity is the key to... ...s talk. The Product Okta’s Auth0 is an easy-to-implement authentication... ...and coach less experienced engineers on sound engineering practices... ...plans and policies. To learn more about our Total Rewards program...Local areaWorldwide$110k - $130k per year
...to Work! Your Role Clarius Mobile Health is seeking a Machine Learning Engineer to contribute to a special project focused on expanding access to ultrasound technology while advancing our next-generation innovations. Over the next 24 months, you'll focus on the development...Permanent employmentFull timeContract workFixed term contractWork at officeRemote workWorldwide$208k - $286k per year
...Secure Every Identity, from AI to Human Identity is the key to... ...business security and growth. At Auth0, an Okta company, we help... ...You will work across Product, Engineering, Security, Developer Relations,... ...applicable plans and policies. To learn more about our Total Rewards...Long term contractFull timeLocal areaWorldwideFlexible hoursShift workEarly shift$149.6k - $187k per year
...disrupt transportation, building a next-generation platform for low-cost, ultra-immersive... ...a critical thinker with experience in machine learning workflows and LLMs, passionate about solving... ...are seeking a Senior Machine Learning Engineer to join the Rider Applied AI team and...Hourly payWork at officeFlexible hours3 days per week- ...We are looking for a Machine Learning Engineer to join our Toronto team and help us take our products to the next level in terms of visual intelligence. Our Company Invision AI is building a universal AI platform for computer vision applications. Powered by a unique...Full timeInternshipManual laborWork at office3 days per week
$170k - $200k per year
...revenue growth. As a Senior ML Engineer within our Benefits... ...role, we would really like to learn more about you. Lettuce share... ...collaborating with Data Scientists, Machine Learning Engineers, Data Engineers... ...Leverage Claude Code and other AI tools as productivity multipliers...Long term contractFull timeWork at officeRemote workWork from homeFlexible hours2 days per week3 days per week$140k - $180k per year
...marketers through data-activated content generation and AI decisioning. The world’s most innovative... ...team where AI isn’t a buzzword—it’s the engine of our success. We’re looking for a... ...development, with a focus on architecting deep learning systems and sophisticated model...- ...We’re looking for Machine Learning Engineers who are passionate about building cutting-edge systems with LLMs and real-world data. In this role, you... ...working with the Python ML stack: Pandas, Numpy, scikit-learn, XGBoost, PyTorch, etc. Proficient with development tools such...
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Staff Machine Learning Engineer, Generative AI (Auth0). Be the first to apply!
- assistant electrical engineer Toronto, ON
- project engineer assistant project manager Toronto, ON
- machine learning engineer Toronto, ON
- junior machine learning developer Toronto, ON
- machine learning researcher Toronto, ON
- machine learning part time Toronto, ON
- software engineer - ai machine learning Toronto, ON
- machine learning Toronto, ON
- intern quantum machine learning for quantum computing Toronto, ON
- assistant electrical engineer