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 31–60 of 572 results
This question evaluates understanding of collaborative filtering approaches and related competencies such as neighborhood-based versus matrix factorization methods, differences for explicit versus implicit feedback, pointwise versus pairwise loss trade-offs, regularization, cold-start and popularity-bias mitigation, evaluation metrics, and scalability considerations. It is asked in the Machine Learning domain to assess both conceptual understanding and practical application for designing, evaluating (e.g., NDCG, MAP) and scaling recommender systems in technical interviews.
AmazonApplied ScientistMachine LearningOnsiteThis question evaluates core machine learning fundamentals including statistical modeling assumptions and loss functions (linear and logistic regression), ensemble methods and feature sampling in random forests, optimization algorithms (Adam versus stochastic gradient descent), and neural network capacity and training dynamics.
AmazonApplied ScientistMachine LearningTechnical ScreenA two-part conversational question for AI infrastructure roles: explain how a decoder-only transformer turns a prompt into next-token probabilities, then explain how vLLM serves such models efficiently. It tests transformer internals, KV cache sizing, prefill versus decode bottlenecks, paged KV memory and continuous batching.
AmazonSoftware EngineerMachine LearningOnsiteThis interview question evaluates core ML concepts, assumptions, math intuition, training/evaluation trade-offs, and practical failure modes in a realistic interview setting. A strong answer for Explain K-Fold Cross-Validation and Its Trade-Offs states assumptions, handles edge cases, explains trade-offs, and shows how to validate the result clearly.
AmazonData ScientistMachine LearningTechnical ScreenThis question evaluates understanding of transformer-based LLM architectures, positional embedding variants, parameter-efficient fine-tuning (PEFT) approaches, regularization strategies, and evaluation methodologies within the Machine Learning domain focused on large language models and NLP, emphasizing both conceptual understanding and practical application. It is commonly asked to assess the interviewee’s ability to reason about architecture and tuning trade-offs, model generalization and stability, and the design of reliable offline and online evaluation pipelines for production-minded language model systems.
AmazonApplied ScientistMachine LearningTechnical ScreenThis question evaluates a candidate's understanding of normalization techniques in deep learning—specifically Layer Normalization in Transformer blocks—covering the LayerNorm equation and the roles of gamma, beta, and epsilon, as well as contrasts with BatchNorm and RMSNorm, in the Machine Learning domain.
AmazonApplied ScientistMachine LearningOnsiteThis question evaluates understanding of logistic regression fundamentals, covering probabilistic modeling, Bernoulli likelihood and logit-link derivation, loss and gradient characterization, calibration and miscalibration remedies, regularization trade-offs, feature scaling and interactions, class imbalance strategies, evaluation metrics for skewed data, coefficient interpretation, and common failure modes. It is commonly asked in technical interviews to assess both theoretical and practical competencies in supervised learning within the Machine Learning domain, with a level of abstraction spanning conceptual derivation and practical application of model behavior and evaluation.
AmazonApplied ScientistMachine LearningOnsiteThis question evaluates a candidate's competency in machine learning model evaluation and metrics, sequence modeling trade-offs between transformers and RNNs, image distributional shift detection, and comparative understanding of optimization algorithms within the Machine Learning domain.
AmazonApplied ScientistMachine LearningOnsiteThis question tests the ability to synthesize concepts across LLM agent architecture, RAG systems, and reinforcement learning, focusing on architectural design trade-offs and decision-making logic.
AmazonApplied ScientistMachine LearningOnsiteThis interview question evaluates core ML concepts, assumptions, math intuition, training/evaluation trade-offs, and practical failure modes in a realistic interview setting. A strong answer for Explain Overfitting and Underfitting in Machine Learning states assumptions, handles edge cases, explains trade-offs, and shows how to validate the result clearly.
AmazonData ScientistMachine LearningTechnical ScreenThis question evaluates a candidate's understanding of parallelism and system-level performance in gradient-boosted tree implementations, covering concepts such as histogram-based split finding, sparse feature handling, cache-friendly data layouts, thread-level work partitioning, and multi-machine data-parallel synchronization.
AmazonApplied ScientistMachine LearningOnsiteThis question evaluates mastery of core Machine Learning concepts and diagnostics, covering statistical inference (p-values), overfitting/underfitting and bias-variance, causal inference approaches, encoding/decoding, optimization and backpropagation, gradient stability, handling imbalanced data, evaluation metrics versus accuracy, and experimental A/B testing. It is commonly asked in ML interviews to assess breadth and depth in the Machine Learning domain and probes both conceptual understanding and practical application by testing theoretical knowledge, recognition of common failure modes, and diagnostic reasoning used to validate models and experiments.
AmazonApplied ScientistMachine LearningTechnical ScreenAn Amazon machine-learning-engineer technical screen covering class-imbalance handling at the data, loss, and decision levels; leakage-aware validation; the bias–variance tradeoff; metric selection for rare-positive fraud data (PR-AUC, precision@k, calibration); and Transformer vs. CNN inductive biases. Includes worked numeric examples for cost-based thresholds and metric interpretation.
AmazonApplied ScientistMachine LearningTechnical ScreenThis interview question evaluates core ML concepts, assumptions, math intuition, training/evaluation trade-offs, and practical failure modes in a realistic interview setting. A strong answer for Explain key ML theory and techniques states assumptions, handles edge cases, explains trade-offs, and shows how to validate the result clearly.
AmazonApplied ScientistMachine LearningOnsiteThis question tests the conceptual understanding of model generalization trade-offs, the mathematical mechanics of regularization, and the practical application of modern LLM architectures.
AmazonSoftware EngineerMachine LearningTechnical ScreenThis question evaluates understanding of probabilistic information measures in language modeling, specifically the concept of surprisal, its units, and its relationship to evaluation metrics such as cross-entropy and perplexity within the Machine Learning/NLP domain.
AmazonApplied ScientistMachine LearningTechnical ScreenThis question evaluates understanding of regularization and feature selection in linear models, covering competencies in LASSO's L1 penalty versus L2, geometric constraint intuition, optimality/KKT conditions, effects of correlated predictors, the role of standardization, hyperparameter selection, and when Elastic Net is appropriate, within the Machine Learning domain for Data Scientist roles. It is commonly asked because it probes both conceptual understanding and practical application of model sparsity, interpretability, preprocessing, and bias–variance trade-offs, testing knowledge of statistical optimization and model selection rather than implementation details.
AmazonData ScientistMachine LearningTechnical ScreenEvaluate the architectural and practical evolution of NLP text embeddings, and formulate a decision framework comparing small bi-encoders versus large LLM-based embedding models based on retrieval quality, latency, and operational cost.
AmazonApplied ScientistMachine LearningOnsiteThis question tests your deep understanding of Group Relative Policy Optimization (GRPO) for post-training reasoning models, including how critic-free optimization alters training dynamics, distributed memory scaling, KV-cache efficiency via Multi-Head Latent Attention (MLA), and reward hacking mitigations.
AmazonApplied ScientistMachine LearningTechnical ScreenImplement the core of Group Relative Policy Optimization in a NumPy notebook: group-normalized advantages, the clipped objective with a KL penalty, per-token credit from process rewards, and shaped rewards. Tests fluency with array axes and broadcasting and an understanding of how reward normalization interacts with shaping terms.
AmazonMachine Learning EngineerMachine LearningOnsiteAmazon data scientist machine learning prompt on writing batch gradient descent pseudocode for linear regression, including MSE loss, vectorized gradients, parameter updates, stopping criteria, and implementation checks.
AmazonData ScientistMachine LearningTechnical ScreenImplement causal multi-head attention from scratch, add a key-value cache for token-by-token decoding, and extend the forward pass to grouped-query attention, where several query heads share one key and value head. Tests tensor shape discipline, masking with cached positions, and reasoning about decode compute and cache memory.
AmazonMachine Learning EngineerMachine LearningOnsiteThis question evaluates understanding of gradient-based optimization and linear model training, including competency in deriving gradients for mean squared error and implementing stochastic (or mini-batch) gradient descent for parameter estimation.
AmazonApplied ScientistMachine LearningTechnical ScreenThis problem tests your understanding of subword tokenization strategies and the theoretical motivation behind KL-divergence constraints during supervised fine-tuning.
AmazonApplied ScientistMachine LearningOnsiteThis question evaluates understanding of hyperparameter tuning methods, including the competencies to compare different search strategies, analyze trade-offs in efficiency and parallelization, and design experiments to optimize model performance.
AmazonApplied ScientistMachine LearningTechnical ScreenEvaluates email targeting and ranking strategy for a new Prime Video series launch. Strong answers cover data needs, cold start, ranking models, offline metrics, A/B testing, guardrails, and model selection.
AmazonData ScientistMachine LearningOnsiteEvaluates end-to-end predictive analytics project communication for a technical data-science interview. Strong answers cover business objectives, data, leakage controls, features, models, evaluation, errors, and impact.
AmazonData ScientistMachine LearningTechnical ScreenOptimize XGBoost for Predicting Marketing Outcomes evaluates core ML concepts, assumptions, math intuition, training/evaluation trade-offs, and practical failure modes in a realistic interview setting. A strong answer states assumptions, handles edge cases, explains trade-offs, and shows how to validate the result clearly.
AmazonData ScientistMachine LearningOnsiteThis question evaluates competency in statistical inference and multivariate analysis, covering hypothesis formulation about equal distributions, preprocessing and normalization considerations, choice of statistical tests or modeling approaches, and interpretation of significance and effect size.
AmazonApplied ScientistMachine LearningOnsiteThis question tests the ability to implement a simple linear regression predictor and analyze the reliability of extrapolating time-series data using basic statistical heuristics.
AmazonSoftware EngineerMachine LearningTechnical Screen