AI Team Lead

Moxie Retail; Hyderabad; On-site; INR 300,000 - 600,000; Full time

Job summary

Moxie Retail in Hyderabad, India seeks a senior AI/ML engineer to design, develop, and deploy models for classification, forecasting, and anomaly detection across large-scale datasets. You will build feature pipelines, evaluate models, apply NLP and LLM techniques, and collaborate with product and engineering to deploy solutions using Docker and cloud platforms.

The role demands 9+ years of hands-on experience, strong Python and ML library skills, and the ability to translate business problems

Qualifications

• 9+ years of hands-on experience in AI/ML model development and deployment.
• Proficiency in Python and AI/ML libraries such as PyTorch, TensorFlow, Scikit-Learn, Pandas, and NumPy.
• Experience with NLP libraries such as Hugging Face Transformers, spaCy, and NLTK.
• Hands-on experience with LLM fine-tuning, prompt engineering, and transformer-based models.
• Experience with SQL and/or NoSQL databases and data manipulation at scale.
• Knowledge of cloud platforms AWS/Azure, Docker, and ML deployment workflows.
• Strong analytical, problem solving, and communication skills, with the ability to explain complex AI concepts to technical and non technical stakeholders.

Responsibilities

• Build and optimize feature engineering pipelines and data preprocessing workflows for structured and unstructured datasets.
• Perform exploratory data analysis (EDA) to identify trends, patterns, and actionable insights from large-scale healthcare and business datasets.
• Develop and evaluate models using appropriate metrics, cross-validation, and experimentation frameworks.
• Implement NLP solutions for text classification, entity extraction, summarization, and information retrieval.
• Design and fine-tune Large Language Models LLMs and transformer-based architectures for domain-specific applications.
• Apply transfer learning and prompt engineering techniques to adapt foundation models to business and healthcare use cases.
• Build AI driven document intelligence solutions using models such as LayoutLM, Donut, and Table Transformers for key-value extraction and document understanding.
• Support MLOps and model deployment, including Docker based packaging, cloud deployment, monitoring, and performance optimization.
• Collaborate with product, engineering, and business teams to translate business problems into AI ML and GenAI solutions.
• Research and experiment with emerging AI technologies, including RAG (Retrieval Augmented Generation), LLM alignment techniques RLHF, DPO, PPO, and KTO, and model optimization methods.

Skills

AI/ML development

Python

NLP

LLM fine-tuning

SQL/NoSQL

Cloud platforms

MLOps

Problem solving

Communication

Education

MS/PhD in CS/AI/ML

Bachelor's in CS/Math/EE

Tools

PyTorch

TensorFlow

Scikit-Learn

Pandas

NumPy

Hugging Face Transformers

spaCy

NLTK

Docker

Jupyter

Job description

Job Summary

Design, develop, and deploy machine learning and deep learning models for classification, regression, clustering, forecasting, anomaly detection, and predictive analytics use cases.

Responsibilities

• Build and optimize feature engineering pipelines and data preprocessing workflows for structured and unstructured datasets.
• Perform exploratory data analysis (EDA) to identify trends, patterns, and actionable insights from large-scale healthcare and business datasets.
• Develop and evaluate models using appropriate metrics, cross-validation, and experimentation frameworks.
• Implement NLP solutions for text classification, entity extraction, summarization, and information retrieval.
• Design and fine-tune Large Language Models LLMs and transformer-based architectures for domain-specific applications.
• Apply transfer learning and prompt engineering techniques to adapt foundation models to business and healthcare use cases.
• Build AI driven document intelligence solutions using models such as LayoutLM, Donut, and Table Transformers for key-value extraction and document understanding.
• Support MLOps and model deployment, including Docker based packaging, cloud deployment, monitoring, and performance optimization.
• Collaborate with product, engineering, and business teams to translate business problems into AI ML and GenAI solutions.
• Research and experiment with emerging AI technologies, including RAG (Retrieval Augmented Generation), LLM alignment techniques RLHF, DPO, PPO, and KTO, and model optimization methods.

Requirements

• 9+ years of hands-on experience in AI/ML model development and deployment.
• Strong understanding of machine learning, deep learning, neural network architectures, training methodologies, and optimization algorithms.
• Proficiency in Python and AI/ML libraries such as PyTorch, TensorFlow, Scikit-Learn, Pandas, and NumPy.
• Experience with NLP libraries such as Hugging Face Transformers, spaCy, and NLTK.
• Hands-on experience with LLM fine-tuning, prompt engineering, and transformer-based models.
• Experience with SQL and/or NoSQL databases and data manipulation at scale.
• Knowledge of cloud platforms AWS/Azure, Docker, and ML deployment workflows.
• Strong analytical, problem solving, and communication skills, with the ability to explain complex AI concepts to technical and non technical stakeholders.

Preferred Skills

• Experience with RAG architectures, vector databases, embeddings, and semantic search.
• Familiarity with LLM alignment techniques such as RLHF, DPO, PPO, and KTO.
• Experience with computer vision or document AI models for OCR and document understanding.
• Knowledge of Airflow or other workflow orchestration tools.
Experience working with HIPAA, PHI, PII, or GDPR compliant systems.
• Familiarity with Linux, Git, Jupyter Notebooks, and Agile development practices.
• Experience with distributed computing and scalable AI infrastructure is a strong advantage.