← All roles

Staff Machine Learning Engineer, Generative AI (Auth0)

Okta ·Toronto, Ontario, Canada ·Hybrid 9d ago
PythonGoTypeScriptLangChainFastAPIPyTorchTensorFlowApache Airflow

About the role

Okta is seeking a Staff Machine Learning Engineer to join the Generative AI team. The role involves architecting and deploying ML/GenAI systems, leading initiatives for agentic applications, and collaborating with product and security teams to deliver AI-powered experiences. The engineer will work on scalable infrastructure, RAG workflows, evaluation pipelines, and mentor team members.

Requirements

7+ years software development, strong Python, hands-on ML experience, GenAI platforms (AWS Bedrock, OpenAI, Anthropic), RAG, embeddings, AI agent frameworks (LiteLLM, LangGraph, etc.), ML frameworks (PyTorch, TensorFlow), evaluation pipelines, collaboration skills. Nice to have: identity/security integration, ethical AI, synthetic data.

About the company

Okta

Okta is a leader in the identity and access management industry, providing a unified platform to secure all types of identities. Their solutions enable organizations to manage access for employees, customers, and partners, ensuring secure and seamless experiences across applications and devices.

View company page →

More from Okta

Staff Client Platform Engineer
iOSLinuxTerraformAnsibleBashPythonGoSwiftGitOIDCAWS Lambda
$194k–267k
Hybrid · San Francisco, California
Staff Client Platform Engineer
$194k–267k
Hybrid · San Francisco, California
iOSLinuxTerraformAnsibleBashPythonGoSwiftGitOIDCAWS Lambda
Staff Software Engineer - Federal
PythonBashGoTerraformAWSGCPAzureInfrastructure as Code
$161k–221k
Hybrid · Chicago, Illinois; New York, New York; Seattle, Washington; Washington, DC
Staff Software Engineer - Federal
$161k–221k
Hybrid · Chicago, Illinois; New York, New York; Seattle, Washington; Washington, DC
PythonBashGoTerraformAWSGCPAzureInfrastructure as Code
Staff Site Reliability Engineer, Federal (TS/SCI)
GoPythonTerraformKubernetesAWSGCPHelmPostgreSQLRedisMySQLArgoCD
$174k–238k
Hybrid · Maryland; Washington, DC
Staff Site Reliability Engineer, Federal (TS/SCI)
$174k–238k
Hybrid · Maryland; Washington, DC
GoPythonTerraformKubernetesAWSGCPHelmPostgreSQLRedisMySQLArgoCD
Senior Software Engineer, PAM
KubernetesAmazon EKSDynamoDBAmazon KinesisRedisDockerTerraformAWSGoJavaC++
CA$136k–187k
Onsite · Toronto, Ontario, Canada
Senior Software Engineer, PAM
CA$136k–187k
Onsite · Toronto, Ontario, Canada
KubernetesAmazon EKSDynamoDBAmazon KinesisRedisDockerTerraformAWSGoJavaC++
Staff Product Security Engineer, Reviews
OWASPJavaGoPythonOIDCiOSAndroid
CA$141k–193k
Hybrid · Toronto, Ontario, Canada
Staff Product Security Engineer, Reviews
CA$141k–193k
Hybrid · Toronto, Ontario, Canada
OWASPJavaGoPythonOIDCiOSAndroid
CA$168k–231k
Hybrid Toronto, Ontario, CanadaToronto, Canada
Posted 9d ago
Apply for this role
Opens www.okta.com ↗
About the company
Okta
https://www.okta.com

Okta is a leader in the identity and access management industry, providing a unified platform to secure all types of identities. Their solutions enable organizations to manage access for employees, customers, and partners, ensuring secure and seamless experiences across applications and devices.

View company page →