Team Lead – Artificial Intelligence (AI)

Responsibilities

We are seeking an experienced Team Lead – AI to lead the design, development, and deployment of cutting-edge AI solutions across enterprise products. The ideal candidate will have strong expertise in Deep Learning, Large Language Models (LLMs), Computer Vision, and NLP, along with proven leadership experience in building and managing high-performing AI teams.

This role involves transforming traditional software systems into AI-driven intelligent platforms, driving innovation across multimodal AI, automation, and large-scale data systems.

Key Responsibilities

• Technical Leadership Lead the end-to-end development of AI solutions: from data collection → model development → deployment. Architect scalable AI systems using LLMs, deep learning, and multimodal models. Design and implement production-grade ML pipelines and AI-driven product features. Guide the team in selecting appropriate models, architectures, and training strategies.
• LLM & Generative AI Systems Design and deploy LLM-based applications (RAG pipelines, chatbots, agents). Work with open-source and proprietary models (e.g., LLaMA, Mistral, Qwen, etc.).
  
  

Implement

Retrieval-Augmented Generation (RAG)

Prompt engineering & fine-tuning

Token streaming and real-time inference systems

Optimize models for latency, cost, and performance in production.

• Computer Vision & Multimodal AI Develop and deploy computer vision models for detection, recognition, and document understanding. Build multimodal pipelines combining text, image, and audio inputs. Work with OCR systems and document intelligence workflows.
• Data Engineering & Model Training Handle large-scale datasets (structured & unstructured).
  
  

Design Data Pipelines For

ETL processes

Feature engineering

Dataset curation and annotation

Train Models Using Best Practices

Distributed training

Mixed precision (FP16)

Efficient fine-tuning techniques

• System Architecture & Deployment
  
  

Build Scalable Backend Systems Using

FastAPI / Python frameworks

Docker-based deployments

GPU-based inference systems

Work With

Databases (SQL/NoSQL)

Vector databases (FAISS, etc.)

Message queues & async processing (Celery, Redis)

• Product Development & Innovation
  
  

Collaborate With Product And Business Teams To

Identify AI use cases

Translate requirements into AI solutions

Drive Innovation In

Automation platforms

Intelligent document processing

Predictive analytics systems

• Team Leadership & Mentorship Lead and mentor a team of AI engineers, data scientists, and interns. Conduct technical reviews and ensure code/model quality.
  
  

Train Junior Engineers In

AI/ML fundamentals

Python programming

Industry best practices

Manage task allocation, timelines, and delivery.

• Research & Continuous Learning Stay updated with state-of-the-art research in AI/ML. Evaluate and integrate new models, tools, and frameworks. Encourage research-driven development and experimentation.
  
  

Desired Candidate Profile

Education

PhD / Master’s in Computer Science, AI, Data Science, or related field (preferred)

Experience

8+ years in AI/ML development

2+ years in team leadership or technical lead role

Proven experience in deploying AI solutions to production

Technical Skills

Core AI/ML

Deep Learning (PyTorch / TensorFlow)

NLP & LLMs

Computer Vision

Classification, regression, and anomaly detection

LLM & GenAI

Transformers, Hugging Face ecosystem

RAG pipelines

Fine-tuning and prompt engineering

Programming

Python (mandatory)

Experience with C++ or other languages is a plus

Frameworks & Tools

PyTorch, Transformers, Scikit-learn

FastAPI / Flask

Docker, Git

Data & Infrastructure

SQL / PostgreSQL / MySQL

Redis, vector DBs (FAISS)

Cloud platforms (AWS/GCP/Azure) – optional but preferred

Preferred Skills

Experience in multimodal AI systems (text + image + audio)

Knowledge of custom model training and optimization

Exposure to edge AI / TinyML

Experience in custom enterprise AI products

Understanding of security, RBAC, and scalable system design

Key Projects You May Work On

AI-powered document processing systems

LLM-based enterprise chat assistants

Predictive analytics and anomaly detection platforms

Multimodal AI applications (text, audio, image)

Intelligent automation for business workflows

Soft Skills

Strong leadership and communication skills

Problem-solving mindset with research orientation

Ability to work in fast-paced, product-driven environments

Ownership and accountability for deliverables

What We Offer

Opportunity to work on cutting-edge AI technologies

High-impact role influencing product direction

Collaborative and innovation-driven environment

Access to real-world large-scale datasets and problems

Bonus (Nice to Have)

Research publications in AI/ML

Experience mentoring teams or conducting training programs

Contributions to open-source AI projects

Back