Machine Learning Engineer Data Manipulation (SQL/Python) Interview Questions

Practice 12 real Data Manipulation (SQL/Python) interview questions for Machine Learning Engineer roles. From companies including DRW, OpenAI, Tesla, Google, Meta.

12 Questions 9 Companies07.02.2026
Showing 12 results
Role
Meta logo
Meta
Hard
Machine Learning Engineer Locked

Find A Low-Quality Annotator From Label Data

Practice a pandas-style data analysis prompt for identifying a low-quality annotator from label data. The question emphasizes cleaning, agreement or g...

Data Manipulation (SQL/Python)
4
0
57 people solved
Jun 2, 2026
Scale AI logo
Scale AI
Medium
Machine Learning Engineer

Debug ML pipeline and build text parser

You are in a hands-on, hour-long ML-engineering working session (Scale AI, Machine Learning Engineer loop). You are given a small ML project — data lo...

Data Manipulation (SQL/Python)
42
0
539 people solved
Sep 6, 2025
Spotify logo
Spotify
Easy
Machine Learning Engineer

Query Top Played Tracks Globally And By Country

Given a play_events table with columns user_id, track_id, timestamp, and country, write SQL to return the top N most played tracks globally and the to...

Data Manipulation (SQL/Python)
8
4
143 people solved
Jul 2, 2026
OpenAI logo
OpenAI
Medium
Machine Learning Engineer

Train and analyze a classifier

You are given a labeled dataset for binary classification. Implement an end-to-end Python solution that trains a classifier and analyzes it to a produ...

Data Manipulation (SQL/Python)
27
0
532 people solved
Jul 31, 2025
OpenAI logo
OpenAI
Medium
Machine Learning Engineer

Implement vectorized NumPy ops and explain broadcasting

Implement vectorized NumPy code for: (a) computing pairwise cosine similarity between two real-valued matrices X (shape n×d) and Y (shape m×d) without...

Data Manipulation (SQL/Python)
23
0
331 people solved
Aug 11, 2025
Google logo
Google
Medium
Machine Learning Engineer

Implement a robust Python generator

Given a list of integers, write a Python generator that yields the integers from the list while handling edge cases such as None values, empty input, ...

Data Manipulation (SQL/Python)
9
0
96 people solved
Sep 6, 2025
DRW logo
DRW
Medium
Machine Learning Engineer

Implement simulation-based portfolio optimizer in Python

Implement simulation-based portfolio optimizer in Python Given a pandas DataFrame 'returns' of daily asset returns (index: dates; columns: tickers) an...

Data Manipulation (SQL/Python)
3
0
53 people solved
Jul 31, 2025
Netflix logo
Netflix
Medium
Machine Learning Engineer

Transform flat keys into nested dictionary

You are given a flat collection of parameter keys like ['layer1.attention.q_proj.weight', 'layer1.attention.k_proj.weight', 'layer1.mlp.fc1.weight', ....

Data Manipulation (SQL/Python)
7
0
119 people solved
Aug 13, 2025
Tesla logo
Tesla
Medium
Machine Learning Engineer

Implement and vectorize NumPy Conv2D

Implement a 2D convolution operation from scratch using NumPy only (no TensorFlow or PyTorch). Assume NCHW input shape (N, C_in, H_in, W_in) and weigh...

Data Manipulation (SQL/Python)
0
0
9 people solved
Sep 6, 2025
DRW logo
DRW
Medium
Machine Learning Engineer

Process CSV for portfolio returns and metrics

Given one or more CSV files containing daily asset prices or returns and optional portfolio weights, write Python (pandas) code to: a) load, clean, an...

Data Manipulation (SQL/Python)
0
0
9 people solved
Jul 29, 2025
Reddit logo
Reddit
Medium
Machine Learning Engineer

Load and prepare JSON for modeling

Using Python in a Jupyter notebook, load a JSON dataset with fields: ( 1) hours spent reading A posts (float), ( 2) hours spent reading B posts (float...

Data Manipulation (SQL/Python)
0
0
6 people solved
Sep 6, 2025
Tesla logo
Tesla
Medium
Machine Learning Engineer

Compute nearest index within threshold after walking distances

You are given: ( 1) points: a list of N 2D coordinates in miles, points[i] = [x_i, y_i], ordered; ( 2) distances: a list of M nonnegative floats (mile...

Data Manipulation (SQL/Python)
0
0
6 people solved
Aug 12, 2025

Explore more Machine Learning Engineer Data Manipulation (SQL/Python) interview questions

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

Other categories for Machine Learning Engineer
Data Manipulation (SQL/Python) questions for other roles
Browse all