Interview questions

Implement Gradient Descent and Explain Adam Updates

Companies: Akuna Capital

ML Coding · ML Fundamentals

Implement Gradient Descent Regression

Companies: Databricks

Implement Grouped-Query Attention (GQA)

Companies: Datadog

ML Coding · Transformers & Attention

Implement K-means and handle train-inference mismatch

Companies: Waymo

ML Coding · Unsupervised Learning

Implement K-Means Without Numerical Libraries

Companies: Lyft

Implement Linear Regression Gradient Descent

Companies: Databricks

Implement Linear Regression with Batch Gradient Descent from Scratch

Companies: Pinterest

ML Coding · ML Fundamentals

Implement Masked Multi-Head Self-Attention

Companies: Apple

ML Coding · Transformers & Attention

Implement MSE Loss and Linear Regression from Scratch

Companies: Indeed

ML Coding · ML Fundamentals

Implement Multi-Head Attention from Scratch

Companies: ByteDance

ML Coding · Transformers & Attention

Implement Multi-Head Attention from Tensor Operations

Companies: ByteDance

ML Coding · Transformers & Attention

Implement Multi-Head Self-Attention

Companies: Uber

ML Coding · Transformers & Attention

Implement robust k-means from scratch

Companies: Microsoft

Implement Softmax Cross-Entropy Forward and Backward Passes and a Training Loop

Companies: Waymo

ML Coding · Neural Networks & Training

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

Companies: Amazon

ML Coding · Transformers & Attention

Improve low R² without p‑hacking

Companies: Instacart

Feature Engineering · Regression & Statistical Modeling

Improve Model Generalization with Cross-Validation and Feature Engineering

Companies: Boston Consulting Group

Inferring Co-Viewing on YouTube: When Several People Watch One Screen

Companies: Google

AI Product Management · ML System Design

Interpret AUC Values and Handle Class Imbalance Techniques

Companies: Boston Consulting Group

iOS vs. Android Usage Gap: Modeling, Causality, Telemetry, Missing Data, and Segmented Actions Context You observe that Instagram usage is substantial...

General practice

Justify Using LLMs for Reporting

Companies: Natoora

K-means Clustering: Concepts, Initialization, Model Selection, Preprocessing, and Business Validation Context: You are clustering customer data with n...

General practice