Lead AI Engineer

We are seeking a highly skilled and motivated Senior/Lead AI Engineer to join our Mahindra AI team. The ideal candidate will have extensive experience in designing, developing, and deploying advanced AI/ML models and solutions, with a proven track record of leading small teams to successfully deliver complex AI projects. This role requires deep technical expertise, strong leadership skills, and a passion for pushing the boundaries of artificial intelligence to solve real-world problems.

Key Responsibilities

• Model Development: Design, develop, and optimize cutting-edge AI/ML models, including deep learning, reinforcement learning, NLP, and computer vision solutions, to address business and technical challenges.
• Project Leadership: Lead a small team of 2-5 engineers and data scientists on 2-3 AI projects simultaneously, ensuring timely delivery, high-quality outcomes, and alignment with organizational goals.
• End-to-End Implementation: Oversee the full lifecycle of AI solutions, from data preprocessing and feature engineering to model training, evaluation, deployment, and monitoring in production environments.
• Collaboration: Work closely with cross-functional teams, including product managers, software engineers, and domain experts, to integrate AI solutions into scalable, production-ready systems.
• Innovation: Stay abreast of the latest advancements in AI/ML research and technologies, and propose innovative approaches to enhance existing systems and workflows.
• Mentorship: Guide and mentor junior engineers and team members, fostering a culture of technical excellence and continuous learning.
• Performance Optimization: Optimize AI models for performance, scalability, and efficiency, ensuring they meet latency, throughput, and resource constraints in production.
• Code Quality: Write clean, maintainable, and well-documented code, adhering to best practices and contributing to shared codebases.

Required Qualifications

• Experience:
  • 5+ years of professional experience in AI/ML engineering, with a focus on developing and deploying production-grade AI solutions.
  • Proven experience leading small teams (2-5 members) on 2-3 AI/ML projects, delivering successful outcomes from ideation to production.
  • Hands-on expertise in building and deploying machine learning models using frameworks such as TensorFlow, PyTorch, Scikit-learn, or similar.
  • Strong programming skills in Python, with experience in production-level coding and software engineering practices.
  • Experience with cloud platforms (e.g., AWS, Azure, GCP) and deploying AI models in distributed, scalable environments.
• Education:
  • Bachelor's or Master's degree in Computer Science, Artificial Intelligence, Data Science, Mathematics, or a related field.
  • Advanced degree (Master's or Ph.D.) from a reputed institute is a strong plus.
• Technical Skills:
  • Deep understanding of machine learning algorithms, neural networks, and statistical modelling.
  • Proficiency in working with large datasets and tools like Pandas, NumPy, or SQL.
  • Experience with model optimization techniques, such as quantization, pruning, and distributed training.
  • Knowledge of MLOps practices, including model versioning, CI/CD pipelines, and monitoring in production.
• Leadership Skills:
  • Demonstrated ability to lead and manage small teams, with strong project management and communication skills.
  • Experience in setting technical direction, prioritizing tasks, and resolving technical challenges under tight deadlines.

Preferred Qualifications

• MS/Mtech/Ph.D. in Artificial Intelligence, Machine Learning, or a related field from a reputed institute.
• Experience with advanced AI domains such as generative AI, large language models (LLMs), or reinforcement learning.
• Contributions to open-source AI/ML projects or publications in top-tier AI conferences/journals.
• Familiarity with containerization and orchestration tools like Docker and Kubernetes.
• Experience mentoring early-career engineers or leading cross-functional initiatives.

Why Join Us?

• Opportunity to work on cutting-edge AI projects with real-world impact.
• Collaborative and innovative work environment with a focus on technical excellence.
• Access to state-of-the-art tools, technologies, and resources to fuel your success.