AI / ML Developer (Lead) with skills AI/ML Development, TensorFlow, NLP, Pytorch, Large Language Models (LLM) for location Any Infogain Base Location (Noida, Gurugram, Bangalore, Mumbai, Pune)
ROLES & RESPONSIBILITIES

Key Responsibilities

  • Design, develop, and optimize Machine Learning & Deep Learning models using Python and libraries such as TensorFlow, PyTorch, and Scikit-learn

  • Work with Large Language Models (e.g., GPT, BERT, T5) to solve NLP tasks such as, semantic search, summarization, chatbots, conversational agents, and document intelligence.

  • Lead the development of scalable AI solution including data preprocessing, embedding generation, vector search, and prompt orchestration.

  • Build and manage vector databases and metadata stores to support high-performance semantic retrieval and contextual memory.

  • Implement caching, queuing, and background processing systems to ensure performance and reliability at scale.

  • Conduct independent R&D to implement cutting-edge AI methodologies, evaluate open-source innovations, and prototype experimental solutions

  • Apply predictive analytics and statistical techniques to mine actionable insights from structured and unstructured data.

  • Build and maintain robust data pipelines and infrastructure for end-to-end ML model training, testing, and deployment

  • Collaborate with cross-functional teams to integrate AI solutions into business processes

  • Contribute to the MLOps lifecycle, including model versioning, CI/CD, performance monitoring, retraining strategies, and deployment automation

  • Stay updated with the latest developments in AI/ML by reading academic papers, and experimenting with novel tools or frameworks

Required Skills & Qualifications

 

  • Proficient in Python, with hands-on experience in key ML libraries: TensorFlow, PyTorch, Scikit-learn, and HuggingFace Transformers

  • Strong understanding of machine learning fundamentals, deep learning architectures (CNNs, RNNs, transformers), and statistical modeling

  • Practical experience working with and fine-tuning LLMs and foundation models

  • Deep understanding of vector search, embeddings, and semantic retrieval techniques.

  • Expertise in predictive modeling, including regression, classification, time series, clustering, and anomaly detection

  • Comfortable working with large-scale datasets using Pandas, NumPy, SciPy etc.

  • Experience with cloud platforms (AWS, GCP, or Azure) for training and deployment is a plus

Preferred Qualifications

 

  • Master’s or Ph.D. in Computer Science, Machine Learning, Data Science, or related technical discipline.

  • Experience with MLOps tools and workflows (e.g., Docker, Kubernetes, MLflow, SageMaker, Vertex AI).

  • Ability to build and expose APIs for models using FastAPI, Flask, or similar frameworks.

  • Familiarity with data visualization (Matplotlib, Seaborn) and dashboarding (Plotly) tools or equivalent

  • Working knowledge of version control, experiment tracking, and team collaboration 

EXPERIENCE
  • 8-11 Years
SKILLS
  • Primary Skill: AI/ML Development
  • Sub Skill(s): AI/ML Development
  • Additional Skill(s): TensorFlow, NLP, Pytorch, Large Language Models (LLM)
ABOUT THE COMPANY

Infogain is a human-centered digital platform and software engineering company based out of Silicon Valley. We engineer business outcomes for Fortune 500 companies and digital natives in the technology, healthcare, insurance, travel, telecom, and retail & CPG industries using technologies such as cloud, microservices, automation, IoT, and artificial intelligence. We accelerate experience-led transformation in the delivery of digital platforms. Infogain is also a Microsoft (NASDAQ: MSFT) Gold Partner and Azure Expert Managed Services Provider (MSP).

Infogain, an Apax Funds portfolio company, has offices in California, Washington, Texas, the UK, the UAE, and Singapore, with delivery centers in Seattle, Houston, Austin, Kraków, Noida, Gurgaon, Mumbai, Pune, and Bengaluru.

Express Application
Upload Microsoft word, PDF file upto 500KB.
Recent Jobs
Posted on June 26, 2025
Cloud Native App Developer (Standard) | 4.5-6 Years | CNA Development - Spring Boot Microservices, Core Java
Posted on June 26, 2025
Frontend React Developer (Standard) | 4.5-6 Years | Frontend Development - ReactJS
Posted on June 26, 2025
Mevn Developer (Senior) | 8-11 Years | Open Source Development - Vue.js, Java Architecture, Node.js, MongoDB DBA
Posted on June 26, 2025
Platform Engineer (Lead) | 8-11 Years | Platform Engineering - Ansible, Docker, Kubernetes, Terraform