Topic
Machine Learning interview questions
572 Machine Learning questions from 133 companies, with hints, runnable code and solution walkthroughs.
59 easy323 medium190 hard
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
Most mentioned in interview experiences
- 1.Answer core ML fundamentals questions2 reports
- 2.Compare deep learning framework trends2 reports
- 3.Evaluate Factual Accuracy and Consistency in a Domain-Specific RAG Chatbot2 reports
- 4.Explain Core ML Interview Concepts2 reports
- 5.Explain Overfitting and Transformer Attention2 reports
- 6.Explain SHAP in an ML System2 reports
- 7.Implement bagging with decision trees2 reports
- 8.Present and defend your data challenge end-to-end2 reports
- 9.Analyze Temperatures and Update Regression1 report
- 10.Analyze Town Temperatures and Select Predictors for NYC1 report
Counted from 122 candidate interview experiences on Katkat. How we collect and count
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 ScreenAn 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 ScreenThis 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 ScreenThis 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 LearningOnsiteThis 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 ScreenDerive 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 LearningOnsiteThis 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 ScreenThis 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 ScreenTests 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 LearningOnsiteDerive 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 LearningOnsiteWrite 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- Implement 1NN with NumPyMedium
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 ScreenThis question tests your ability to implement fundamental neural network components using NumPy, focusing on matrix operations, broadcasting, and numerical stability.
OpenAIData ScientistMachine LearningOnsiteWork 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 LearningOnsiteImplement 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 ScreenImplement 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 ScreenThis 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 ScreenThis 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 LearningOnsiteThis 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 ScreenFind 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 LearningOnsiteThis 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 ScreenTests vectorized distance computation in array libraries and the mathematical equivalence between nearest-prototype classification and an affine neural network layer.
OpenAIMachine Learning EngineerMachine LearningTechnical ScreenThis 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 ScreenThis 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 ScreenThis 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 AssessmentThis 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 ScreenThis question evaluates probabilistic reasoning, expected-value calculation, risk management, and market-making intuition under time pressure within a simulated trading game.
OptiverQuantitative TraderMachine LearningTechnical ScreenThis 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 ScreenThis 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