Lead AI Engineer
Company: Keywords Studios
Role summary
Lead AI Engineer is a full-time role at Keywords Studios in Pune, Maharashtra, India (hybrid). Posted 2026-10-21. Experience level: 10+.
- Job title
- Lead AI Engineer
- Company
- Keywords Studios
- Location
- Pune, Maharashtra, India
- Work arrangement
- hybrid
- Employment type
- full-time
- Category
- AI & Machine Learning
- Experience level
- 10+
- Date posted
- 2026-10-21
You’ll be joining an innovative team at the forefront of AI-driven customer support, where you’ll build advanced agents and productivity tools that directly impact how users interact with technology. It’s an exciting opportunity to lead the development of sophisticated LLM systems and RAG pipelines in a collaborative environment that values cutting-edge research and practical implementation.
Job description
About the team:
You will be part of the AI team responsible for building next-generation AI capabilities at Helpshift.
The team works on:
AI Agents for automated issue resolution
User Intent Detection
AI-Powered Answers and Knowledge Base Enhancements
Agent CoPilot for agent productivity
LLM-based systems, proprietary ML models, and multimodal RAG pipelines
The team includes Backend Engineers, Full-stack Developers, ML Engineers, Data Scientists, and Frontend Developers—collaborating to deliver intelligent and scalable AI experiences.
About the role:
We are looking for an experienced engineer with deep technical expertise in AI and strong leadership skills. You will help design and deliver advanced AI products and AI Agents, lead a group of engineers, and drive end-to-end execution of LLM-powered initiatives.
This role combines hands-on AI/ML and LLM engineering with technical leadership—guiding backend, frontend, and full-stack developers to build reliable, scalable production AI systems.
What you'll do:
Lead and mentor a team of backend and frontend developers, fostering a high-performing, collaborative engineering culture.
Own the technical direction, architecture, and strategy for LLM, AI Agent, and ML initiatives.
Architect and build AI Agents, RAG pipelines (including multimodal RAG), LLM-based services, and evaluation frameworks.
Design, build, and deploy scalable AI/ML systems from concept to production.
Collaborate closely with PMs, ML Engineers, and Data Scientists to convert business requirements into AI-driven solutions.
Oversee full SDLC for your squad—ensuring strong engineering practices around coding, testing, quality, and reliability.
Conduct detailed technical reviews and provide guidance to maintain code quality.
Troubleshoot performance, scalability, and reliability issues in production AI systems.
Champion continuous improvement, automation, and innovation within the team.
Keep the team up-to-date with advances in LLMs, LLMOps, fine-tuning techniques, vector search, and generative AI.
What you'll bring:
8–10 years of experience in software development.
More than 5 years of hands-on experience in AI/ML.
More than 2 years in an engineering leadership role, preferably leading a cross-functional team.
Hands-on experience building:
AI Agents / Agentic workflows
LLM pipelines (inference, orchestration, guardrails)
RAG systems (vector search, knowledge bases, multimodal RAG)
LLM fine-tuning (SFT, LoRA, instruction tuning)
LLMOps workflows (monitoring, evaluation, optimization, versioning)
Strong understanding of ML algorithms, embeddings, transformers, vector semantics, and distributed systems.
Experience with Vector Databases such as Elasticsearch, Weaviate, Pinecone, or Milvus.
Proficiency with ML frameworks (PyTorch, TensorFlow, scikit-learn).
Experience with cloud platforms (AWS, GCP, Azure) and MLOps tooling.
Good to have:
Solid foundation in backend technologies such as Python, Java, Go, Node.js, and experience with Kafka.
Knowledge of the Gaming industry or Customer Support domain.
Familiarity with frontend frameworks (React, Angular, Vue.js).
Exposure to functional programming (Clojure preferred; our team uses Clojure).
Experience in startup/high-growth environments.
Open-source contributions in AI/ML/LLM projects.
Experience with containerization and orchestration (Docker, Kubernetes).
Bonus Points:
Experience building production-grade LLM-powered automation for customer support or gaming.
Publications, presentations, or contributions related to LLMs, RAG, or AI Agents.
Hybrid setup
Worker's insurance
Paid Time Offs
Other employee benefits to be discussed by our Talent Acquisition team in India.
Helpshift embraces diversity. We are proud to be an equal opportunity workplace and do not discriminate on the basis of sex, race, color, age, sexual orientation, gender identity, religion, national origin, citizenship, marital status, veteran status, or disability status
Privacy Notice
By providing your information in this application, you understand that we will collect and process your information in accordance with our Applicant Privacy Notice. For more information, please see our Applicant Privacy Notice at https://www.keywordsstudios.com/en/applicant-privacy-notice.