Senior Machine Learning Engineer (Identity Threat Protection)

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 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.

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.

The Identity Threat Protection Team

Join our Identity Threat Protection (ITP) Team, where we're redefining cloud authentication through cutting-edge AI and machine learning. Our ITP platform with Okta AI goes beyond traditional login security to provide real-time protection against evolving threats by seamlessly integrating with existing security infrastructure and analyzing user behavior and system data to proactively identify and mitigate risks.

The Senior MLE Opportunity

As a Senior Machine Learning Engineer, you'll be part of a dynamic team of full-stack engineers, data scientists, and ML engineers building a next-generation security platform that continuously evaluates threats and constructs comprehensive risk graphs. ITP is at the forefront of trust and security, prioritizing risk in authentication and authorization decisions to solve critical security problems at scale.

You'll join an engineering team that believes in designing for scale, rigorous code reviews, automated testing, and iterative processes to build high-quality enterprise software. Our team is fast, innovative, and flexible, with agile release cycles and individual ownership, offering you the opportunity to work on stimulating security projects, emerging technologies, and complex security and ML challenges.

Identity Threat Protection with Okta AI

What You'll Do

• Architect & Deploy ML Infrastructure: Design, build, and maintain robust, scalable ML infrastructure using modern tools (Airflow, MLFlow, feature stores, vector databases) while implementing automated CI/CD workflows for model training, validation, and deployment.
• Build Scalable Data Pipelines: Design and maintain ETL/ELT pipelines to ingest, process, and transform threat data from various sources into Snowflake data warehouse, ensuring data quality, security, and compliance for real-time threat detection.
• Collaborate & Evangelize: Work with cross-functional teams to align ML platform strategies with company objectives while acting as a data evangelist to foster data-driven culture and provide mentorship to analysts and stakeholders.
• Optimize & Monitor Systems: Deploy and monitor ML models with data scientists and engineers, troubleshoot infrastructure issues, and optimize performance of large-scale ML systems for threat prevention and access management.
• Drive Innovation & Security: Evaluate and adopt new ML technologies while ensuring data security, privacy, and compliance within the ML infrastructure, contributing to the comprehensive risk graph that powers ITP's threat detection capabilities.
  
  

What You'll Bring

• Architect & Deploy ML Infrastructure: Design, build, and maintain robust, scalable ML infrastructure using modern tools (Airflow, MLFlow, feature stores, vector databases) while implementing automated CI/CD workflows for model training, validation, and deployment.
• Build Real-Time Inference & Scoring Services: Design, deploy, and optimize high-performance ML inference platforms that deliver sub-second risk scores for authentication and authorization decisions. Build and maintain real-time scoring services that process high-volume security events, implementing advanced optimization techniques like quantization and model pruning to balance performance with resource efficiency. Ensure inference systems can handle tens of thousands of concurrent requests while maintaining low latency for threat detection.
• Build Scalable Data Pipelines: Design and maintain ETL/ELT pipelines to ingest, process, and transform threat data from various sources into Snowflake data warehouse, ensuring data quality, security, and compliance. Architect streaming data pipelines that enable real-time model inference for fraud detection, anomaly detection, and behavioral analysis.
• Collaborate & Evangelize: Work with cross-functional teams to align ML platform strategies with company objectives while acting as a data evangelist to foster data-driven culture and provide mentorship to analysts and stakeholders.
• Optimize & Monitor Systems: Deploy and monitor ML models with data scientists and engineers, troubleshoot infrastructure issues, and optimize performance of large-scale ML systems for threat prevention and access management. Implement comprehensive monitoring for inference throughput, latency, and resource utilization across distributed environments.
• Drive Innovation & Security: Evaluate and adopt new ML technologies while ensuring data security, privacy, and compliance within the ML infrastructure, contributing to the comprehensive risk graph that powers ITP's threat detection capabilities.

Nice to Have

• Experience with Java and/or Java EE web applications.
• Familiarity with authentication protocols, access management, or cloud security best practices.
• Exposure to prompt engineering, LLM ecosystems (Hugging Face, LangChain, vector databases), and security-focused ML applications.
  
  

#LI-Hybrid 

#BB-1

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:$130,000—$194,000 CAD

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/.