Principal Software Engineer, Applied AI at Okta | CA, US | Rezi

Principal Software Engineer, Applied AI at Okta

Principal Software Engineer, Applied AI

Okta · CA, US

3 weeks ago

Principal Software Engineer, Applied AI

Okta · CA, US

23 days ago
Resume preview

Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now
Resume preview

Tailor your resume to this Principal Software Engineer, Applied AI role.

Rezi rewrites your resume against Okta's job description. Free.

Resume score gauge reading 58 out of 100

Don't guess if your resume is good enough.

See how it scores against the Principal Software Engineer, Applied AI posting at Okta — free, in seconds.

About the Role

Okta is building Digital Workers, a new suite of autonomous agents for identity operations and security. This role offers the opportunity to shape a product that will serve Okta's 20,000+ customers, combining startup ownership with public company resources.

Responsibilities

  • Build the agent harness for planning, tool use, memory, context compaction, interruption, and recovery during long-running work.
  • Integrate OpenAI and Anthropic model APIs, selecting models based on measured quality, latency, and cost, with clear failure handling.
  • Evaluate existing harnesses and build only the missing pieces, ensuring the harness remains connected to AWS workflow execution and approval state.
  • Implement authorization and sensitive writes behind application policy checks, requiring evidence of resulting state before marking work complete.
  • Build evaluation datasets and replay tests for valid findings, access decisions, app setup, and AI agent onboarding.
  • Measure false positives, missed risks, unsafe actions, completion rate, and cost, converting production failures into regression tests.
  • Test for prompt injection, stale evidence, conflicting instructions, and unsafe tool use, protecting tenant boundaries in retrieval and memory.
  • Improve prompts, retrieval, tools, and model routing, evaluating fine-tuning or other post-training methods when justified and approved.
  • Set product engineering direction across workflows and implement shared components to accelerate team shipping.

Requirements

  • Production builder with experience shipping AI software and understanding of real-user impact.
  • Careful evaluator who investigates failures and uses evidence to judge quality.
  • Strong owner who follows improvements through code, evaluation, release, and customer use.
  • 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.
  • 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.

Skills

  • Python
  • FastAPI
  • AWS Step Functions
  • Amazon Bedrock AgentCore
  • PostgreSQL
  • Amazon RDS
  • pgvector
  • OpenAI API
  • Anthropic API
  • Claude Agent SDK
  • OpenAI Codex SDK
  • Strands Agents
  • LangChain Deep Agents
  • LangGraph
  • ClickHouse
  • BigQuery
  • Agent observability
  • Security evaluations
  • Synthetic evaluation data
  • Fine-tuning
  • Preference data
  • Post-training evaluation
  • Model distillation
  • Identity and security workflows
  • Foundation-model research

Location

  • San Francisco Bay Area

Work Type

  • In-person
  • Full-time

Experience Level

  • Principal

Salary/Compensations

  • $224,000—$308,000 USD

Benefits

  • Equity
  • Bonus
  • Health insurance
  • Dental insurance
  • Vision insurance
  • 401(k)
  • Flexible spending account
  • Paid leave (including PTO and parental leave)

About the Company

  • Okta is building the trusted, neutral infrastructure that enables organizations to safely embrace the new era of AI.
  • Okta has a global community spanning over 20 offices worldwide.
  • Okta is intentional about connection and fostering community.

Equal Opportunity

  • 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.
  • Okta considers for employment qualified applicants with arrest and convictions records, consistent with applicable laws.
  • Reasonable accommodation is available for the job application, interview process, or onboarding.
  • Okta may use Automated Employment Decision Tools (AEDT) that use artificial intelligence, machine learning, or other automated processes to assist in recruitment and hiring.