Machine Learning Engineer Machine Learning Interview Questions

Practice 217 real Machine Learning interview questions for Machine Learning Engineer roles. From companies including Amazon, OpenAI, Snapchat, Pinterest, TikTok.

217 Questions 73 Companies07.25.2026
Showing 20 results
Role
OpenAI logo
OpenAI
Medium
Machine Learning Engineer Locked

Implement Masked Cross-Entropy with Label Smoothing

Implement numerically stable masked cross-entropy in NumPy with label smoothing and optional temperature scaling. Validate shapes, dtypes, finite logi...

Machine Learning
65
1
558 people solved
Jun 27, 2026
Meta logo
Meta
Hard
Machine Learning Engineer Locked

Self-Attention: Implementation, Complexity, and Efficient Variants

This question evaluates a machine learning candidate's understanding of the self-attention mechanism, including its implementation, computational comp...

Machine Learning
40
0
344 people solved
Jun 27, 2026
OpenAI logo
OpenAI
Medium
Machine Learning Engineer Locked

Implement 1NN with NumPy

This question evaluates implementing a 1-nearest-neighbor classifier with NumPy, testing skills in vectorized numerical computation, distance metrics,...

Machine Learning
287
3
2722 people solved
May 19, 2026
Spacex logo
Spacex
Medium
Machine Learning EngineerSenior+

Diagnose a Production Classifier and Defend Its Systems Design

Diagnose a Production Classifier and Defend Its Systems Design You own a binary classifier used by a downstream production service. The positive class...

Machine Learning
2
0
18 people solved
Jul 25, 2026
Pinterest logo
Pinterest
Hard
Machine Learning Engineer

Prevent Vanishing Gradients in Deep Networks

Prevent Vanishing Gradients in Deep Networks Clarifying Questions to Ask - Are we discussing feed-forward networks, recurrent networks, or both? - Sho...

Machine Learning
11
0
87 people solved
Jul 3, 2026
OpenAI logo
OpenAI
Medium
Machine Learning Engineer

Compute entropy and implement 1-NN

You are given two short ML coding problems from a machine-learning engineer screen. Both are implementation-focused but probe whether you understand t...

Machine Learning
121
1
1152 people solved
Apr 24, 2026
OpenAI logo
OpenAI
Medium
Machine Learning Engineer

Derive Sharded Matrix Multiplication and Backpropagation

Derive Sharded Matrix Multiplication and Backpropagation Consider Y = X @ W, where X has shape (B, D) and W has shape (D, H). Training runs on P devic...

Machine Learning
27
0
289 people solved
Apr 19, 2026
Bridgewater logo
Bridgewater
Medium
Machine Learning Engineer

Diagnose Strong Validation and Weak Production Performance

Diagnose Strong Validation and Weak Production Performance You are given a training repository for a model whose validation metric looks strong while ...

Machine Learning
8
0
63 people solved
Jul 8, 2026
Mistral AI logo
Mistral AI
Medium
Machine Learning Engineer

Answer an LLM Systems Quick-Fire Interview

You are given a rapid-fire LLM systems interview. Be prepared to explain Transformer architecture, attention and masking, multi-head attention, normal...

Machine Learning
17
0
141 people solved
Jul 2, 2026
Lyft logo
Lyft
Medium
Machine Learning Engineer

Reason About an Intercept Shift in Logistic Regression

Reason About an Intercept Shift in Logistic Regression A trained binary classifier produces p(x) = sigmoid(w^T x + b) and predicts the positive class ...

Machine Learning
9
0
64 people solved
Jul 4, 2026
Amazon logo
Amazon
Medium
Machine Learning Engineer

Implement Stable Sigmoid, Softmax, and Scaled Dot-Product Attention

Implement Stable Sigmoid, Softmax, and Scaled Dot-Product Attention Implement the mathematical core of three common neural-network operations without ...

Machine Learning
4
0
52 people solved
Jul 2, 2026
OpenAI logo
OpenAI
Hard
Machine Learning Engineer

Improve Training With Noisy Annotators

You are given a labeled training dataset as a Pandas DataFrame. Each row contains feature columns, an observed label, and an annotator_id identifying ...

Machine Learning
90
0
781 people solved
Apr 2, 2026
Meta logo
Meta
Hard
Machine Learning Engineer Locked

Machine Learning Fundamentals: Optimizers, Scaling Laws, and Clustering

This question evaluates conceptual grasp of core machine learning fundamentals: gradient-based optimizers, neural scaling laws, and unsupervised clust...

Machine Learning
11
0
147 people solved
Jun 27, 2026
Amazon logo
Amazon
Hard
Machine Learning EngineerNew Grad

Explain Core ML Interview Concepts

You are in a phone screen for an applied scientist / machine-learning engineer role and are asked to verbally explain a set of machine-learning fundam...

Machine Learning
235
0
1922 people solved
Apr 27, 2026
Mercor logo
Mercor
Medium
Machine Learning Engineer

Review a Compact GPT-Style Transformer Implementation

Review a Compact GPT-Style Transformer Implementation You are reviewing a compact decoder-only Transformer implementation intended for next-token lang...

Machine Learning
2
0
40 people solved
Jul 8, 2026
Mistral AI logo
Mistral AI
Hard
Machine Learning Engineer

Explain Transformer Internals and Implement Top-p Decoding

You are interviewing for an AI Scientist role. Explain the Transformer architecture in detail, including attention, positional encoding, encoder versu...

Machine Learning
13
0
94 people solved
Jul 2, 2026
OpenAI logo
OpenAI
Hard
Machine Learning Engineer

Improve classifier with noisy multi-annotator labels

Problem You are given a text dataset for a binary classification task (label in $\{0,1\\}$). Each example has been labeled by multiple human annotator...

Machine Learning
905
3
6117 people solved
Feb 11, 2026
Lyft logo
Lyft
Medium
Machine Learning Engineer

Implement K-Means Without Numerical Libraries

Implement K-Means Without Numerical Libraries Implement K-means clustering for a list of finite numeric points using only core language features and e...

Machine Learning
3
0
37 people solved
Jul 4, 2026
Amazon logo
Amazon
Hard
Machine Learning Engineer Locked

GRPO Deep Dive: Critic-Free RL, Parallelism, MLA, and Reward Design for a Reasoning LLM

This question assesses understanding of reinforcement learning algorithms used to post-train large reasoning language models, including critic-free po...

Machine Learning
8
0
98 people solved
Jun 21, 2026
OpenAI logo
OpenAI
Medium
Machine Learning Engineer

Debug MiniGPT and Backpropagate Matmul

This is a hands-on PyTorch screen with two independent tasks. You share a code editor with the interviewer and are expected to run the code, read trac...

Machine Learning
106
0
1062 people solved
Apr 3, 2026

Explore more Machine Learning Engineer Machine Learning interview questions

Real questions from candidate reports, grouped by company, topic and role.

By company
Other categories for Machine Learning Engineer
Machine Learning questions for other roles
Browse all