Staff Software Engineer, AI Platform & Infrastructure (AI)

Location: San Francisco, California; in-person onboarding is required.

Annual base salary: $194,000–$267,300 USD for candidates located in the San Francisco Bay Area. Equity, bonus, health, dental and vision insurance, 401(k), flexible spending account, and paid leave may also apply.

Company

Okta develops identity and security infrastructure, including autonomous agents for identity operations and enterprise security.

What you will do

• Build Python FastAPI services, data models, and reusable connectors for Okta, AWS, ServiceNow, and enterprise applications.
• Implement durable workflow execution with Step Functions, SQS, and EventBridge, including approval waits, rate limits, duplicate events, and safe retries.
• Manage PostgreSQL schemas, indexes, query performance, pgvector retrieval, and tenant isolation across APIs, storage, and credentials.
• Build model API integrations and controls for OpenAI, Anthropic, and supported AWS paths, including credentials, quotas, timeouts, and cost management.
• Develop analytics, usage metering, reconciliation, deployment pipelines, infrastructure as code, observability, and safe release controls.
• Implement scoped write services, kill switches, audit retention, backup recovery, and failure testing for sensitive changes.

Requirements

• 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 database performance diagnosis and troubleshooting asynchronous system failures.
• Experience with infrastructure as code, deployment automation, observability, and production incident response.
• Sound judgment regarding customer data, tenant isolation, credentials, and safe changes to external systems.
• For Staff scope: ability to lead a domain such as execution, tenant security, data infrastructure, or reliability across teams.

Nice to have

• ClickHouse, BigQuery, or comparable analytics systems, event ingestion, and usage metering experience.
• Experience with multi-tenant SaaS, OAuth, OIDC, SAML, SCIM, or enterprise connector development.
• Experience deploying agent frameworks such as Claude Agent SDK, Codex SDK, Strands Agents, Deep Agents, or LangGraph.
• Prior identity experience is optional.

Culture & Benefits

• Work with startup-like ownership and pace while building for Okta's public-company resources and customer base.
• Collaborate with a global community spanning more than 20 offices worldwide.
• Complete immersive, in-person onboarding designed to connect employees with the mission and team.
• Benefits include health, dental and vision insurance, 401(k), flexible spending account, PTO, parental leave, equity where applicable, and bonus eligibility.