Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Staff Software Engineer, AI Platform & Infrastructure

$194k - $267.3k

Okta

Secure Every Identity, from AI to Human

Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence.

This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk.

The Software Engineer Opportunity

The team that built Jeeva AI is building Digital Workers, a new suite of autonomous agents for identity operations and security inside Okta. The team works directly with Todd McKinnon, Okta's co-founder and CEO, with the opportunity to bring this product to Okta's 20,000+ customers. You will help shape the product with the ownership and pace of a startup and the resources of a public company.

You will build the backend and AWS platform that keep these workers running around the clock. The team owns services, connectors, workflow execution, data and production reliability. You will write Python services and infrastructure code that let workers make approved changes and preserve evidence.

About You

Reliable builder: You design for failure and prove that recovery works.

Hands-on engineer: You write production services and improve the tools engineers use every day.

Strong owner: You follow changes through deployment, monitoring and production support.

What you will be doing

  • Build Python FastAPI services, data models and reusable connectors for Okta, AWS, ServiceNow and enterprise applications.
  • Implement durable execution with Step Functions, SQS and EventBridge. Handle approval waits, rate limits, duplicate events and safe retries.
  • Manage PostgreSQL schemas, indexes, query performance and pgvector retrieval. Enforce tenant isolation across APIs, storage and credentials.
  • Build model API access for OpenAI, Anthropic and supported AWS paths, including credentials, quotas, timeouts and cost controls.
  • Add an analytics pipeline for worker events, traces, tokens, findings and verified outcomes. Evaluate ClickHouse and comparable systems such as BigQuery.
  • Create reliable usage records and reconciliation. Prevent retries, duplicate findings and incomplete work from being counted as new outcomes.
  • Build deployment pipelines, infrastructure as code, observability and safe release controls across development, staging and production.
  • Implement scoped write services, kill switches, audit retention and backup recovery. Test failures and verify that sensitive changes actually took effect.

What you will bring to the role

  • 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. Experience diagnosing database performance and asynchronous system failures.
  • Experience with infrastructure as code, deployment automation, observability and production incident response.
  • Sound judgment about customer data, tenant isolation, credentials and safe changes to external systems.

Scope by level

Senior: Own a backend or platform component, including deployment, tests and production operation.

Staff: Lead a domain such as execution, tenant security, data infrastructure or reliability across teams.

Principal: Set shared backend and platform contracts, resolve system tradeoffs and implement foundational services.

Our stack

Python FastAPI, AWS Step Functions, Amazon Bedrock AgentCore, Lambda, SQS, EventBridge, API Gateway and containers. Data and security use RDS PostgreSQL, pgvector, S3 Object Lock, KMS, Secrets Manager, CloudWatch and CloudTrail. The frontend uses React and AWS Amplify.

OpenAI and Anthropic model integrations are planned. We are also adding an agent harness and dedicated analytics. ClickHouse is a hiring focus for analytics, and BigQuery or equivalent experience is relevant. The harness and analytics choices remain open.

And extra credit for

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

P26060_3596531

#LI-RWI

Below is the annual base salary range for candidates located in San Francisco Bay Area. Your actual base salary will depend on factors such as your skills, qualifications, experience, and work location. In addition, Okta offers equity (where applicable), bonus, and benefits, including health, dental and vision insurance, 401(k), flexible spending account, and paid leave (including PTO and parental leave) in accordance with our applicable plans and policies. To learn more about our Total Rewards program please visit: .

The annual base salary range for this position for candidates located in the San Francisco Bay area is between: $194,000—$267,300 USD

The Okta Experience

We are intentional about connection. Our global community, spanning over 20 offices worldwide, is united by a drive to innovate. Your journey begins with an immersive, in-person onboarding experience designed to accelerate your impact and connect you to our mission and team from day one.

Okta is an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, ancestry, marital status, age, physical or mental disability, or status as a protected veteran. We also consider for employment qualified applicants with arrest and convictions records, consistent with applicable laws.

If reasonable accommodation is needed to complete any part of the job application, interview process, or onboarding please use this Form to request an accommodation.

Notice for New York City Applicants & Employees: Okta may use Automated Employment Decision Tools (AEDT), as defined by New York City Local Law 144, that use artificial intelligence, machine learning, or other automated processes to assist in our recruitment and hiring process. In accordance with NYC Local Law 144, if you are an applicant or employee residing in New York City, please click here to view our full NYC AEDT Notice.
Vacancy posted 4 days ago
Similar jobs that could be interesting for youBased on the Staff Software Engineer, AI Platform & Infrastructure in San Francisco, CA vacancy
  • $194k - $267.3k

     ...Every Identity, from AI to HumanIdentity is the...  ...the trusted, neutral infrastructure that enables...  ...too, let's talk.The Software Engineer OpportunityThe team that...  ...the backend and AWS platform that keep these workers...  ...production operation.Staff: Lead a domain such as... 
    Suggested
    Local area
    Worldwide
    Flexible hours

    Okta

    San Francisco, CA
    7 hours ago
  • $240k - $400k

     ...About Sentry Software runs the world and the...  ...team is building its AI-native future....  ...distributed systems, cloud infrastructure, and services that...  ...at scale. Our Platform and Infrastructure Engineering teams build and...  ...We're looking for Staff Software Engineers... 
    Suggested
    Hourly pay
    Immediate start

    Sentry

    San Francisco, CA
    3 days ago
  •  ...com addresses.About the roleRippling’s Infrastructure organization builds the technical...  ...daily events and system complexity. As a Staff Software Engineer on the Cloud Infrastructure team, you'...  ...-region disaster recovery and agentic AI workflows. You will not just maintain... 
    Suggested
    Work at office
    3 days per week

    Rippling

    San Francisco, CA
    4 days ago
  •  ...about ourTotal Rewards philosophy.AI is a fundamental part of how work...  ...interview process.About the Role:As a Staff Software Engineer on the Cloud Infrastructure team, you'll build automation and...  ...to someone with a DevOps, SRE, or platform engineering background who enjoys... 
    Suggested
    Full time
    Work at office
    Local area
    2 days per week
    3 days per week

    Gusto

    San Francisco, CA
    4 days ago
  • $300 per month

     ...only vertically integrated AI infrastructure company built from the ground...  ...Data Center Infrastructure Engineering (DCIE) team is fundamental...  ...skilled and motivated Software Engineer to join Crusoe’s Data...  ...automation tooling for GPU platforms including NVIDIA A100, H200... 
    Suggested
    Temporary work

    Crusoe

    San Francisco, CA
    5 days ago
  • $198k - $299k

     ...9Amplitude is the leading AI analytics platform, helping over 5,200 customers...  .... Our in-house OLAP engine, Nova, processes trillions...  ...That makes Nova the critical infrastructure in the loop, and as non-stop...  ....We're looking for a Staff Software Engineer who wants to go deep... 
    Immediate start
    Worldwide
    Home office

    Amplitude

    San Francisco, CA
    4 days ago
  • $245.4k - $429.45k

     ...around the world come to our platform to find creative ideas,...  ...It’s Possible.At Pinterest, AI isn't just a feature, it's...  ...recruiting process here.Sr. Staff Software Engineer, Product ML InfrastructureMillions...  ...AI.Pinterest’s Product ML Infrastructure (PMLI) team enables fast,... 
    Work at office
    Local area
    Remote work
    Relocation
    Relocation package

    Pinterest

    San Francisco, CA
    1 day ago
  • $199k - $270k

    Staff Software Engineer - AI Research InfrastructureP-1215At Databricks, we are obsessed with enabling...  ...the world’s best data and AI platform so our customers can focus on the high...  ...Staff Software Engineer, AI Research Infrastructure, you will be developing and running... 
    Local area
    Worldwide

    DataBricks

    San Francisco, CA
    4 days ago
  • # Staff Software Engineer, Cloud InfrastructureAccordSan Francisco, CAFull TimeAugust...  ...InfrastructureApply with AI Cover Letter## Job...  ...looking for our first Cloud Infrastructure Engineer to own and expand...  ...ll also own our engineering platform, improving how engineers build... 
    Work at office
    Immediate start
    3 days per week

    OfferGenie

    San Francisco, CA
    4 days ago
  •  ...About Genius AI Genius AI is building technology...  ..., finances, staff and inventory management...  ...have committed to a software factory model: a standardized...  ...day. As our Staff Infrastructure Engineer, you'll be one of the...  ...Do Scale our CI platform as our orchestration... 
    Work at office
    Local area
    3 days per week

    US Health Partners, LLC

    San Francisco, CA
    18 hours ago
  • $100k - $300k

     ...looking for? Hercules is looking for a senior or staff level infrastructure engineer to help us build the future of AI software development What will you do? You will...  ...Hercules CMS, Hercules Files & Media, etc Platform , our enterprise-grade platform that... 
    Work at office
    Remote work

    Hercules

    San Francisco, CA
    2 days ago
  • $230k - $290k

     ...Staff Software Engineer, Mobile (App Infrastructure) Remote US At Affirm , we exist for the moments that matter—...  ...shared mobile app infrastructure — the platform every other mobile feature team...  ...are proficient and up to date on AI-assisted development — Claude Code... 
    For contractors
    Work at office
    Remote work
    Flexible hours

    Affirm

    San Francisco, CA
    2 days ago
  • $194k - $267k

     ...Every Identity, from AI to HumanIdentity...  ...trusted, neutral infrastructure that enables organizations...  .... The Okta Platform is an independent...  ...experiences. Okta's Core Engineering team is...  ...'re looking for a staff level backend engineer...  ...authorizationDemonstrably followed best software engineering... 
    Permanent employment
    Local area
    Worldwide
    Flexible hours

    Okta

    San Francisco, CA
    5 days ago
  •  ...about our Total Rewards philosophy. AI is a fundamental part of how work...  ...process. About The Role As a Staff Software Engineer on the Cloud Infrastructure team, you'll build automation and...  ...to someone with a DevOps, SRE, or platform engineering background who enjoys... 
    Full time
    Work at office
    Local area
    Remote work
    2 days per week
    3 days per week

    Gusto

    San Francisco, CA
    1 day ago
  • $232k

     ...About the Role As a Staff Engineer in Core Infrastructure, you will be a key technical...  ...compute demands of Generative AI and Autonomous Vehicle...  ...The Core Infrastructure Platform team builds the next generation...  ...~8+ years of full-time Software Engineering work experience... 
    Full time
    Work experience placement
    Work at office
    Remote work

    Uber Technologies Inc.

    San Francisco, CA
    4 hours ago
  • $236k - $290k

     ...services operate. By combining frontier agentic AI, an enterprise-grade platform, and deep domain expertise, we’re reshaping...  ...just getting started. Role Overview As a Staff Software Engineer on the Core Infrastructure team at Harvey, you'll play a critical role in... 
    Relocation package

    Harvey AI

    San Francisco, CA
    4 hours ago
  • $231k - $340k

     ...services operate. By combining frontier agentic AI, an enterprise-grade platform, and deep domain expertise, we’re reshaping...  ...just getting started. Role Overview As a Staff Software Engineer on the Model Infrastructure team, you'll lead the design and development of... 

    Harvey

    San Francisco, CA
    4 hours ago
  • $176.5k

     ...team and role: Core Infrastructure builds and operates...  ...and governance platform for all content across...  ...existing team of Senior, Staff, and Principal engineers building the Content...  ...who loves owning software end-to-end: design,...  ...You've made agentic AI tools part of your daily... 
    Local area
    Flexible hours

    Scribd, Inc.

    San Francisco, CA
    4 hours ago
  • $188k - $275k

     ...CoreWeave, the AI Hyperscaler™ acquired...  ...powerful end-to-end platform to develop, deploy...  ...-leading cloud infrastructure with the best-in-class...  ...intersection of software, hardware, and AI,...  ...challenging engineering problems at scale,...  ...About The Role As a staff engineer on Marimo... 
    Permanent employment
    Temporary work
    Casual work
    Work at office
    Remote work
    Flexible hours

    Weights & Biases

    San Francisco, CA
    4 days ago
  •  ...Lightfield Engineer PositionLightfield is reimagining CRM as a world...  ...of how a company works—so AI can help teams anticipate what...  ...Greylock to attack the biggest software category in the world. We...  ...building, and scaling the core infrastructure and systems powering Lightfield... 

    LIGHTFIELD INC

    San Francisco, CA
    5 days ago
  • $405k

     ...Staff Software Engineer, Environments InfrastructureSan Francisco, CA | New York...  ..., and steerable AI systems. We want AI to be safe...  ...organization builds and maintains the infrastructure that improves Claude's...  ...errors structurallyOwn the platform layers that sit beneath every... 
    Remote work
    Visa sponsorship

    anthropic

    San Francisco, CA
    13 hours ago
  •  ...security-first enterprise AI company. We build...  ...team of researchers, engineers, designers, and more,...  ...next generation of AI platforms powering advanced NLP...  ...Members of Technical Staff to join the Model Serving...  ...experience running production infrastructure at a large... 
    Full time
    Work experience placement
    Work at office
    Local area
    Remote work
    Home office

    Cohere

    San Francisco, CA
    4 days ago
  • $320k

     ...Staff+ Software Engineer, Claude App InfrastructureSan Francisco, CA | New York...  ...interpretable, and steerable AI systems. We want AI to be...  ...systems.About the RoleClaude App Infrastructure owns the reliability,...  ...capabilities of claude.ai. We're the platform team for Anthropic's... 
    Work at office
    Visa sponsorship
    Flexible hours

    anthropic

    San Francisco, CA
    5 days ago
  • $320k

     ...Staff + Senior Software Engineer, InferenceSan Francisco, CA | New York City, NY | Seattle...  ..., and steerable AI systems. We want AI to be...  ...high-performance inference infrastructure they need to develop next-...  ...running in multiple cloud platforms.Inference systems are highly... 
    Work at office
    Worldwide
    Visa sponsorship
    Flexible hours

    anthropic

    San Francisco, CA
    4 days ago
  • $260k - $350k

     ...ship category-defining software that enables Ryder and...  ...software to the data platform that will power the next...  ...impact for the engine of the American economy...  ...ll love it here.Role: Staff Software Engineer - InfrastructurePod...  ..., optimize cloud infrastructure, and champion best... 
    Full time
    Work at office
    Immediate start

    Baton Trucking

    San Francisco, CA
    4 days ago
  •  ...About the Role The Infrastructure team builds and...  ...strong infrastructure engineers whose experience spans...  ...neatly within a single platform team. You do not need...  ...Connector Platform, or AI Acceleration role clearly...  ...years of professional software engineering experience... 

    Maven Ventures

    San Francisco, CA
    4 hours ago
  • $200k - $275k

     ...speed and accuracy. Our AI-enabled platform turns siloed and disconnected...  .... TeamAs an engineering team, we believe strongly...  ...RoleWe are looking for a Staff Data Infrastructure Engineer to join our growing...  ...Airflow or similar toolsStrong software engineering fundamentals... 
    Local area

    Peregrine Technologies

    San Francisco, CA
    3 days ago
  • $248k - $282k

     ...Asana’s core reactive data loading platform, powering our GraphQL-like query...  ...billions of daily queries. As a Staff Software Engineer on the Data Loading Infrastructure team, you will lead the next chapter...  ...~ Demonstrates curiosity about AI tools and emerging technologies,... 
    Work at office
    Local area
    Work from home
    Worldwide

    Asana, Inc.

    San Francisco, CA
    4 hours ago
  • $197.3k - $313.7k

    ## Staff Software Engineer, Data InfrastructureApply: Office Tech-Flexible: California...  ...****Salesforce is the #1 AI CRM, where humans with...  ...Engineer to join the Data Infrastructure team within the broader Data...  ...and operate the core data platforms that enable data engineers,... 
    Full time
    Work at office
    Flexible hours

    Slack Enterprise

    San Francisco, CA
    2 days ago
  • $320k

     ...Staff+ Software Engineer, Infrastructure (Distributed Systems)San Francisco, CA | New York City, NY | Seattle, WAAbout AnthropicAnthropic's mission is to create reliable, interpretable, and steerable AI systems. We want AI to be safe and beneficial for our users and for... 
    Work at office
    Visa sponsorship
    Flexible hours

    anthropic

    San Francisco, CA
    5 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Staff Software Engineer, AI Platform & Infrastructure. Be the first to apply!