Skip to content
OktaOkta

Principal Software Engineer, AI Engineering

Leads architecture and hands-on development of secure, production-grade AI applications, including LLM orchestration, RAG pipelines, agentic experiences, and resilient backend services. Requires 12+ years of software engineering experience, strong Python and frontend expertise, and technical leadership across global teams.

About the job

Responsibilities

  • Drive the architecture and design of AI solutions, leading cross-functional initiatives across Product, Design, and Data Science.
  • Design, build, and refine backend services powering AI solutions, including LLM orchestration, RAG pipelines, and generative AI features.
  • Build high-performance agentic experiences for web and mobile, optimized for streaming responses and low latency.
  • Champion observability and operational excellence for enterprise-grade reliability and performance.
  • Enable delivery of key AI projects through technical leadership and hands-on execution.
  • Mentor engineers and raise technical craftsmanship and solution quality across the full stack.
  • Collaborate with cross-geo peers on aligned designs, shared roadmaps, and scalable architectures.
  • Partner with Security, Data, and Infrastructure teams to ensure secure, reliable, enterprise-ready AI adoption.

Requirements

  • 12+ years of software engineering experience, emphasizing backend systems design and development.
  • 8+ years of Python preferred.
  • 6+ years with modern frontend frameworks such as React, Next.js, and TypeScript.
  • Knowledge of accessibility principles (a11y) and effective collaboration with Product and Design.
  • Experience building and deploying production-grade AI applications and services.
  • Experience with distributed systems, microservices, and resilient REST or GraphQL APIs on cloud platforms.
  • Ability to influence technical strategy and lead architecture decisions across global teams.
  • Excellent communication skills and experience working across time zones and cultures.

Nice to Have

  • Experience with agentic frameworks such as LangChain, LangGraph, and LangSmith.

Benefits

  • Immersive, in-person onboarding experience.
  • Opportunities to work on applied AI projects and enterprise-scale platform foundations.
  • Well-being support, social impact initiatives, talent development, and community connection.

Skills

Python, React, Next.js, TypeScript, Llm Orchestration, RAG, Generative AI, LangChain, LangGraph, Langsmith, Distributed Systems, Microservices, Rest, GraphQL, AWS

Databricks

Databricks

Bengaluru, India

Staff Software Engineer - Search Platform
No salary listedOn-site10+ YOEML Engineering

Leads the development of ML- and NLP-powered search relevance systems, including query understanding, ranking, retrieval, and evaluation. Requires 10+ years of search relevance experience and a bachelor’s degree, with advanced study preferred.

Payabli

Payabli

Remote

Staff Machine Learning Engineer
No salary listedRemote8+ YOEML Engineering

Sets the technical direction for production machine learning across a payments platform, building and scaling models for risk, authorization, disputes, and forecasting. Requires 8+ years of ML engineering experience, including production model ownership and strong technical leadership.

Databricks

Databricks

Bengaluru, India

Staff Software Engineer
No salary listedOn-site10+ YOEML Engineering

Leads the development of machine-learning search relevance systems, including query understanding, ranking, retrieval, and evaluation at scale. Requires 10+ years of search relevance experience and expertise in ML, NLP, or related discovery technologies.

Databricks

Databricks

Bengaluru, India

Staff Software Engineer - Machine Learning
No salary listedOn-site10+ YOEML Engineering

Leads the development of machine-learning search relevance systems, including query understanding, ranking, retrieval, and evaluation pipelines. The role requires 10+ years of search relevance experience and expertise in NLP, LLMs, or related discovery technologies.

GitLab

GitLab

Bengaluru, India

Senior Software Engineer, NLP
No salary listedRemote5+ YOEML Engineering

Build and ship autonomous, agentic software development lifecycle capabilities, including AI agents, orchestration, and safety guardrails. The role requires senior software engineering experience, proficiency in Ruby, Go, or Python, distributed systems knowledge, and experience with AI/ML applications.