numpy Interview Questions
29 interview questions in our bank cover numpy, most of them Coding & Leetcode-style Questions. They average 3.2/5 difficulty — medium — and each one was reported by a candidate after a real interview. Companies known to ask about numpy: Tesla, Apple, Datadog, Waymo, Capital One, and 15 more.
Practice these on the problems board →Companies that ask about numpy
Question mix
- Coding & Leetcode-style Questions18
- ML Fundamentals & Algorithms8
- Research & Paper Understanding1
- LLMs & Prompt Engineering1
- Deep Learning & Architectures1
Difficulty
- 3/5 — medium23
- 4/5 — hard6
Questions tagged numpy
ML Take-Home: 4-Hour Experiment plus Live Review
4/5Prepare for a rigorous research-oriented take-home assessment reported from Anthropic that tests your practical machine learning expertise under strict time limits. Candidates must independently investigate a structured modeling challenge, build functional code solutions, and interpret numerical findings effectively. The follow-up discussion evaluates your experimental methodology, communication clarity, and ability to defend technical decisions. The full problem and model solution require a subscription.
Research & Paper UnderstandingAnthropicNumPy Puzzle: 1-NN Vectorization → Wx+b Network
3/5Master high-performance matrix operations and spatial classification techniques inspired by coding evaluations at OpenAI. This exercise challenges practitioners to implement distance-based prediction logic entirely through vectorized array manipulations, avoiding standard iteration loops, before translating the identical procedure into a neural network forward propagation framework. Learn how to leverage advanced mathematical libraries for optimal efficiency by accessing the full problem breakdown and expert solution.
Coding & Leetcode-style QuestionsOpenAIDistributed Matrix Multiplication — DP and FSDP
4/5This advanced infrastructure coding assessment from xAI explores the principles of distributed model training by simulating device communication for matrix operations. Candidates must build both data-parallel and fully sharded strategies to handle heavy computational workloads across multiple nodes effectively. To unlock the full problem description, architectural insights, and the reference code, a subscription is required.
Coding & Leetcode-style QuestionsxAIImplement an Image Filter from Scratch
3/5This foundational coding interview question reported at Apple requires building a two-dimensional convolution operator completely from scratch using only standard library constructs. Candidates must implement custom zero-padding, kernel transformations, and nested iteration loops to support image processing primitives like blurring and edge detection without external scientific libraries. Mastering this task proves deep comprehension of spatial filtering mechanics. Unlock the complete specification and optimal Python implementation by subscribing.
Coding & Leetcode-style QuestionsAppleLinear Regression via Gradient Descent (MSE)
3/5Practice fundamental machine learning implementation with this core optimization exercise reported from Databricks hiring rounds. You are asked to construct an iterative parameter estimation algorithm from scratch using standard loss metrics and gradient update rules without relying on pre-built machine learning libraries. This task gauges your mathematical intuition regarding convergence behavior, learning rate tuning, and gradient debugging. View the complete problem breakdown and optimal mathematical solution by securing a subscription.
Coding & Leetcode-style QuestionsDatabricksImplement Binary Focal Loss
3/5This machine learning challenge, commonly asked in interviews at Datadog, focuses on implementing the Binary Focal Loss function to help models handle difficult classification tasks with class imbalance. You will practice handling probability clipping, mathematical transformations, and various reduction strategies to compute per-sample penalties accurately. To access the complete problem requirements, mathematical formulas, and the full model solution, a paid subscription is required.
ML Fundamentals & AlgorithmsDatadogML Coding: 2-D Convolution, Decaying Attention, Training Loop
3/5Sharpen your deep learning implementation skills with this practical coding challenge featured in NVIDIA interview loops. You will write core neural network components from scratch, including a custom two-dimensional convolution in NumPy, a decayed attention mechanism, and a complete training loop utilizing standard machine learning frameworks. This exercise evaluates your fluency with tensor operations, gradient descent workflows, and fundamental deep learning mechanics. Gain access to the full problem breakdown and clean implementation code with an active subscription.
Coding & Leetcode-style QuestionsNVIDIAHand-Code Self-Attention and Cross-Entropy
4/5Conquer a rigorous machine learning coding challenge frequently seen in ByteDance interview processes by implementing core deep learning components from scratch. You will be tasked with writing scaled dot-product attention in NumPy and deriving the binary cross-entropy loss function mathematically, alongside oral questions on architectural layers. This exercise examines your mathematical rigor and ability to translate theoretical formulations into stable vector operations. Access the comprehensive problem breakdown and ideal reference implementation with a subscription.
Coding & Leetcode-style QuestionsByteDanceML Coding — Handwritten MHA and Sparse Matmul
3/5Master core machine learning implementation tasks frequently encountered in technical interviews for artificial intelligence roles at Oracle. This challenge requires building foundational neural network components from scratch using fundamental tensor operations, specifically focusing on custom attention mechanisms and optimized sparse matrix multiplication without relying on high-level framework abstractions. Unlock the complete technical walkthrough, architectural insights, and production-ready reference solution by upgrading to a paid subscription.
ML Fundamentals & AlgorithmsOracleImplement top-k sampling using NumPy
3/5This Cohere interview question tests your proficiency in numerical computing and probabilistic sampling techniques commonly used in modern language models. You are asked to implement a token selection mechanism that restricts generation choices to the highest-scoring candidates using pure array operations without heavy machine learning frameworks. The task evaluates your mastery of vector manipulation, probability normalization, and efficient indexing strategies. To access the complete problem statement, performance considerations, and clean model implementation, a subscription is needed.
ML Fundamentals & AlgorithmsCohereMLE CodeSignal OA (MCQ + NN + Classical-ML Coding)
3/5This Coinbase machine learning engineer assessment bundle covers a mix of conceptual multiple-choice questions, a neural network forward-pass calculation, and foundational coding tasks such as implementing classical algorithms from scratch. It tests your theoretical understanding of model evaluation metrics, loss functions, ensemble techniques, and dimensionality reduction alongside practical coding fluency. To explore the full question bank, detailed test scenarios, and expert solutions, a subscription is required.
ML Fundamentals & AlgorithmsCoinbaseFollow-up: Implement Gaussian Mixture Model (GMM) With EM
4/5This machine learning challenge, reported during a technical interview at Tubi, requires candidates to build and optimize a Gaussian Mixture Model from scratch using the Expectation-Maximization framework. Applicants need to initialize parameters effectively and iteratively update mixing coefficients, means, and covariance matrices to capture underlying data distributions. The assessment evaluates a deep understanding of probabilistic clustering and latent variable models. Access to the comprehensive problem description and optimal code solution requires a paid subscription.
ML Fundamentals & AlgorithmsTubiImplement Grouped Query Attention Forward Pass
4/5In modern machine learning engineering interviews reported by Datadog, implementing core transformer mechanisms from scratch is a crucial test of tensor manipulation skills. This challenge focuses on building the forward computation for a specialized attention variant where multiple query projections share designated key and value heads to optimize memory and compute overhead. Candidates must efficiently manage high-dimensional tensor shapes and apply scaling equations without relying on high-level wrapper libraries. Discovering the optimal vectorization strategy and reviewing the complete production-ready solution requires a full subscription.
ML Fundamentals & AlgorithmsDatadogImplement Dense Layer with Numpy
3/5Understanding the mathematical mechanics behind neural network layers is essential for machine learning engineers, as explored in this interview question reported by Etsy. You are asked to manually construct a fully connected layer using NumPy, implementing both forward propagation and gradient backpropagation alongside a custom activation function. To explore the complete implementation guidelines and fully worked model solution, a paid subscription is required.
ML Fundamentals & AlgorithmsEtsyVectorized K-Means in NumPy or PyTorch
3/5Sharpen your machine learning fundamentals with this Apple-inspired coding challenge that requires implementing K-Means clustering entirely from scratch using only standard libraries. You will write a vectorized version of Lloyd's algorithm to iteratively update cluster centroids and assign data points using squared Euclidean distance metrics. This task evaluates your numerical programming abilities, matrix operations efficiency, and understanding of unsupervised learning mathematics. Unlock the complete specification, optimization tips, and reference code through our paid subscription.
Coding & Leetcode-style QuestionsAppleHand-Written K-Means
3/5Frequently asked during Microsoft screening rounds, this programming challenge requires candidates to build the unsupervised clustering algorithm completely from scratch using standard Python libraries or numerical computing frameworks. You will need to implement iterative centroid updates and cluster assignment logic while demonstrating a strong grasp of underlying operational complexities. Preparing for this exercise ensures you can articulate foundational machine learning concepts fluidly during live technical evaluations. Access the complete breakdown and verified implementation by purchasing a subscription.
Coding & Leetcode-style QuestionsMicrosoftNumPy Top-p Sampling and Multi-Head Attention
3/5Implement foundational machine learning operations from scratch in this Scale AI interview question focusing on tensor manipulation and probabilistic sampling. Working within a notebook environment, you are expected to construct core architectural primitives and sampling strategies efficiently using numerical computing libraries. This problem assesses your fluency with vectorization, probability distributions, and fundamental transformer mechanics without relying on high-level framework abstractions. Discover the complete problem walkthrough and optimal solution by unlocking your subscription.
Coding & Leetcode-style QuestionsScale AICompute Euclidean Distance Between Two Tensors
3/5Sourced from XPeng, this machine learning fundamentals problem requires calculating pairwise Euclidean distances across high-dimensional batched tensor inputs without relying on standard slow iteration loops. You will need to leverage vectorization techniques and matrix operations to produce an output tensor of specified shape efficiently. This exercise evaluates your deep understanding of tensor manipulation and linear algebra operations in deep learning frameworks. The full problem statement and optimized code solution are available exclusively with a paid subscription.
ML Fundamentals & AlgorithmsXPengHand-Write K-Means Clustering (NumPy)
3/5Practice implementing an unsupervised clustering algorithm entirely from scratch using only array manipulation libraries, avoiding high-level machine learning frameworks. As a staple of core machine learning interviews at Waymo, this task assesses your ability to translate mathematical definitions of distance and centroid updates into highly vectorized, efficient code without relying on slow iterative loops. You will focus on optimizing broadcasting operations to handle multi-dimensional data swiftly and accurately. Gain immediate access to the complete problem breakdown and optimal reference code by purchasing a subscription.
Coding & Leetcode-style QuestionsWaymoNumPy Conv2D Forward and Parameter Count
3/5This reported Tesla interview question challenges you to build a foundational Conv2D forward pass from scratch using only NumPy, bypassing standard deep-learning libraries. You will need to calculate output tensor dimensions, figure out parameter counts including bias terms, and optimize your code beyond slow nested loops through effective vectorization techniques. Master this essential tensor manipulation exercise to strengthen your grasp of convolutional neural network internals. The comprehensive problem statement and complete model solution require a subscription.
Coding & Leetcode-style QuestionsTeslaStandardize Each Column of a 2D Tensor
3/5In this numerical computing challenge reported during Bridgewater interviews, programmers are asked to normalize a two-dimensional tensor column by column using statistical z-scores. The task requires calculating population standard deviations and means efficiently to handle large dataset matrices without performance bottlenecks. This exercise tests your familiarity with tensor manipulation libraries, vectorized operations, and numerical stability. Unlock the full problem details, mathematical guidelines, and a high-performance model solution by purchasing a subscription.
Coding & Leetcode-style QuestionsBridgewaterImplement Top-P (Nucleus) Sampling
3/5Implementing modern generative artificial intelligence utilities requires precise probabilistic modeling, as demonstrated in this reported Capital One interview question focused on prompt engineering and sampling strategies. The challenge involves processing vocabulary logits numerically in a stable manner, sorting cumulative probabilities, and isolating a dynamic subset for random token selection. Candidates must apply robust mathematical transformations while guarding against floating-point underflow or overflow issues. This scenario evaluates your capability to bridge theoretical machine learning concepts with production-ready code. Unlocking the complete problem details and model solution requires a subscription.
LLMs & Prompt EngineeringCapital OneML Coding: Implement a Simple Attention / Layer Forward Pass
3/5This machine learning engineering exercise, frequently asked during Google technical interviews, requires you to construct the forward pass mechanism for a fundamental neural network component from scratch. Working directly with raw tensors and weight matrices, you must apply core mathematical operations to replicate behaviors like scaled dot-product attention or standard linear transformations. The task assesses your deep understanding of tensor manipulation, gradient-friendly architecture design, and numerical computation primitives. To view the full problem specifications and clean, production-grade model implementation, subscribe today.
Deep Learning & ArchitecturesGoogleLinear Regression from Scratch in NumPy
3/5Reported at Capital One, this machine learning coding exercise asks you to construct a foundational predictive model from scratch without relying on external data science libraries. You will implement both analytical matrix solutions and iterative optimization techniques to fit linear data accurately. To examine the complete instructions, mathematical prerequisites, and clean model code, a subscription is required.
Coding & Leetcode-style QuestionsCapital OneCompute EMA Indicators and Detect Crossovers in Pandas
3/5In this Squarepoint quantitative engineering interview problem, candidates must process financial time-series data using native library capabilities without resorting to standard iteration loops. The task involves calculating dual moving averages across different time horizons and pinpointing precise intersection moments where the indicators cross paths. Success relies on writing highly optimized, vectorized expressions that execute efficiently over large datasets. Unlock the complete evaluation guidelines, constraints, and an expert pandas implementation with a subscription.
Coding & Leetcode-style QuestionsSquarepointBatched Trajectory Suffix Sum
3/5Evaluating tensor trajectories efficiently is a critical algorithmic task in autonomous driving development, as seen in AI foundation interviews at Tesla. This problem requires candidates to process multidimensional coordinate arrays and compute cumulative suffix metrics across sequential waypoints for multiple batches simultaneously. Success hinges on avoiding explicit looping constructs through clever vectorization and reverse accumulation techniques. To review the optimal tensor operation approach and complete reference code, unlock full access with a subscription.
Coding & Leetcode-style QuestionsTeslaML Coding from Scratch: Regression, Markov, Facility Location
4/5During this Uber machine learning onsite interview, candidates must build core modeling primitives entirely from scratch using only NumPy. The evaluation focuses on mathematical derivation, optimization mechanics, and fluent coding without relying on high-level framework abstractions. Engineers are tested on fundamental concepts underlying regression and probability models. Unlock the complete technical walkthrough, derivation notes, and production-grade solution by purchasing a subscription.
Coding & Leetcode-style QuestionsUberSpeed-Limit RL Reward from Trajectory Samples
3/5This reinforcement learning simulation task, reported from Tesla interviews, asks you to design a custom reward function that penalizes speed limit infractions using raw trajectory data sampled at high frequencies. You will reason about trajectory tensors, evaluate various penalty formulations based on duration and magnitude of violations, and adapt to dynamic speed limit changes over time. Get full access to the complete problem specification and reference code with a subscription.
Coding & Leetcode-style QuestionsTeslaML Coding: Debug a NumPy / Tensor Framework
3/5Test your ability to troubleshoot and repair subtle defects in a custom tensor computation library during this mock engineering interview inspired by Waymo. This hands-on debugging challenge examines your understanding of memory aliasing, dimension reduction semantics, and numeric type conversions within custom array frameworks. Quick diagnosis and clear explanations are key to succeeding in this evaluation. Unlock the full troubleshooting guide, bug list, and corrected implementation by subscribing today.
Coding & Leetcode-style QuestionsWaymo
Studied alongside
numpy interview FAQ
- How many numpy interview questions are there?
- 29 reported questions, mostly Coding & Leetcode-style Questions.
- Which companies ask numpy questions?
- Tesla (3), Apple (2), Datadog (2), Waymo (2), Capital One (2), Anthropic (1), OpenAI (1), xAI (1).
- How hard are numpy questions?
- They average 3.2 out of 5: 23 at 3/5, 6 at 4/5.