Senior Software Engineer, Applied AI

Okta Inc

Confirmed live today High trust

Quick summary

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

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $175k
This role $196k
$107k $239k
below market most similar roles pay here above market

This role pays more than 72% of similar roles. Most pay $149,500–$201,250 — the blue band above. At the midpoint, this role pays about $196k versus about $175k 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 · Senior Software Engineer, Applied AI

The Senior 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 agent reliability as they investigate risks, use tools, and complete tasks across enterprise systems. You will integrate OpenAI and Anthropic model APIs, manage AWS workflow execution, and build evaluation datasets to measure false positives and completion rates. Key responsibilities include improving prompts, retrieval, and model routing while ensuring authorization remains behind application policy checks. The technical stack includes Python, FastAPI, AWS Step Functions, Amazon Bedrock AgentCore, PostgreSQL on Amazon RDS, and pgvector. You will solve complex problems regarding agent execution, memory, and context compaction to provide trustworthy autonomous identity software.

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 production metrics like false positives and missed risks, converting production failures into regression tests.
  • Test for prompt injection, stale evidence, conflicting instructions, and unsafe tool use to protect tenant boundaries.
  • Improve prompts, retrieval, and model routing while evaluating fine-tuning methods when results justify the effort.

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

Staff Software Engineer, Applied AI

Okta Inc

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

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

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

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

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

Senior Applied AI Engineer

Nvidia

Santa Clara, CA 19 days ago $152,000–$241,500
Python LLM Retrieval-Augmented Generation Agentic Workflows Machine Learning AI Systems Git Production Monitoring Observability Model Evaluation Backend Services
5+ yrs exp