Staff Software Engineer, AI & Automation

at Okta

Bengaluru, India

Get to know Okta

Okta is The World’s Identity Company. We free everyone to safely use any technology, anywhere, on any device or app. Our flexible and neutral products, Okta Platform and Auth0 Platform, provide secure access, authentication, and automation, placing identity at the core of business security and growth.

At Okta, we celebrate a variety of perspectives and experiences. We are not looking for someone who checks every single box - we’re looking for lifelong learners and people who can make us better with their unique experiences.

Join our team! We’re building a world where Identity belongs to you.

About the Team

Okta’s Business Technology organization is on a mission to simplify and secure internal operations through automation, intelligence, and scale. Our Business Technology Shared
Services team builds core capabilities that power smarter, faster experiences for employees and customers.
We work across engineering, data science, security, and internal business teams to deliver GenAI- and automation-powered services — including virtual agents, workflow orchestration,and intelligent recommendations. We’re looking for engineers who enjoy solving tough problems and building with purpose

We are seeking a Staff Engineer – Applied AI to join our AI & Automation Engineering team in India. In this role, you’ll design, build, and scale AI-powered solutions that transform how our employees work and how our business operates. You’ll focus on applied AI use cases that deliver measurable business outcomes, while also shaping the underlyingplatforms and frameworks that make AI development faster, safer, and more reliable across the company.

As a senior individual contributor, you’ll collaborate closely with peers in the US and India, partnering with product managers, designers, and platform engineers. You’ll act as both a technical expert driving standards, mentoring engineers, and setting the bar for what great AI engineering looks like.

What you will do :

Design and deliver Applied AI solutions: Build end-to-end systems that leverage LLMs, ML models, retrieval pipelines, and integrations with enterprise data.
Shape technical direction: Define best practices for reliability, security, and scalability in AI-powered applications.
Partner across teams: Collaborate with platform engineers, data teams, and product managers to accelerate AI adoption across functions such as GTM, support, operations, and finance.
Mentor and elevate others: Provide guidance and code reviews, and influence architecture beyond your immediate team.
Contribute to platform foundations: Improve reusable components (RAG frameworks, prompt management, vector search, orchestration layers) that raise the baseline for AI engineering.
Evaluate and integrate emerging tech: Experiment with new AI/ML techniques, libraries, and vendor offerings, translating innovation into practical enterprise value.

What we are looking for:

Experience: 7+ years in software engineering with at least 2+ years in AI/ML, Applied AI, or data-intensive systems.
Expertise in Applied AI: Hands-on experience with LLMs, embeddings, RAG pipelines, prompt engineering, and model integration into production systems.
Programming skills: Proficiency in Python (preferred), plus experience with TypeScript/JavaScript or other modern languages.
System design: Strong background in designing distributed systems, APIs, and event-driven architectures.
Enterprise mindset: Familiarity with compliance, security, and data governance requirements in enterprise AI deployments.
Leadership without authority: Ability to influence cross-functional teams, mentor engineers, and set long-term technical direction.
Collaboration skills: Comfortable working across geographies (India, US) with excellent communication skills.

Nice to have:

Contributions to AI frameworks (LangChain, LangGraph, Hugging Face, etc.).
Experience with enterprise integration platforms (e.g., Boomi, Workato, Zapier, RPA).
Prior experience in high-scale SaaS or enterprise technology companies.

Why join us:

This is a unique opportunity to shape the future of enterprise AI adoption at scale. You’ll work on applied AI projects that directly improve employee productivity and customer experience, while building foundational platforms that make AI safer and easier to use across the organization.

Okta is an Equal Opportunity Employer and values diversity at all levels of the organization.

#hybrid role

P23034_3163149

What you can look forward to as a Full-Time Okta employee!

Amazing Benefits
Making Social Impact
Developing Talent and Fostering Connection + Community at Okta

Okta cultivates a dynamic work environment, providing the best tools, technology and benefits to empower our employees to work productively in a setting that best and uniquely suits their needs. Each organization is unique in the degree of flexibility and mobility in which they work so that all employees are enabled to be their most creative and successful versions of themselves, regardless of where they live. Find your place at Okta today! https://www.okta.com/company/careers/.

Some roles may require travel to one of our office locations for in-person onboarding.

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.

Okta is committed to complying with applicable data privacy and security laws and regulations. For more information, please see our Personnel and Job Candidate Privacy Notice at https://www.okta.com/legal/personnel-policy/.