Katkat

Topic

Machine Learning interview questions

572 Machine Learning questions from 133 companies, with hints, runnable code and solution walkthroughs.

Questions572

59 easy323 medium190 hard

Your progress0/572

Machine Learning interview questions by the numbers

Last updated

Katkat has 572 Machine Learning interview questions from 133 companies; the most come from Amazon (10%), OpenAI (6%) and Meta (5%). By difficulty: 10% easy, 56% medium and 33% hard. The most common round is Technical Screen (61%). Frequent concepts include Machine Learning, Feature Engineering, Model Evaluation and Optimization.

Companies (133)

Difficulty

  • Easy10% · 59
  • Medium56% · 323
  • Hard33% · 190

Interview rounds

  • Technical Screen61% · 348
  • Onsite28% · 163
  • Online Assessment8% · 46
  • HR Screen3% · 15

Roles

  • Data Scientist46% · 264
  • Machine Learning Engineer35% · 202
  • Software Engineer9% · 53
  • Applied Scientist5% · 26
  • Quantitative Researcher2% · 13

Frequent concepts

  • Machine Learning7% · 39
  • Feature Engineering5% · 29
  • Model Evaluation4% · 23
  • Optimization4% · 23
  • Deep Learning4% · 21
  • System Design3% · 18

Showing 361–390 of 572 results

  • This question evaluates a candidate's ability to diagnose a poorly generalizing ML classifier, testing practical knowledge of data leakage, feature drift, and model calibration. It assesses applied machine learning debugging skills — including cross-validation design and error analysis — commonly tested in senior engineering and ML roles.

    OpenAISoftware EngineerMachine LearningTechnical Screen
  • An OpenAI ML engineer technical-screen debugging question: given a PyTorch Transformer training pipeline whose loss diverges and validation accuracy stays near chance (plus mask IndexErrors, AMP crashes, and nondeterminism), identify the distinct bugs and fix them. It tests Transformer internals - attention/causal masking, teacher-forcing label shift, positional encodings, ignore_index loss, optimizer and GradScaler order, determinism, and DDP - along with writing unit tests and a structured debugging plan.

    OpenAIMachine Learning EngineerMachine LearningTechnical Screen
  • This question evaluates debugging and practical implementation skills for transformer-based text classification, covering model correctness, training loop integrity, data preprocessing, and evaluation metrics within the Machine Learning / Natural Language Processing domain.

    OpenAIMachine Learning EngineerMachine LearningTechnical Screen
  • This question evaluates a candidate's ability to synthesize the state of the art in Machine Learning, defend a research direction, and design rigorous experiments, measuring competencies in literature analysis, methodological justification, experimental design, and technical communication.

    OpenAIMachine Learning EngineerMachine LearningOnsite
  • This question evaluates understanding of backpropagation through linear blocks, matrix calculus, and the computation of gradients with respect to individual weight matrices in a product of trainable matrices, including reasoning about tensor shapes and matrix differentials.

    OpenAIMachine Learning EngineerMachine LearningTechnical Screen
  • Derive forward and backward computation for column-sharded matrix multiplication across multiple devices. Track every local tensor shape, identify the collective needed for replicated input gradients, and propose tests for shard ordering, scaling, and numerical errors.

    OpenAIMachine Learning EngineerMachine LearningOnsite
  • This question tests the ability to optimize restart intervals for a randomized algorithm with heavy-tailed runtime distributions, balancing the trade-off between progress and computation waste.

    OpenAISoftware EngineerMachine LearningHR Screen
  • This question evaluates practical debugging skills for Transformer-based sequence models, targeting machine learning engineers who must diagnose intermittent shape/dtype errors and convergence failures simultaneously. It tests deep familiarity with the full training stack — data pipelines, attention masking, positional encoding, optimizer precision, and distributed setup — and the ability to reason about how a single root cause can manifest across multiple symptoms.

    OpenAIMachine Learning EngineerMachine LearningTechnical Screen
  • Tests the ability to detect and filter low-quality human labels in large-scale dataset pipelines by modeling annotator reliability, distinguishing genuine ambiguity from noise, and evaluating downstream model performance.

    OpenAIMachine Learning EngineerMachine LearningOnsite
  • Derive and implement, without an autodiff library, the backward pass of a running (cumulative) product in linear time without division, then a numerically stable softmax cross-entropy and its gradient. Tests chain-rule fluency, prefix/suffix tricks, and gradient checking.

    OpenAIMachine Learning EngineerMachine LearningOnsite
  • Write gradients by hand: the forward and backward pass of the prefix-product operation, in linear time and without division so inputs may contain zeros, and a numerically stable softmax cross-entropy loss with its gradient in NumPy. It tests the chain rule, prefix and suffix products, and gradient checking.

    OpenAIMachine Learning EngineerMachine LearningOnsite
  • Tests your ability to implement a fast, vectorized 1-Nearest Neighbor (1NN) classifier using NumPy, focusing on memory-efficient distance computations and tie-breaking.

    OpenAIMachine Learning EngineerMachine LearningTechnical Screen
  • Implement a numerically stable, masked cross-entropy loss function with label smoothing and temperature scaling in NumPy, returning both the loss and analytical gradients.

    OpenAIMachine Learning EngineerMachine LearningTechnical Screen
  • This question tests your ability to implement fundamental neural network components using NumPy, focusing on matrix operations, broadcasting, and numerical stability.

    OpenAIData ScientistMachine LearningOnsite
  • Work through inclusive prefix products in separate-output, in-place, reverse-gradient, and parallel variants. Explain zeros, overflow, mutation and aliasing, constant temporary workspace, synchronization, floating-point effects, and the work/depth trade-offs of each part.

    OpenAIMachine Learning EngineerMachine LearningOnsite
  • Implement a 1-nearest-neighbor classifier that labels each query point with the label of its closest training point under Euclidean distance, then extend it to Manhattan distance. Tests vectorized distance computation under memory limits, tie handling and how the choice of metric changes predictions.

    OpenAIMachine Learning EngineerMachine LearningTechnical Screen
  • Implement the core of a PyTorch-style autograd system: a tensor wrapper that records addition, multiplication, matrix multiplication, ReLU and sum, plus a backward method that fills in gradients for every input. Tests reverse-mode differentiation, correct ordering of the backward pass, gradient accumulation for tensors used more than once, and verification against PyTorch.

    OpenAISoftware EngineerMachine LearningTechnical Screen
  • This question evaluates understanding of backpropagation and the chain rule, proficiency in vectorized NumPy implementation, numerical stability of softmax/cross-entropy, and competency in gradient checking and debugging of model gradients.

    OpenAISoftware EngineerMachine LearningTechnical Screen
  • This question evaluates a candidate's ability to handle label noise by estimating annotator reliability and applying data-cleaning, relabeling, or reweighting methods while measuring model performance and understanding classification metrics.

    OpenAIMachine Learning EngineerMachine LearningOnsite
  • This question evaluates a candidate's ability to analyze noisy multi-annotator labels, design label-aggregation and dataset-splitting strategies, and choose modeling and evaluation approaches for binary text classification.

    OpenAIMachine Learning EngineerMachine LearningTechnical Screen
  • Find four planted bugs in a compact PyTorch decoder-only language model (some produce NaNs, some silently ruin learning), then add key/value caching so generation processes one new token per step.

    OpenAIMachine Learning EngineerMachine LearningOnsite
  • This question evaluates proficiency in end-to-end supervised machine learning workflows—covering data validation, exploratory data analysis, preprocessing, model training and selection, evaluation metrics and calibration, fairness assessments, and reproducibility—and tests competencies in applied machine learning engineering, statistical reasoning, and model governance within the Machine Learning domain. It is commonly asked in technical interviews because it verifies practical implementation skills and judgment for deploying reliable classifiers, aligning metrics to business goals and diagnosing performance and fairness across slices, with emphasis on practical application informed by conceptual understanding.

    OpenAIMachine Learning EngineerMachine LearningTechnical Screen
  • Tests vectorized distance computation in array libraries and the mathematical equivalence between nearest-prototype classification and an affine neural network layer.

    OpenAIMachine Learning EngineerMachine LearningTechnical Screen
  • This question evaluates a candidate's understanding of probabilistic modeling and expectation calculations for stochastic arrival processes, touching on renewal-theory concepts and length-biased sampling phenomena relevant to timing and queuing problems.

    OptiverData ScientistMachine LearningTechnical Screen
  • This question evaluates a candidate's quantitative estimation and uncertainty-quantification skills, focusing on decomposition into estimable factors, clear articulation of assumptions, calibrated confidence intervals, and updating estimates when new aggregate data are provided.

    OptiverQuantitative ResearcherMachine LearningTechnical Screen
  • This problem tests your ability to translate statistical distributions and categorical data into ranked probabilistic likelihoods, accounting for uncertainty and sample size variance.

    OptiverData ScientistMachine LearningOnline Assessment
  • This question evaluates probabilistic reasoning, decision-theoretic optimization of confidence intervals, and sequential risk–reward trade-offs by requiring interval estimation under a nonstandard scoring rule, relevant to a Data Scientist and categorized in Machine Learning/statistical decision theory.

    OptiverData ScientistMachine LearningTechnical Screen
  • This question evaluates probabilistic reasoning, expected-value calculation, risk management, and market-making intuition under time pressure within a simulated trading game.

    OptiverQuantitative TraderMachine LearningTechnical Screen
  • This question evaluates quantitative probability reasoning, expected-value computation, risk-sizing (including Kelly-style sizing), arbitrage recognition, and market-making intuition within a live trading simulation.

    OptiverQuantitative ResearcherMachine LearningTechnical Screen
  • This question tests the ability to design an end-to-end evaluation and data pipeline for a clinical conversational agent, emphasizing safety, bias mitigation, and rigorous validation in a high-stakes environment.

    OracleData ScientistMachine LearningTechnical Screen

More topics