Staff Machine Learning Engineer, Generative AI (Auth0)
Core
Architect, design, and deploy robust Machine Learning & Generative AI systems, ensuring seamless integration with diverse platform services and establishing scalable LLMOps pipelines in production.
Role type
Staff Machine Learning Engineer (Generative AI)
Builds
Production-ready AI/ML systems, LLM-powered features, reusable components, and agentic applications for Okta's security and developer experience platforms.
Domain
Identity & Access Management (IAM) + Generative AI
Deliverable
production ML models
Required skills
Python, applied machine learning, Generative AI platforms (AWS Bedrock, OpenAI, Anthropic), RAG, embeddings, knowledge-base workflows, AI agent frameworks (LiteLLM, LangGraph, LangChain, LlamaIndex, MCP), ML frameworks (FastAPI, PyTorch, TensorFlow, Spark ML), workflow orchestration (Airflow), evaluation metrics and pipelines
Preferred skills
Identity/authentication/security product integration, ethical AI and model risk frameworks, evaluation datasets, synthetic data generation, LLM-as-a-judge methods
Technologies
AWS Bedrock, OpenAI, Anthropic, LiteLLM, LangGraph, LangChain, LlamaIndex, MCP, FastAPI, PyTorch, TensorFlow, Spark ML, Airflow
Responsibilities
Architect and deploy scalable ML/GenAI systems with retrieval, inference, and evaluation pipelines; Develop and iterate on structured prompting, context retrieval, and RAG workflows; Build automated evaluation pipelines for model quality and safety; Implement schema validation, structured output enforcement, and guardrails; Mentor and coach engineers on team growth and community development.
Seniority
Staff, hands-on IC with strategic influence and mentorship
