← All roles
Staff Machine Learning Engineer, Generative AI (Auth0)
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
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
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
GoPythonTerraformKubernetesAWSGCPHelmPostgreSQLRedisMySQLArgoCD
Senior Software Engineer, PAM
KubernetesAmazon EKSDynamoDBAmazon KinesisRedisDockerTerraformAWSGoJavaC++
CA$136k–187k
Onsite · Toronto, Ontario, Canada
Senior Software Engineer, PAM
CA$136k–187k
KubernetesAmazon EKSDynamoDBAmazon KinesisRedisDockerTerraformAWSGoJavaC++
Staff Product Security Engineer, Reviews
OWASPJavaGoPythonOIDCiOSAndroid
CA$141k–193k
Hybrid · Toronto, Ontario, Canada
Staff Product Security Engineer, Reviews
CA$141k–193k
OWASPJavaGoPythonOIDCiOSAndroid
Similar roles
Senior Staff Machine Learning Engineer, GenAI Platform
$293k–410k
Senior Staff Machine Learning Engineer, GenAI Platform
$293k–410k
Staff Applied AI Engineer
$199k–261k
Staff Applied AI Engineer
$199k–261k
Principal AI Architect
$254k–356k
Principal AI Architect
$254k–356k
Software Engineer, GenAI Platform
$153k–197k
Software Engineer, GenAI Platform
$153k–197k
Senior Staff Machine Learning Engineer
$201k–352k
Senior Staff Machine Learning Engineer
$201k–352k