Data And AI Lead

Welldoc; Bengaluru; On-site; INR 3,000,000 - 6,000,000; Full time

Benefits offered by this job

Flexible work environment

Generous PTO

Pre-IPO stock options

Medical insurance

Job summary

Welldoc is seeking an experienced AI Engineer to lead the design and scaling of Generative AI solutions, focusing on Retrieval-Augmented Generation and frontier models. You will collaborate with product, data, and engineering teams to prototype, evaluate, and productionize AI-first systems.

The role requires deep hands-on expertise in RAG pipelines and experience with orchestration frameworks like CrewAI, LangGraph, Autogen, or OpenAI SDK, along with strong Python and cloud deployment skills.

Qualifications

• 10–14 years total experience with at least 2–3 years in Generative AI/LLM projects
• Strong understanding of Frontier LLMs and their domain applications
• Hands-on experience with IDEs (Cursor)
• Experience building Retrieval-Augmented Generation systems using vector databases (FAISS, Weaviate, or Pinecone)
• Working knowledge of CrewAI, LangGraph, Autogen, or similar agent orchestration frameworks
• Proficiency in Python with LangChain, Transformers, and OpenAI SDKs
• Familiar with API integrations, asynchronous tasks, and cloud-native deployment (Azure)
• Excellent problem-solving and analytical skills and strong communication skills

Responsibilities

• Design and implement scalable RAG pipelines using enterprise knowledge bases
• Evaluate and integrate Frontier LLMs and open-source APIs for tasks
• Build autonomous agent workflows with CrewAI, LangGraph, Autogen, or OpenAI SDK
• Develop prompts, chains, and multi-agent systems
• Stay updated on GenAI developments and advise internal adoption
• Collaborate with product, data and engineering to turn prototypes into production systems
• Ensure responsible AI practices including data privacy and model explainability

Skills

Python

LangChain

Transformers

OpenAI SDKs

FAISS

Weaviate

Pinecone

Cursor IDE

CrewAI

LangGraph

Autogen

API integrations

Azure

Education

Bachelor's degree in CS/EE/SE

MS or PhD desirable

Tools

Cursor IDE

FAISS

Weaviate

Pinecone

Job description

Job Description:

Job Purpose

We are seeking a highly motivated and skilled AI Engineer to join our team and drive innovation in the fast-evolving landscape of Generative AI. The ideal candidate will have deep hands-on expertise in RAG (Retrieval-Augmented Generation) pipelines and working familiarity with Frontier Models (e.g., GPT-4, Claude, LLaMA, etc.). Exposure to cutting-edge orchestration frameworks like CrewAI, LangGraph, Autogen, or OpenAI SDK is highly desirable.

You will work closely with product, data, and engineering teams to prototype, build, and scale AI-first solutions that deliver tangible business impact. Strong communication skills are key, as you'll be working with cross-functional teams and occasionally interfacing with clients.

Responsibilities

• Design and implement scalable RAG pipelines to enable contextual, high-quality responses using enterprise knowledge bases.
• Evaluate and integrate Frontier LLMs /Open source LLM and APIs for specific tasks, ensuring optimal performance and efficiency.
• Build autonomous agent workflows using tools like CrewAI, LangGraph, Autogen, or OpenAI SDK.
• Develop and optimize prompts, chains, and multi-agent systems.
• Stay updated on the latest in the GenAI ecosystem and make recommendations for internal adoption.
• Collaborate with stakeholders to understand use cases, define solution architectures, and iterate prototypes into production-ready systems.
• Ensure responsible AI practices, including data privacy, model explainability, and compliance.

Required Skills & Experience

• 10 to 14 years of overall industry experience with at least 2-3 years in Generative AI /LLM-focused projects.
• Strong understanding of Frontier LLMs and their application across domains.
• Hands-on experience in IDE's such as Cursor
• Hands-on experience building Retrieval-Augmented Generation systems using vector databases (like FAISS, Weaviate, or Pinecone).
• Working knowledge of CrewAI, LangGraph, Autogen, or similar agent orchestration frameworks.
• Proficiency in Python, with experience using libraries like LangChain, Transformers, and OpenAI SDKs.
• Familiarity with API integrations, asynchronous task handling, and cloud-native deployment (Azure).
• Excellent problem-solving and analytical skills.
• Strong written and verbal communication skills.

Preferred Qualifications

• Exposure to US Healthcare systems, data formats (FHIR, HL7), or domain-specific use cases is a plus.
• Experience in fine-tuning or evaluating LLMs using domain-specific corpora.
• Understanding of security and compliance requirements in regulated industries.

What We Offer

• Opportunity to work on cutting-edge AI projects with meaningful real-world applications.
• Collaborative and inclusive work culture.
• Flexible work environment and competitive compensation.

Required Education

Bachelor's degree in Computer Science, Electrical and Computer Engineering, Software Engineering, or Equivalent. An MS or PhD in such technical fields is very desirable.

Compensation & Benefits

Welldoc offers a competitive compensation package, including generous PTO, pre-IPO stock options, and medical insurance