Staff Software Engineer, AI

Staff Software Engineer, AI at Okta

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 Global AI Engineering Team

At Okta, you’ll be part of a unified global AI engineering team that spans the US and India. You’ll work on cutting-edge applied AI projects while shaping the platform foundations that make secure, scalable AI possible at enterprise scale. If you’re passionate about building AI that is smarter, faster, and always secure, this is your opportunity to make an impact.

The Staff Software Engineer Opportunity

At Okta, we’re building the future of secure, enterprise-grade AI Agents. We’re looking for a Staff Engineer to join our global AI Engineering team. In this role, you will be designing and building the intelligent, user-facing experiences and the end-to-end AI solutions that power them.

This role will you’ll play a key role in designing and developing platforms and services that support Okta’s internal transformation. You’ll help build the infrastructure and applications that make intelligent automation and GenAI real for our workforce — all while ensuring performance, security, and reliability at enterprise scale.This is a hands-on IC role for someone who thrives on system design, loves to build, and wants to grow their influence across a fast-moving engineering team. This person will also set technical direction, mentor others, and partner closely with our India-based counterparts as part of one global team.

What you'll be doing

• Design, build, and refine the front-end interfaces that bring our AI capabilities to life for employees and customers.
• Build high-performance Agentic Experiences (AX) for web and mobile, engineered for streaming responses and low latency.
• Champion observability and operational excellence to ensure our AI services meet enterprise-grade standards for reliability and performance.
• Develop robust backend services to power our AI solutions, including LLM orchestration, RAG pipelines, and generative AI features
• Enable the successful delivery of key AI projects through technical leadership and hands-on execution.
• Mentor engineers, raising the bar on technical craftsmanship and solution quality across the full stack.
• Collaborate with India-based peers to ensure globally aligned designs, shared roadmaps, and scalable architectures.
• Partner with Security, Data, and Infrastructure teams to ensure AI adoption is secure, reliable, and enterprise-ready

What you'll bring to the role

• 7+ years in software engineering, with deep expertise in front-end development (5+ years in modern frameworks like React, Vue, or Angular).
• 4+ years with backend systems design and development (Python preferred, Java, Node.js, or Go).
• A strong commitment to building accessible (a11y) and inclusive user experiences while collaborating effectively with Product & Design.
• Proven experience integrating LLMs and MCPs into production-ready, user-facing applications.
• Experience with agentic frameworks like LangChain, LangGraph, and LangSmith is a strong plus.
• Strong background in building distributed systems, microservices, and resilient APIs (REST/GraphQL) on cloud platforms (AWS/GCP/Azure).
• Demonstrated ability to influence technical strategy and lead architecture decisions across global teams.
• Excellent communication skills with experience working effectively across time zones and cultures.