Staff Software Engineer, AI Platform & Infrastructure

Okta Inc

Confirmed live today High trust

Quick summary

Work type: On-site
Location: San Francisco, CA
Salary: $194,000–$267,300 / yr
Posted: 4 days ago
Freshness: Confirmed live today
Nearby: 99+ roles within 25 mi

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $213k

This role $231k

$164k $278k

below market most similar roles pay here above market

This role pays more than 60% of similar roles. Most pay $179,582–$246,750 — the blue band above. At the midpoint, this role pays about $231k versus about $213k for comparable roles.

Based on 240 similar postings.

Employer

About Okta Inc

Okta, Inc. is an American identity and access management company based in San Francisco. It provides cloud software that helps companies manage and secure user authentication into applications, and for developers to build identity controls into applications, websites, web services, and devices.[

Okta Inc currently has 103 open roles on FindRole.

Listed pay typically runs $174,000–$238,000 across 103 roles with salary data.

Most-posted roles

View all roles at Okta Inc

At a glance

TL;DR · Staff Software Engineer, AI Platform & Infrastructure

Role Posting Log in to save

The Staff Software Engineer, AI Platform & Infrastructure joins the team developing Digital Workers, a suite of autonomous agents for identity operations and security. You will build the backend and AWS platform to keep these workers running, owning services, connectors, workflow execution, and production reliability. Day-to-day tasks include writing Python FastAPI services, managing PostgreSQL schemas with pgvector, and implementing durable execution using AWS Step Functions, SQS, and EventBridge. You will also build model API access for OpenAI and Anthropic, manage infrastructure as code, and develop analytics pipelines using ClickHouse or BigQuery. This role solves the technical challenge of enabling autonomous agents to perform approved changes while preserving evidence and enforcing tenant isolation across APIs, storage, and credentials within a secure identity infrastructure.

What does a Software Engineer earn in California?

Median $214500 from 1193 postings across 76 companies.

See salary data

Skills

Python FastAPI AWS PostgreSQL Step Functions SQS EventBridge pgvector ClickHouse BigQuery Lambda API Gateway S3 KMS Secrets Manager CloudWatch CloudTrail React AWS Amplify OpenAI Anthropic Infrastructure as Code OAuth OIDC SAML SCIM LangGraph

What you'll do

Build Python FastAPI services, data models, and reusable connectors for enterprise applications.
Implement durable execution using AWS Step Functions, SQS, and EventBridge to handle retries and rate limits.
Manage PostgreSQL schemas, query performance, and pgvector retrieval while enforcing tenant isolation.
Build model API access for OpenAI and Anthropic with integrated cost controls and quotas.
Develop an analytics pipeline for worker events and traces using ClickHouse or BigQuery.
Create reliable usage records and reconciliation systems to prevent duplicate findings or incomplete work.
Build deployment pipelines, infrastructure as code, and observability tools across all environments.
Implement scoped write services, kill switches, and audit retention to ensure safe system changes.

What we're looking for

Experience building and operating production backend or distributed systems with strong coding skills in Python or a comparable language.
Hands-on AWS experience with APIs, queues, containers, networking, IAM, and databases.
Strong SQL and data modeling skills, including experience diagnosing database performance and asynchronous system failures.
Experience with infrastructure as code, deployment automation, observability, and production incident response.
Ability to lead a domain such as execution, tenant security, data infrastructure, or reliability across teams.
Experience with ClickHouse, BigQuery, or comparable analytics systems, event ingestion, and usage metering (preferred).
Experience with multi-tenant SaaS, OAuth, OIDC, SAML, SCIM, or enterprise connector development (preferred).
Experience deploying agent frameworks, including Claude Agent SDK, Codex SDK, Strands Agents, Deep Agents, or LangGraph (preferred).

8+ yrs exp Remote

Save