Principal Machine Learning Engineer, SecureAI

Okta
San Francisco
Workplace: HybridFull timeUSD 238,000 - 326,000 annuallyFunction: Data Science & Machine LearningExperience: 10+ yearsEducation: bachelorsSkills: ["Ownership","Collaboration","Mentoring","Iteration","Executing with excellence"]

Build authorization capabilities for AI agents within Okta’s Identity Security Fabric. Replace static, rule-based systems with intent-based enforcement that inspects agent runtime requests, evaluates intent and behavior, and delivers real-time semantic verification. You’ll architect scalable ML/GenAI systems (RAG, embeddings, evaluation pipelines, and guardrails), integrate inference into API gateway paths, and help mentor engineers while advancing secure, autonomous agent access at scale.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Okta
Okta
4 days ago

Principal Machine Learning Engineer, SecureAI

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 2 hours agoStatus: Live

Job Summary

Build authorization capabilities for AI agents within Okta’s Identity Security Fabric. Replace static, rule-based systems with intent-based enforcement that inspects agent runtime requests, evaluates intent and behavior, and delivers real-time semantic verification. You’ll architect scalable ML/GenAI systems (RAG, embeddings, evaluation pipelines, and guardrails), integrate inference into API gateway paths, and help mentor engineers while advancing secure, autonomous agent access at scale.
Location: San Francisco
Workplace: Hybrid
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key Responsibilities

  • •Architect and drive features for a unified control plane enabling secure, intent-based authorization for AI agents.
  • •Implement real-time LLM reasoning and prompt parsing to interpret prompts, tool payloads, and intent.
  • •Integrate low-latency inference or semantic evaluation into API gateway request paths.
  • •Design confidence-scored decision engines and evaluation benchmarks with defenses against prompt injection and guardrails to prevent bypasses or false positives.
  • •Build scalable ML/GenAI systems and automated evaluation pipelines, optimizing RAG workflows and enforcing reliable, compliant structured outputs, and mentor engineers.

Pay and Benefits

Salary: USD 238,000 - 326,000 annually
Equity and Bonus:Equity
Perks:Health InsuranceDentalVision401kFlexible SpendingPaid LeaveParental Leave

Key Requirements

  • •10+ years of software development experience with strong Python expertise (Go or TypeScript a plus).
  • •Hands-on applied machine learning experience across feature engineering, training, and fine-tuning.
  • •Hands-on experience with modern Generative AI platforms such as AWS Bedrock, OpenAI, and Anthropic.
  • •Deep understanding of RAG, embeddings, and knowledge-base workflows.
  • •Experience with AI agent frameworks (e.g., LiteLLM, LangGraph, LangChain, LlamaIndex, MCP) plus ML frameworks and orchestration tools (e.g., FastAPI, PyTorch, TensorFlow, Airflow).
Experience:10+ yearsGenerative AIRAGRetrieval-augmented generationAI agentsEnterprise securityDeveloper toolsML evaluation
Education:Bachelor's in Computer Science or related field
Skills:OwnershipCollaborationMentoringIterationExecuting with excellence
Tech Stack:PythonGoTypeScriptAWS BedrockOpenAIAnthropicLLMClaudeCedarLLM-as-a-judgeEmbeddingsVector searchZero-shot classificationRAGFastAPIPyTorchTensorFlowSpark MLAirflowLiteLLM

Company Brief

Okta
Provides identity and access management cloud solutions that help organizations secure and manage user authentication, single sign-on, multi-factor authentication, and lifecycle management across applications and devices.
Industry: Cybersecurity
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: San Francisco, United States
Founded: 2009
WebsiteLinkedIn