Katkat

practice a little every day, paws crossed ♡

Machine Learning Interview Questions

Practice 9403 coding, system design and machine learning questions from 546 companies, sorted by role, topic and interview round. Every question has step-by-step hints, a code runner that supports 8 languages and an AI mentor; members also get the full solution walkthrough.

9403
Questions
546
Companies
11
Topics
8
Languages

Showing 31–60 of 572 results

  • AmazonHardApplied Scientist
    Explain Collaborative Filtering Approaches

    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.

    Machine LearningOnsite
  • AmazonHardApplied Scientist
    Explain Core ML Interview Concepts

    This 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.

    Machine LearningTechnical Screen
  • AmazonMediumSoftware Engineer
    Explain Decoder-Only LLM Architecture and How vLLM Serves It Efficiently

    A 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.

    Machine LearningOnsite
  • AmazonMediumData Scientist
    Explain K-Fold Cross-Validation and Its Trade-Offs

    This 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.

    Machine LearningTechnical Screen
  • AmazonMediumApplied Scientist
    Explain LLM architecture, tuning, evaluation

    This 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.

    Machine LearningTechnical Screen
  • AmazonMediumApplied Scientist
    Explain Layer Normalization in Transformers

    This 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.

    Machine LearningOnsite
  • AmazonHardApplied Scientist
    Explain Logistic Regression Fundamentals

    This 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.

    Machine LearningOnsite
  • AmazonMediumApplied Scientist
    Explain ML evaluation, sequence models, and optimizers

    This 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.

    Machine LearningOnsite
  • AmazonHardApplied Scientist
    Explain NLP/RL concepts used in LLM agents

    This 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.

    Machine LearningOnsiteLLM-AgentsRAGReinforcement-Learning
  • AmazonMediumData Scientist
    Explain Overfitting and Underfitting in Machine Learning

    This 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.

    Machine LearningTechnical Screen
  • AmazonMediumApplied Scientist
    Explain XGBoost Parallelism Strategies

    This 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.

    Machine LearningOnsite
  • AmazonMediumApplied Scientist
    Explain core ML concepts and diagnostics

    This 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.

    Machine LearningTechnical Screen
  • AmazonHardApplied Scientist
    Explain imbalance, metrics, bias-variance, Transformers vs. CNNs

    An 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.

    Machine LearningTechnical Screen
  • AmazonHardApplied Scientist
    Explain key ML theory and techniques

    This 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.

    Machine LearningOnsite
  • AmazonMediumSoftware Engineer
    Explain overfitting, regularization, and LLM techniques

    This question tests the conceptual understanding of model generalization trade-offs, the mathematical mechanics of regularization, and the practical application of modern LLM architectures.

    Machine LearningTechnical ScreenRegularizationLLMRAG
  • AmazonMediumApplied Scientist
    Explain surprisal and its units

    This 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.

    Machine LearningTechnical Screen
  • AmazonMediumData Scientist
    Explain why LASSO selects features

    This 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.

    Machine LearningTechnical Screen
  • AmazonMediumApplied Scientist
    From Early NLP Embeddings to Modern LLM Embeddings

    Evaluate 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.

    Machine LearningOnsiteEmbeddingsNLPInformation Retrieval
  • AmazonHardApplied Scientist
    GRPO Deep Dive: Critic-Free RL, Parallelism, MLA, and Reward Design for a Reasoning LLM

    This 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.

    Machine LearningTechnical ScreenReinforcement LearningLLM Post-TrainingGRPO
  • AmazonMediumMachine Learning Engineer
    GRPO in NumPy: Group-Relative Advantages, Process Rewards and Shaped Rewards

    Implement 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.

    Machine LearningOnsite
  • AmazonEasyData Scientist
    Implement Batch Gradient Descent for Linear Regression

    Amazon 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.

    Machine LearningTechnical Screen
  • AmazonMediumMachine Learning Engineer
    Implement Multi-Head Attention with a KV Cache and Cached Grouped-Query Attention

    Implement 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.

    Machine LearningOnsite
  • AmazonMediumApplied Scientist
    Implement SGD for linear regression and derive gradients

    This 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.

    Machine LearningTechnical Screen
  • AmazonMediumApplied Scientist
    LLM Fundamentals: Tokenization Design and KL-Regularized SFT

    This problem tests your understanding of subword tokenization strategies and the theoretical motivation behind KL-divergence constraints during supervised fine-tuning.

    Machine LearningOnsiteLLMTokenizationSFT
  • AmazonEasyApplied Scientist
    List hyperparameter tuning methods

    This 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.

    Machine LearningTechnical Screen
  • AmazonMediumData Scientist
    Optimize Email Strategy for New Prime Video Series Launch

    Evaluates 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.

    Machine LearningOnsite
  • AmazonMediumData Scientist
    Optimize Predictive Analytics: Feature Engineering to Model Evaluation

    Evaluates 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.

    Machine LearningTechnical Screen
  • AmazonMediumData Scientist
    Optimize XGBoost for Predicting Marketing Outcomes

    Optimize 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.

    Machine LearningOnsite
  • AmazonMediumApplied Scientist
    Test whether two user populations differ

    This 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.

    Machine LearningOnsite
  • AmazonHardSoftware Engineer
    Use a Fitted Line to Predict a Future Data Point

    This question tests the ability to implement a simple linear regression predictor and analyze the reliability of extrapolating time-series data using basic statistical heuristics.

    Machine LearningTechnical ScreenRegressionTime SeriesStatistics