Lead Software Engineer - AI | Mumbai City

Job Description:
We are looking for an experienced Lead Software Engineer – AI to head a cross-functional AI engineering team. This role involves leading the development of AI-powered solutions across computer vision, NLP and predictive modeling. You will architect scalable AI systems, implement MLOps practices, and guide a team of engineers. This is a full-time role based in Mumbai, Rajkot or Ahmedabad.

Key Responsibilities:

  • AI Product Development
    • Lead end-to-end development of AI-powered solutions in computer vision, NLP, and predictive modeling.
    • Develop use cases in image enhancement, document intelligence, entity extraction, and automated data validation.
    • Apply deep learning, transformers, and foundation model fine-tuning for scalable, real-world applications.
    • Design scalable AI services integrated via REST APIs and microservices.
    • Explore cutting-edge technologies like generative AI, multilingual AI, and biometric verification.
  • MLOps & AI Engineering
    • Build and maintain CI/CD pipelines for model training, deployment, and testing.
    • Automate monitoring, retraining workflows, and ensure traceability for models and datasets.
    • Utilize tools such as MLflow, Kubeflow, Airflow, Docker, and Kubernetes for production-grade deployment.
  • NLP & Language Model Applications
    • Lead development of domain-specific language models and PoCs.
    • Implement transformer-based architectures for tasks like summarization, semantic search, and QA systems.
    • Fine-tune large language models (LLMs) for unstructured data insights.
    • Integrate NLP modules into business workflows and real-time systems.
  • Engineering Leadership
    • Mentor ML engineers and software developers.
    • Collaborate with cross-functional stakeholders on product and engineering roadmaps.
    • Lead sprint planning, delivery oversight, and code/architecture reviews.

What We're Looking For:

  • Required Technical Skills
    • 5+ years in AI/ML development with experience in:
    • Computer Vision (e.g., OCR, Deepfake Detection)
    • NLP (e.g., Text Parsing, Transformers, GPT fine-tuning)
    • ML frameworks: PyTorch, TensorFlow, Hugging Face
    • MLOps tools: MLflow, Kubeflow, Airflow, Docker, Kubernetes
    • Strong programming skills in Python; Java/Go/Node.js are a plus
    • Proficiency in REST APIs, microservices architecture
    • Experience with DevOps tools: Jenkins, GitHub Actions, Prometheus, Grafana
  • Leadership Skills
    • Demonstrated success leading teams of 5–10 engineers
    • Strong communication and project management skills

Ability to translate product requirements into scalable AI systems

Preferred Qualifications

  • B.Tech/M.Tech in Computer Science, AI, or related field
  • Exposure to financial or regulated domains is a plus
  • Experience working with multilingual or international datasets
Apply now