ML Coding & PyTorch interview questions.
Practice 27 high-signal questions covering the concepts and decisions interviewers probe in ML Coding & PyTorch.
Implement Byte Pair Encoding.ML Coding & PyTorch · hard · Evidence 77/100Sample From a Non-Uniform RNGML Coding & PyTorch · hard · Evidence 77/100Design an efficient sparse-matrix representation.ML Coding & PyTorch · medium · Evidence 77/100Implement self-attention.ML Coding & PyTorch · hard · Evidence 77/100Implement rejection samplingML Coding & PyTorch · medium · Evidence 77/100Implement masked self-attention.ML Coding & PyTorch · hard · Evidence 77/100Implement the transpose of a sparse matrix efficiently.ML Coding & PyTorch · medium · Evidence 77/100How would you vectorize a K-means implementation?ML Coding & PyTorch · hard · Evidence 77/100Implement K-means from scratch.ML Coding & PyTorch · medium · Evidence 77/100Implement single-head attention in PyTorch.ML Coding & PyTorch · hard · Evidence 75/100Implement KL divergence in Python.ML Coding & PyTorch · medium · Evidence 75/100Add a causal mask that hides future tokens.ML Coding & PyTorch · hard · Evidence 75/100Implement self-attention from scratch in code.ML Coding & PyTorch · hard · Evidence 75/100Implement multi-head attention in PyTorch.ML Coding & PyTorch · hard · Evidence 75/100Implement a Categorical SamplerML Coding & PyTorch · medium · Evidence 66/100Implement Cross-Entropy Loss From ScratchML Coding & PyTorch · medium · Evidence 41/100Large-Scale Weighted Random SamplingML Coding & PyTorch · hard · Evidence 41/100Implement a Contrastive Learning LossML Coding & PyTorch · hard · Evidence 41/100Transformation + Sorting AI Coding ProblemML Coding & PyTorch · medium · Evidence 41/100How would you reduce memory movement in a custom CUDA inference kernel?ML Coding & PyTorch · hard · Evidence 41/100What bottlenecks would you optimize in a CUDA kernel used for model inference?ML Coding & PyTorch · hard · Evidence 41/100How would you optimize a deep-learning operation for GPU execution?ML Coding & PyTorch · hard · Evidence 40/100How do CPU and GPU architecture change ML data movement?ML Coding & PyTorch · hard · Evidence 40/100Implement and Optimize a Custom Neural Network LayerML Coding & PyTorch · hard · Evidence 34/100Implement a Transformer BlockML Coding & PyTorch · hard · Evidence 34/100Code Multi-Head Self-AttentionML Coding & PyTorch · hard · Evidence 34/100Implement a Transformer encoder layerML Coding & PyTorch · medium · Evidence 31/100