Vijay Sales logo

AI/ML Engineer

0 years

6 - 15 Lacs

pune maharashtra india

Posted:1 day ago| Platform: Linkedin logo

Apply

Skills Required

ai ml data analytics automation architecture cutting learning development forecasting personalization inventory optimization pricing retrieval support engineering databricks pyspark mlflow unity model tracking sql aggregation erp crm automate deployment rest docker github azure devops monitoring drift training regression pytorch tensorflow spark programming writing git sagemaker segmentation python visualization

Work Mode

On-site

Job Type

Full Time

Job Description

About Vijay Sales

Vijay Sales is one of India’s leading electronics retail brands with 160+ stores nationwide and a fast-growing digital presence. We are on a mission to build the most advanced data-driven retail intelligence ecosystem—using AI, predictive analytics, LLMs, and real-time automation to transform customer experience, supply chain, and omnichannel operations.

Role Overview

We are looking for a highly capable

AI Engineer

who is passionate about building production-grade AI systems, designing scalable ML architecture, and working with cutting-edge AI/ML tools. This role involves hands-on work with

Databricks

,

SQL

,

PySpark

, modern

LLM/GenAI frameworks

, and full lifecycle ML system design.

Key Responsibilities

Machine Learning & AI Development

  • Build, train, and optimize ML models for forecasting, recommendation, personalization, churn prediction, inventory optimization, anomaly detection, and pricing intelligence.
  • Develop GenAI solutions using modern LLM frameworks (e.g., LangChain, LlamaIndex, HuggingFace Transformers).
  • Explore and implement RAG (Retrieval Augmented Generation) pipelines for product search, customer assistance, and support automation.
  • Fine-tune LLMs on company-specific product and sales datasets (using QLoRA, PEFT, and Transformers).
  • Develop scalable feature engineering pipelines leveraging Delta Lake and Databricks Feature Store.

Databricks / Data Engineering

  • Build end-to-end ML workflows on Databricks using PySpark, MLflow, Unity Catalog, Delta Live Tables.
  • Optimize Databricks clusters for cost, speed, and stability.
  • Maintain reusable notebooks and parameterized pipelines for model ingestion, validation, and deployment.
  • Use MLflow for tracking experiments, model registry, and lifecycle management.

Data Handling & SQL

  • Write advanced SQL for multi-source data exploration, aggregation, and anomaly detection.
  • Work on large, complex datasets from ERP, POS, CRM, Website, and Supply Chain systems.
  • Automate ingestion of streaming and batch data into Databricks pipelines.

Deployment & MLOps

  • Deploy ML models using REST APIs, Databricks Model Serving, Docker, or cloud-native endpoints.
  • Build CI/CD pipelines for ML using GitHub Actions, Azure DevOps, or Databricks Workflows.
  • Implement model monitoring for drift, accuracy decay, and real-time alerts.
  • Maintain GPU/CPU environments for training workflows.

Must-Have Technical Skills

Core AI/ML

  • Strong fundamentals in machine learning: regression, classification, time-series forecasting, clustering.
  • Experience in deep learning using PyTorch or TensorFlow/Keras.
  • Expertise in LLMs, embeddings, vector databases, and GenAI architecture.
  • Hands-on experience with HuggingFace, embedding models, and RAG.

Databricks & Big Data

  • Hands-on experience with Databricks (PySpark, SQL, Delta Lake, MLflow, Feature Store).
  • Strong understanding of Spark execution, partitioning, and optimization.

Programming

  • Strong proficiency in Python.
  • Experience writing high-performance SQL with window functions, CTEs, and analytical queries.
  • Knowledge of Git, CI/CD, REST APIs, and Docker.

MLOps & Production Engineering

  • Experience deploying models to production and monitoring them.
  • Familiarity with tools like MLflow, Weights & Biases, or SageMaker equivalents.
  • Experience in building automated training pipelines and handling model drift/feedback loops.

Preferred Domain Experience

  • Retail/e-commerce analytics
  • Demand forecasting
  • Inventory optimization
  • Customer segmentation & personalization
  • Price elasticity and competitive pricing
Skills:- Python, Artificial Intelligence (AI), Generative AI, databricks and Data Visualization

More Jobs at Vijay Sales

E Commerce - Technical Co-Ordinator & Ops Executive.

Mumbai Suburbs, Mumbai (All Areas)

2 - 5 yrs
|

INR 1 - 5 Lacs

Cashier

Kukatpalli, Andhra Pradesh

2.0 - 2.0 yrs
|

INR 0 - 0 Lacs

Cashier

India

2.0 - 2.0 yrs
|

INR 0 - 0 Lacs

Cashier Executive

Navi Mumbai, Mumbai (All Areas)

1.0 - 4.0 yrs
|

INR 2 - 3 Lacs

Assistant Manager - Accounts (MIS)

Mumbai Suburban, Mumbai (All Areas)

3.0 - 7.0 yrs
|

INR 6 - 8 Lacs

Mock Interview

Practice Video Interview with JobPe AI

Start PySpark Interview
cta

Start Your Job Search Today

Browse through a variety of job opportunities tailored to your skills and preferences. Filter by location, experience, salary, and more to find your perfect fit.

Job Application AI Bot

Job Application AI Bot

Apply to 20+ Portals in one click

Download Now

Download the Mobile App

Instantly access job listings, apply easily, and track applications.

Google Play App Store
coding practice

Enhance Your Python Skills

Practice Python coding challenges to boost your skills

Start Practicing Python Now
Vijay Sales logo
Vijay Sales

Retail / Consumer Electronics

Mumbai

RecommendedJobs for You

Cozzera logo

AI/ML Engineer

Cozzera

india

Devlats Pvt Ltd logo

AI/ML Engineer

Devlats Pvt Ltd

pune, maharashtra, india

TransPerfect logo

AI/ML Engineer

TransPerfect

bengaluru, karnataka, india

UBS logo

AI/ML Engineer

UBS

pune, maharashtra, india

Vijay Sales logo

AI/ML Engineer

Vijay Sales

pune, maharashtra, india

SourcingXPress logo

AI/ML Engineer

SourcingXPress

jaipur, rajasthan, india

Kanhasoft Pvt.Ltd logo

AI/ML Engineer

Kanhasoft Pvt.Ltd

india

EA Tech Corporations Technology For Automation Engineering Private Limited logo

AI/ML Engineer

EA Tech Corporations Technology For Automation Engineering Private Limited

bhubaneswar, odisha, india

hackajob logo

AI/ML Engineer

hackajob

hyderabad, telangana, india

Techify Solutions Pvt Ltd logo

AI/ML Engineer

Techify Solutions Pvt Ltd

ahmedabad, gujarat, india