Skip to content
PPreptima

Data & AI Engineering interview questions

Questions for data engineering, ML engineering, and the LLM-application roles now appearing in mainstream hiring loops.

13 published across 8 topics.

Data and AI engineering fundamentals58 short answers on one page, for revising rather than studying.

Python for Data

1 question

Idiomatic Python, pandas and vectorisation, memory behaviour, and writing correct transformations.

Data Pipelines

2 questions

Batch versus streaming, orchestration, idempotent backfills, and lakehouse and medallion layering.

hardDesignScenario

How would you design a data pipeline you can safely re-run?

Make the output of a run a pure function of its partition key and write it by overwriting that partition rather than appending, so a re-run replaces instead of duplicating. Then design for late events, a bounded lookback, and additive schema change, and assume at-least-once delivery into an idempotent sink.

6 minmid, senior, staff

Spark & Big Data

1 question

Partitioning and shuffles, skew, joins at scale, catalyst and lazy evaluation, and cluster tuning.

Warehousing & Modelling

2 questions

Star and snowflake schemas, slowly changing dimensions, and dbt-style transformation layering.

ML Fundamentals

2 questions

Bias and variance, evaluation-metric selection, leakage, imbalanced data, and validation strategy.

mediumConceptScenario

Your fraud model is 99.4% accurate. Is that good?

Almost certainly not: if fraud is 0.6% of transactions, a model predicting 'not fraud' for everything scores 99.4% and catches nothing. The metric has to follow from which error costs more, which makes it a business decision expressed in numbers rather than a modelling one.

4 minmid, senior
mediumConceptScenario

How do you tell that a model is overfitting, and what do you do about it?

Overfitting shows up as a validation score materially worse than the training score, but only if the split respects time order and group structure. Fix it by improving the data first, then simplifying, regularising and stopping early, and keep a test set you read once.

4 minentry, mid, senior

MLOps

2 questions

Experiment tracking, model registries, deployment patterns, drift detection, and retraining triggers.

hardDesignScenario

What has to be in place before you would let a model serve live traffic?

Features must be computable at decision time, training and serving must share one transformation implementation, the model and dataset version must be pinned together, drift and realised performance monitored separately because labels arrive late, and rollback must move model and features as one unit.

6 minmid, senior, staff, lead

LLMs & RAG

2 questions

Embeddings and vector search, chunking, retrieval evaluation, prompt design, and cost and latency control.

hardScenarioDesign

Your RAG system returns confident but wrong answers. How do you debug it?

Isolate the layer before changing anything: check whether the correct passage was retrieved at all. If it was not, the fault is in the retrieval stack; if it was, the generator is at fault. Then measure both separately with retrieval recall@k and faithfulness scored on a labelled eval set.

5 minmid, senior, staff

Statistics & Experimentation

1 question

Distributions, hypothesis testing, A/B design, and reading a result honestly.