Skip to content
OktaOktaSan Francisco, CA

Staff Software Engineer, AI-Core

Build and own core layers of a scalable Agentic AI platform for autonomous engineering agents at Okta, including agent identity/auth, knowledge/memory, observability, governance/safety, and orchestration. Requires 5+ years backend engineering experience plus AI/agent exposure; strong distributed systems and security knowledge.

194k – 267k
Hybrid7+ YOEML Engineering

About the role

What You'll Work On

  • Design and implement backend APIs and services that make up the agentic platform
  • Build the agent identity and machine-to-machine authentication system, including credential management and delegated access flows
  • Build the agent knowledge base and memory layer so agents retain context within and across sessions
  • Build the observability layer for agents: tracing, cost tracking, audit logs, and dashboards that make agent behavior debuggable in production
  • Build the governance and safety layer: policy enforcement on tool calls, content filtering, PII protection, and human-in-the-loop approval flows
  • Build the orchestration layer that coordinates multi-step agent workflows with state persistence and error recovery
  • Collaborate with Product, SRE, Security, Data Platform, Observability and other domain partners to shape what each layer looks like and how it integrates with the broader Okta platform

You Might Be a Good Fit If You Have

  • 5+ years of software engineering experience building production backend systems
  • 1+ years of exposure to AI/ML or agentic applications, whether through production work, side projects, or hands-on experimentation
  • Strong proficiency in one or more backend languages (Python, TypeScript/Node.js, Go, or similar)
  • Hands-on experience designing and operating distributed systems: APIs, microservices, container orchestration, and serverless technologies
  • A security-conscious mindset around credential handling, trust boundaries, and what can go wrong at integration points; familiarity with OAuth, OIDC, or other modern auth patterns
  • Comfort operating in ambiguous, fast-moving environments where the problem definition evolves alongside the solution and the right abstractions are still being invented

Bonus

  • Exposure to LLM integration, RAG pipelines, MCP, or agent orchestration frameworks like LangChain, LangGraph, or the Claude/OpenAI SDKs
  • Experience with policy-as-code authorization (Cedar, OPA), agent identity patterns, or building developer-facing APIs and SDKs

Skills

PythonTypeScriptNode.jsGoDistributed SystemsMicroservicesContainer OrchestrationServerlessOAuthOIDCLlm IntegrationRAGLangChainLangGraphCedar

Similar roles

ML Engineering jobs
Nuro

Senior/Staff Machine Learning Engineer, Sensor Simulation

NuroMountain View, CA

Develop synthetic sensor simulation models and algorithms using ML techniques like NeRF and Gaussian splatting to generate photorealistic images and realistic lidar/radar data for autonomous vehicles. Requires advanced degree plus 3-5+ years experience, strong ML fundamentals, and Python/deep learning expertise.

194k – 291k
On-site5+ YOEML Engineering
Nuro

Senior/Staff Software Engineer, ML Infrastructure, Optimization

NuroMountain View, CA

Build and optimize ML infrastructure for autonomous vehicles, focusing on model optimization, compilers, and deployment of large models to Nuro's self-driving fleet. Requires 3+ years ML optimization experience and strong Python/C++/CUDA skills.

194k – 352k
On-site3+ YOEML Engineering
Nuro

Senior/Staff Software Engineer, ML Data

NuroMountain View, CA

Senior/Staff Software Engineer architects ML data pipelines for autonomy AI at Nuro, transforming massive datasets into training signals. Requires 7+ years experience, C++/Python proficiency, and expertise in end-to-end ML data systems.

194k – 352k
On-site7+ YOEML Engineering
Square

Staff Machine Learning Engineer, Credit Products (Square Financial Services)

SquareSeattle, WA +1

Develops and deploys ML models for credit underwriting in regulated banking, owning full stack from data curation to production decisioning. Leads MLOps initiatives and improves policies using advanced techniques for underserved segments. Requires 8+ YOE in ML production and technical degree.

195k – 343k
Hybrid8+ YOEML Engineering
Databricks

Staff Software Engineer

DatabricksNew York, NY

Staff ML Engineer building CustomerLake, Databricks' Customer Data Platform for enterprise ML/AI personalization, recommendations, churn, and LTV modeling. Requires 10+ years shipping production ML/LLM systems with strong product mindset in 0-to-1 environments.

192k – 260k
On-site10+ YOEML Engineering