jobapplier

Principal Machine Learning Engineer (SecureAI)

Okta

Skills

Required

  • ML
  • AI/ML
  • AI-native
  • LiteLLM
  • LangGraph
  • LangChain
  • FastAPI
  • PyTorch
  • TensorFlow
  • Spark
  • Airflow
  • Python
  • Go / Typescript

Preferred

  • evaluation datasets / synthetic data generation / LLM-as-a-judge
  • frameworks
  • Implementing intent-based enforcement
  • semantic verification results
  • retrieval-augmented generation
  • RAG
  • embeddings
  • workflows

Join Okta, a leading identity security company, as a Principal Machine Learning Engineer. In this role, you will advance Okta's authorization capabilities by replacing static, rule-based systems with dynamic AI security mechanisms. You will architect and drive features for our unified control plane, establish the premier Identity Security Fabric for the agentic era, and mentor and coach engineers to support team and community growth. This is an opportunity to do career-defining work in a mission-driven company.

Exposure to ethical AI, model risk, or compliance frameworks

About Okta

Proven ability to collaborate with product and engineering teams to drive greenfield initiatives forward, navigate unknowns, and iterate quickly and frequently

Responsibilities

  • Architecting and driving features for a unified control plane to establish the premier Identity Security Fabric for the agentic era.
  • Implementing intent-based enforcement to verify agent runtime requests match their intended purpose, requiring key capabilities.
  • Designing confidence-scored decision engines that feed semantic verification results into policy frameworks (e.g., Cedar).
  • Deep understanding of retrieval-augmented generation (RAG), embeddings, and knowledge-base workflows

Requirements

  • Experience building tools or infrastructure for AI/ML applications, with a deep understanding of the developer lifecycle in an AI-native world
  • Hands-on experience with LiteLLM, LangGraph, LangChain, LlamaIndex, MCP, or other related AI agent frameworks
  • Familiarity with ML frameworks (FastAPI, PyTorch, TensorFlow, Spark ML) and workflow orchestration tools (Airflow, etc.)
  • 10+ years of software development experience, with strong programming expertise in Python (and familiarity with Go or Typescript a plus)
  • Hands-on experience with applied machine learning, from feature engineering to training and fine-tuning models
  • Experience defining evaluation metrics, pipelines, and feedback loops for ML/GenAI systems
  • Hands-on experience with modern Generative AI platforms (AWS Bedrock, OpenAI, Anthropic, etc.)
  • Experience integrating AI-driven systems with identity, authentication, or security products

Preferred

  • Familiarity with evaluation datasets, synthetic data generation, or LLM-as-a-judge methods

Pay

USD 238,000 - 326,000

Every job at Okta