Coding Interview Prep · 65 min · numpy · pandas · Python

ML Coding Under Pressure: numpy, pandas & ML Micro-Tasks

ML take-homes test whether you can manipulate data correctly, fast, without Stack Overflow.

Hiring signal: Implements ML primitives from scratch fluently — no library lookup under pressure

What you will learn

Perform vectorised numpy operations without Python loops
Build feature-engineering pipelines in pandas from scratch
Implement common ML metrics (accuracy, precision/recall, ROC-AUC) from numpy
Manipulate and join DataFrames under timed conditions
Spot and fix common bugs: broadcasting errors, off-by-one, NaN propagation

The Problem

ML interviews are increasingly bifurcated: standard LeetCode patterns and ML-specific coding tasks — implement softmax from scratch, compute a rolling feature, write a confusion matrix from arrays. These tasks are easy with libraries and hard under pressure because muscle memory matters. The candidates who pass spend time drilling these primitives until they're automatic. This lesson gives you the drill.

The Concept

Why ML Coding Is a Separate Skill

Standard DSA interviews test algorithmic thinking. ML coding interviews test something different: can you manipulate numerical data correctly without libraries doing the thinking for you? The interviewer wants to see that you understand what softmax does, not just that you can call torch.nn.Softmax().

This matters because in production ML, you constantly need to: implement metrics from scratch (your eval pipeline doesn't have sklearn), vectorize operations for performance (loops are 100x slower than numpy), manipulate tensors with broadcasting (shape errors are the most common bug), and implement small ML primitives (custom loss functions, custom metrics, data transforms). If you can only do these with library calls, you can't debug when the library doesn't expose what you need.

The three skill areas:

numpy vectorization: replacing Python loops with array operations — the single most important skill for ML coding
ML metrics from scratch: precision, recall, F1, confusion matrix, cross-entropy — using only numpy
ML primitives from scratch: softmax, k-means step, kNN, gradient descent step — proving you understand the math

Unlock the full lesson

You've read the first 2 sections. The rest of this lesson covers The Patterns, The Drill Protocol, Ship It, Use It, Evaluation, Exercises, Key Terms, Common Pitfalls, Interview Framing — plus a hands-on lab, quiz, and project artifact.

Create a free account to unlock Phase 0 and Phase 1 of every course — no credit card.

Key terms in this lesson

Softmax