Staff Software Engineer, Applied AI

Okta Inc

Confirmed live today High trust

Quick summary

Work type
On-site
Location
San Francisco, CA
Salary
$194,000–$267,000 / yr
Posted
12 days ago
Freshness
Confirmed live today

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $217k
This role $230k
$157k $287k
below market most similar roles pay here above market

This role pays more than 60% of similar roles. Most pay $187,436–$246,000 — the blue band above. At the midpoint, this role pays about $230k versus about $217k 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, Applied AI

The Staff Software Engineer, Applied AI joins the team developing Digital Workers, a suite of autonomous agents for identity operations and security. In this role, you will build harnesses and evaluations to ensure these agents reliably investigate risk, use tools, and complete tasks across enterprise systems. You will integrate OpenAI and Anthropic model APIs, manage model routing, and develop evaluation datasets to measure false positives and unsafe actions. Key responsibilities include building for planning, memory, and recovery while maintaining authorization policy checks. You will utilize a technical stack including Python, FastAPI, AWS Step Functions, Amazon Bedrock AgentCore, PostgreSQL, and pgvector. The work focuses on solving complex identity and security problems by creating trusted, neutral infrastructure for autonomous agents to perform complex operations safely.

What does a Software Engineer earn in California?

Median $214500 from 1193 postings across 76 companies.

See salary data

What you'll do

  • Build agent harnesses for planning, tool use, memory, context compaction, and recovery during long-running tasks.
  • Integrate OpenAI and Anthropic model APIs while selecting models based on quality, latency, and cost.
  • Evaluate existing harnesses and build missing components connected to AWS workflow execution and approval states.
  • Enforce application policy checks for authorization and sensitive writes, requiring evidence of state before completion.
  • Build evaluation datasets and replay tests for valid findings, access decisions, and AI agent onboarding.
  • Measure false positives, missed risks, and unsafe actions to turn production failures into regression tests.
  • Test for prompt injection, stale evidence, conflicting instructions, and unsafe tool use to protect tenant boundaries.
  • Improve prompts, retrieval, tools, and model routing, including evaluating fine-tuning or post-training methods.

What we're looking for

  • Strong Python engineering skills and experience shipping an LLM application that uses tools.
  • Hands-on experience building agent workflows or evaluation systems for tool-using LLM applications.
  • Experience integrating model APIs, structured outputs, and asynchronous services.
  • Experience with OpenAI or Anthropic is especially relevant.
  • Depth in agent execution or evaluation systems with working code and failure analysis.
  • Ability to distinguish a plausible model response from a supported finding or verified action.
  • Open-source contributions, agent observability, security evaluations, or synthetic evaluation data (preferred).
  • Fine-tuning, preference data, post-training evaluation, or model distillation using approved datasets (preferred).

More like this

Similar roles

Senior Software Engineer, Applied AI

Okta Inc

San Francisco, CA 12 days ago $165,000–$226,000
Python FastAPI AWS AWS Step Functions Amazon Bedrock PostgreSQL pgvector OpenAI Anthropic LangChain LangGraph ClickHouse BigQuery LLM Fine-tuning

Principal Software Engineer, Applied AI

Okta Inc

San Francisco, CA 12 days ago $224,000–$308,000
Python FastAPI AWS AWS Step Functions Amazon Bedrock PostgreSQL pgvector OpenAI Anthropic LangChain LangGraph ClickHouse BigQuery LLM Fine-tuning Model Distillation

Staff Software Engineer, Applied AI

Apple Inc

Cupertino, CA 40 days ago $194,600–$293,900
LLM Python Go Java RAG PostgreSQL Snowflake SQL ETL HTTP APIs Model Monitoring Prompt Optimization ML Workflows Data Warehouse
10+ yrs exp

Staff Software Engineer, AI Platform & Infrastructure

Okta Inc

San Francisco, CA 4 days ago $194,000–$267,300
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

Senior Software Engineer, AI Platform & Infrastructure

Okta Inc

San Francisco, CA 4 days ago $165,000–$226,600
Python FastAPI AWS Lambda SQS EventBridge Step Functions PostgreSQL pgvector ClickHouse BigQuery S3 KMS Secrets Manager CloudWatch CloudTrail React AWS Amplify OpenAI Anthropic Infrastructure as Code OAuth OIDC SAML SCIM LangGraph

Software Architect, AI Systems and Platform

Okta Inc

San Francisco, CA 4 days ago $274,000–$376,200
Python FastAPI TypeScript React AWS Amazon Bedrock Lambda SQS EventBridge API Gateway PostgreSQL pgvector S3 ClickHouse BigQuery LangChain LangGraph Anthropic API OAuth OIDC SAML SCIM API Design