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