Staff Software Engineer, AI & Automation

Reposted 24 Days Ago
Be an Early Applicant
Bengaluru, Karnataka
In-Office
Senior level
Cloud
The Role
Design and develop full-stack applications integrating AWS Bedrock AI solutions. Build scalable AI/ML systems, ensuring smooth integration of LLMs into applications, while collaborating with stakeholders to meet business needs.
Summary Generated by Built In

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/.

Top Skills

Angular
Aws Bedrock
Boto3
C/C++
Go
Hugging Face Transformers
Numpy
Pandas
Python
PyTorch
Rust
TensorFlow
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: San Francisco, CA
6,000 Employees
Year Founded: 2009

What We Do

Okta is the leading independent identity provider. The Okta Identity Cloud enables organizations to securely connect the right people to the right technologies at the right time. With more than 7,000 pre-built integrations to applications and infrastructure providers, Okta provides simple and secure access to people and organizations everywhere, giving them the confidence to reach their full potential. More than 10,000 organizations, including JetBlue, Nordstrom, Siemens, Slack, T-Mobile, Takeda, Teach for America, and Twilio, trust Okta to help protect the identities of their workforces and customers.

Similar Jobs

In-Office
Bengaluru, Karnataka, IND
6000 Employees

Altium Logo Altium

Quality Assurance Engineer

Cloud • Enterprise Web • Software • Analytics • Design
Hybrid
Bengaluru, Bengaluru Urban, Karnataka, IND
900 Employees
Hybrid
Bengaluru, Bengaluru Urban, Karnataka, IND
289097 Employees
Hybrid
Bengaluru, Bengaluru Urban, Karnataka, IND
289097 Employees

Similar Companies Hiring

Toro TMS Thumbnail
Transportation • Software • Sales • Enterprise Web • Cloud
Chicago, IL
64 Employees
Yooz Thumbnail
Software • Machine Learning • Fintech • Financial Services • Cloud • Automation • Artificial Intelligence
Aimargues, FR
470 Employees
Amplify Platform Thumbnail
Fintech • Financial Services • Consulting • Cloud • Business Intelligence • Big Data Analytics
Scottsdale, AZ
62 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account