Team Lead – AI Engineering

Arvillium; Pune District; On-site; INR 4,000,000 - 7,000,000; Full time

Job summary

Arvillium is seeking a Team Lead – AI Engineering to spearhead the design, development, and deployment of cutting-edge AI solutions for enterprise clients. This role balances hands-on coding with technical leadership, driving architecture, reviews, and delivery excellence.

You will mentor a small team, engage directly with clients, and guide projects from concept to production, leveraging Generative AI, LLMs, and AI frameworks. Strong Python and cloud experience are essential.

Qualifications

• 5–7 years of hands-on software development with AI/ML projects.
• Strong academic or professional background in Data Science, AI, Computer Science, or a related field.
• Expert-level Python programming for production-grade applications.
• Experience leading or mentoring small engineering teams while remaining an active developer.

Responsibilities

• Lead the design, development, and deployment of AI and Generative AI solutions for enterprise customers.
• Act as a technical lead while contributing hands-on to software development and solution architecture.
• Manage a small team of AI engineers by providing technical guidance, mentoring, and code reviews.
• Participate in client discovery sessions, technical discussions, solution presentations, and project planning.
• Own technical delivery, ensuring projects are completed on time with high quality.
• Conduct regular client calls, provide project updates, manage expectations, and address technical queries.
• Prepare and present weekly project status reports, delivery metrics, and risk assessments.
• Develop scalable AI applications using Python and modern AI frameworks.
• Work with LLMs, AI Agents, Retrieval-Augmented Generation, vector databases, and ML models.
• Collaborate with data engineers, solution architects, and business stakeholders to deliver end-to-end AI solutions.
• Drive engineering best practices including code quality, testing, documentation, and continuous improvement.

Skills

Hands-on development

Python

AI/ML

Team mentoring

Client interaction

Architectural design

Education

Bachelor's degree in Data Science/AI/CS

Tools

Git

CI/CD

Docker

Kubernetes

Job description

Arvillium is looking for a highly motivated Team Lead – AI Engineering to lead the design and delivery of cutting‑edge AI solutions for enterprise customers. This is a hands‑on technical leadership role that combines software development, data science, client engagement, and team mentoring. You will lead a small team while remaining an active individual contributor, working on Generative AI, Machine Learning, AI Agents, and enterprise AI applications.

About the Role

As a Team Lead – AI Engineering, you will be responsible for delivering high‑quality AI solutions from concept to production. You will work directly with clients to understand business requirements, design technical solutions, guide a team of AI engineers, and ensure successful project execution. This is not a people‑manager‑only role—you will be expected to actively contribute to architecture, coding, code reviews, and technical problem‑solving while mentoring your team and driving delivery excellence.

Responsibilities

• Lead the design, development, and deployment of AI and Generative AI solutions for enterprise customers.
• Act as a technical lead while contributing hands‑on to software development and solution architecture.
• Manage a small team of AI engineers by providing technical guidance, mentoring, and code reviews.
• Participate in client discovery sessions, technical discussions, solution presentations, and project planning.
• Own technical delivery, ensuring projects are completed on time with high quality.
• Conduct regular client calls, provide project updates, manage expectations, and address technical queries.
• Prepare and present weekly project status reports, delivery metrics, and risk assessments.
• Develop scalable AI applications using Python and modern AI frameworks.
• Work with LLMs, AI Agents, Retrieval‑Augmented Generation (RAG), vector databases, and machine learning models.
• Collaborate with data engineers, solution architects, and business stakeholders to deliver end‑to‑end AI solutions.
• Drive engineering best practices including code quality, testing, documentation, and continuous improvement.
• Stay updated with the latest advancements in AI, machine learning, and software engineering, and bring innovative ideas to customer engagements.
• Champion modern AI‑assisted development practices, including effective use of AI coding assistants and vibe coding techniques to improve engineering productivity.

Requirements

• 5–7 years of hands‑on software development experience, with significant exposure to AI, Machine Learning, or Data Science projects.
• Strong academic or professional background in Data Science, Artificial Intelligence, Computer Science, or a related field.
• Expert‑level Python programming skills with experience building production‑grade applications.
• Experience with Generative AI, LLMs, AI Agents, RAG architectures, prompt engineering, and modern AI frameworks.
• Proven experience leading or mentoring small engineering teams while remaining an active developer.
• Excellent problem‑solving and system design skills.
• Strong communication and presentation skills with the ability to interact confidently with customers.
• Experience managing client calls, technical discussions, project status reporting, and stakeholder communication.
• Ability to balance technical leadership with hands‑on development responsibilities.
• Familiarity with Git, CI/CD, cloud platforms, APIs, and modern software engineering practices.
• Mandatory: Practical experience using AI‑assisted software development tools and vibe coding workflows to accelerate development while maintaining code quality.
• Ability to work independently in a fast‑paced consulting environment and manage multiple priorities.

Added Advantage

• Onsite customer engagement or international project experience.
• Experience delivering AI solutions for enterprise clients in consulting or IT services environments.
• Knowledge of cloud platforms such as AWS, Azure, or Google Cloud.
• Experience with MLOps, Docker, Kubernetes, and deployment automation.
• Exposure to data engineering technologies, vector databases, and enterprise integrations.