Careers at Square DataLabs

Build technology that moves businesses forward.

We are looking for people who enjoy solving real engineering problems, learning continuously and taking ownership. Join a team working across AI, digital commerce, cloud and modern product engineering.

View Open Positions →
Current Opening

Technical Lead – AI / Generative AI

We are looking for a hands-on AI Technical Lead who can design, build and guide production-ready AI solutions using modern LLM and retrieval technologies.

Engineering · Artificial Intelligence

Technical Lead – AI / Generative AI

2–3+ Years Relevant AI ExperienceFull TimeKochi / Hybrid
Apply Now →

About the role

You will lead the technical design and implementation of AI-powered applications, with a strong focus on Large Language Models, Retrieval-Augmented Generation (RAG), intelligent agents and production AI services. This is a hands-on role for an engineer who can make architecture decisions, write high-quality code and guide other developers.

What you'll do

  • Design and develop LLM-powered applications, RAG pipelines and AI agent workflows.
  • Build reliable retrieval systems using embeddings, vector search, document processing and context-management strategies.
  • Evaluate and integrate foundation models and APIs from leading LLM providers, as well as appropriate open-source models.
  • Develop backend AI services and APIs using Python and frameworks such as FastAPI.
  • Improve prompt design, retrieval quality, model responses, latency, reliability and cost efficiency.
  • Design evaluation, observability and guardrail approaches for production AI systems.
  • Work with databases, vector stores, caching, queues and cloud infrastructure used by AI applications.
  • Lead technical discussions, review code and mentor engineers while remaining actively involved in development.
  • Collaborate with product, frontend and backend teams to convert business requirements into scalable AI solutions.

Core technology areas

PythonLLMsRAGAI AgentsPrompt EngineeringEmbeddingsVector DatabasesFastAPIREST APIsRedisSQL / NoSQLCloudGit

Please include your resume and, where available, links to GitHub, AI projects or other technical work relevant to the role.

Hiring Process

Clear steps. Focused conversations.

Our process is designed to understand your technical depth, problem-solving approach and how you work with a team.

01Apply
02Profile Review
03Technical Discussion
04Final Conversation
05Offer
Life at Square DataLabs

Our moments.

Technology is built by people. A look at some moments from our team.