AI Engineer - NLP & Generative AI
At a glance
Mid-level Machine Learning Engineering role at Keywords Studios. Pune · full-time.
Pay not stated AI Engineer salary
aimlroles summary, based on the employer's posting.
What you'll do
- Build and maintain Python backend services and APIs, including systems that support AI Agents
- Connect RAG pipelines, embedding services and inference APIs to backend workflows for customer support automation
- Use MongoDB, PostgreSQL or another listed database for workloads that need low latency at scale
- Package services with Docker, run them on Kubernetes, and deploy across AWS and GCP
- Monitor service health, resolve production performance issues, and improve reliability using observability tools
What you bring
- You have three to five years of experience developing backend software
- You bring hands-on Python and backend framework experience, especially with FastAPI
- You have built backend APIs and workflow integrations, and understand databases such as MongoDB or PostgreSQL
- You have used Docker, understand Kubernetes, and know the basics of AWS, GCP and CI/CD
Who this fits
This mid-level role suits someone focused on backend engineering who wants to build and operate AI-supported workflows. The role is based onsite in Pune, India, with a hybrid setup listed among the benefits.
From the employer
About the Role
We are looking for a Backend AI Engineer to join our AI engineering team and help build and scale backend systems that power AI Agents for the gaming industry, handling millions of customer support tickets at scale.
In this role, you will work on high-throughput backend services, AI-powered workflows, and retrieval-based systems that support large-scale customer support automation. You will collaborate closely with AI Engineers, ML Engineers, and Product teams to design, build, deploy, and operate reliable backend systems that integrate AI components such as RAG pipelines and AI Agent orchestration.
This is an excellent opportunity to work on production-grade AI systems operating at massive scale, where performance, reliability, and observability are critical.
What You’ll Do:
- Design, develop, and maintain scalable backend services and APIs using Python and frameworks like FastAPI.
- Build and integrate backend workflows supporting AI Agents and customer support automation use cases.
- Develop secure and performant REST APIs for internal and external consumers.
- Integrate AI components such as RAG pipelines, embedding services, and inference APIs into backend systems.
- Work with databases including MongoDB, YugabyteDB, and PostgreSQL to support high-scale, low-latency workloads.
- Containerize applications using Docker and deploy/manage services on Kubernetes (K8s).
- Deploy and operate services on AWS and GCP cloud platforms.
- Implement and maintain CI/CD pipelines, and actively participate in CM/CR release processes.
- Use observability tools such as Grafana, Elasticsearch, and Kibana to monitor system health, debug issues, and improve reliability.
- Write clean, testable, and maintainable code; participate in code reviews and design discussions.
- Troubleshoot and resolve production issues related to scalability, performance, and reliability.
Requirements
What You’ll Bring
- 3–5 years of experience in backend software development.
- Strong hands-on experience with Python and backend frameworks, especially FastAPI.
- Experience building backend APIs and workflow integrations.
- Solid understanding of databases such as MongoDB, PostgreSQL, and YugabyteDB.
- Hands-on experience with Docker and working knowledge of Kubernetes.
- Familiarity with AWS and GCP cloud services.
- Good understanding of CI/CD concepts, build pipelines, and CM/CR release processes.
- Working knowledge of observability and logging systems like Grafana, Elasticsearch, and Kibana.
- Basic to intermediate understanding of AI systems, especially Retrieval-Augmented Generation (RAG).
- Strong problem-solving skills and ability to work effectively in a collaborative team environment.
Good to Have / Plus Skills
- Experience with Jenkins for CI/CD automation.
- Familiarity with MCP concepts and AI orchestration frameworks.
- Hands-on exposure to AI/LLM-based systems, embeddings, or vector search.
- Experience or familiarity with Clojure.
- Knowledge of the Gaming industry or Customer Support domain.
- Experience working in high-scale, high-traffic production systems.
Bonus Points
- Experience building or operating AI Agents in production environments.
- Exposure to distributed systems, event-driven architectures, or message queues.
- Interest in AI reliability, evaluation, and system observability at scale.
Benefits
- 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.