Weekday AI

Data Scientist

Weekday AI Bengaluru, Karnataka, India · ₹3M–₹5M/yr

Technology, Information and Internet · 11-50 employees

Oct 19
data-scientist Junior (0-2 yrs) Full-time India
Create a free account to apply — email only, no card. You can also save this posting or score it against your profile with AI.

About the role

Fine-tune and deploy machine learning, NLP, and LLM solutions for entity extraction, classification, semantic retrieval, ranking, and knowledge graph applications. Build and evaluate scalable production pipelines, analyze large datasets, monitor deployed models, and collaborate with product, data, and engineering teams.

What they look for

Python Machine Learning Natural Language Processing Large Language Models PyTorch Hugging Face Transformers Sentence Transformers Pandas Scikit-Learn SQL Transformer Fine-Tuning LoRA QLoRA Semantic Search Knowledge Graphs Model Evaluation

Requirements

Requires at least one year of relevant experience and strong Python skills, with hands-on experience in applied machine learning, NLP, transformer models, and LLM fine-tuning. Candidates should understand embeddings, information retrieval, model evaluation, SQL, and production ML practices; experience with graph technologies and large-scale taxonomies is an advantage.

Full description

𝗧𝗵𝗶𝘀 𝗿𝗼𝗹𝗲 𝗶𝘀 𝗳𝗼𝗿 𝗼𝗻𝗲 𝗼𝗳 𝘁𝗵𝗲 𝗪𝗲𝗲𝗸𝗱𝗮𝘆'𝘀 𝗰𝗹𝗶𝗲𝗻𝘁𝘀

𝗦𝗮𝗹𝗮𝗿𝘆 𝗿𝗮𝗻𝗴𝗲: 𝗥𝘀 𝟯𝟬𝟬𝟬𝟬𝟬𝟬 - 𝗥𝘀 𝟱𝟬𝟬𝟬𝟬𝟬𝟬 (𝗶𝗲 𝗜𝗡𝗥 𝟯𝟬-𝟱𝟬 𝗟𝗣𝗔)

Experience: 1+ yrs

Location: Bangalore

We are looking for a talented and technically strong Data Scientist I / II to develop and deploy production-grade machine learning, Natural Language Processing (NLP), and Large Language Model (LLM) solutions. The role focuses on building intelligent systems for entity extraction, classification, semantic retrieval, ranking, and knowledge graph applications using large-scale enterprise datasets.

The ideal candidate will have hands-on experience with Python, transformer models, applied NLP, and machine learning, along with a strong understanding of model evaluation and production deployment. This is an opportunity to work on complex AI challenges involving millions of records while building scalable solutions that deliver measurable business value.

Key Responsibilities

  • Fine-tune transformer models and LLMs using techniques such as LoRA and QLoRA for entity extraction, classification, and reranking.
  • Develop NLP models to extract, classify, and link entities from resumes, job descriptions, and other unstructured enterprise data.
  • Build prediction models, knowledge graphs, and graph traversal solutions to represent relationships between skills, roles, and competencies.
  • Design and implement semantic retrieval, clustering, and learning-to-rank pipelines across large taxonomies and datasets.
  • Develop embedding-based search solutions using bi-encoders, cross-encoders, and related retrieval techniques.
  • Apply classical machine learning methods, including gradient boosting, logistic regression, clustering, and dimensionality reduction.
  • Design evaluation frameworks, select appropriate metrics, analyse model errors, and validate performance through systematic experiments.
  • Explore datasets containing millions of records using Python, SQL, and data-processing tools.
  • Deploy, monitor, and improve machine learning models in production environments.
  • Use experiment tracking, model versioning, Docker, and cloud infrastructure to support reproducible ML workflows.
  • Collaborate with product, data, and engineering teams to translate business problems into effective AI solutions.
  • Document experiments, communicate findings, and continuously improve model accuracy, efficiency, and reliability.

What Makes You a Great Fit

  • 1+ years of relevant experience in applied machine learning, NLP, or data science; experience expectations may vary between Data Scientist I and II.
  • Strong proficiency in Python, with hands-on experience using PyTorch, Hugging Face Transformers, sentence-transformers, pandas, and scikit-learn.
  • Practical experience fine-tuning transformer models on custom datasets for named entity recognition, classification, or related NLP tasks.
  • Experience with LLM fine-tuning and parameter-efficient techniques such as PEFT, LoRA, or QLoRA.
  • Strong understanding of embeddings, attention mechanisms, contrastive learning, semantic search, and information retrieval.
  • Working knowledge of classical machine learning algorithms and statistical evaluation methods.
  • Experience with SQL and the ability to analyse large-scale datasets.
  • Understanding of model evaluation, sampling, statistical significance, calibration, and metric selection.
  • Familiarity with ML deployment practices, experiment tracking, model versioning, Docker, and GPU-based training or serving on AWS.
  • Experience with learning-to-rank, knowledge graphs, entity linking, or large-scale taxonomies is an advantage.
  • Familiarity with Neo4j, Cypher, pgvector, graph ML, GLiNER, NuNER, vLLM, ONNX, AWS Bedrock, or SageMaker is a plus.
  • Strong analytical thinking, problem-solving, and written communication skills.
  • Ability to work collaboratively, take ownership of experiments, and translate research into production-ready solutions.
  • A strong interest in applied AI, modern NLP, LLM systems, and continuous experimentation.

Similar roles