Data Science Engineer

FirstHive; Bengaluru; On-site; INR 600,000 - 900,000; Full time

Job summary

FirstHive in Bengaluru is seeking a junior ML Engineer to join the AI and Data Science team. This full-time role targets 0–1 year of experience and centers on building ML capabilities for our customer data platform (CDP).

You'll develop models for segmentation, churn, and recommendations, write SQL for feature engineering, and deploy APIs with FastAPI, collaborating with data engineers to ensure robust data pipelines and production-ready systems.

Qualifications

• Strong Python programming skills are required.
• Familiarity with Pandas, NumPy and scikit-learn.
• Basic understanding of LLM APIs, embeddings and REST APIs.

Responsibilities

• Develop ML models for customer segmentation and entity resolution.

Skills

Python

Pandas

NumPy

scikit-learn

SQL

LLM APIs

Git

Education

B.E./B.Tech/M.Tech/M.Sc in CS/DS/Statistics

Tools

FastAPI

Job description

About the role

This is a full-time position within the AI and Data Science team. The role is based in Bengaluru and is open to candidates with 0–1 years of experience. The work focuses on building machine-learning and AI capabilities for FirstHive's CDP platform.

What you'll work on

• Develop ML models for customer segmentation and entity resolution.
• Build models addressing churn and conversion use cases.
• Work on recommendation systems for the platform.
• Develop LLM-powered features using prompting techniques, embeddings and RAG.
• Write SQL for feature engineering and analytical workflows.
• Build model-serving APIs using FastAPI.
• Support deployment and ongoing monitoring of ML models.
• Collaborate with data engineers to create dependable data and machine-learning pipelines.

What we\'re looking for

• Strong Python programming skills are required.
• Candidates should be comfortable with Pandas, NumPy and scikit-learn.
• A good understanding of core machine-learning concepts and SQL is expected.
• Basic familiarity with LLM APIs, embeddings and REST APIs is required.
• Applicants should know Git and demonstrate a strong willingness to learn.
• Academic ML projects, internships, Kaggle work or GitHub machine-learning projects are considered a plus.
• Eligible qualifications include B.E., B.Tech, M.Tech or M.Sc. in Computer Science, Data Science, Statistics, Mathematics or a related field.

Good to know

• The role offers exposure to real-world AI and ML projects and production-focused engineering.
• Mentorship, high ownership and experience in a fast-growing B2B SaaS startup are highlighted.