← All companies

Company

DeepMind

12 problems tagged for DeepMind.

Medium× 6Hard× 5Expert× 1How these tags are sourced

Implement 2D Positional Embeddings

Build 2D sinusoidal position encodings for vision transformers, encoding both row and column positions in patch grids.

Medium

Implement Beam Search for LLM Decoding

Maintain and expand top-scoring partial sequences at each decoding step with length normalization and early stopping.

Medium

Implement Contrastive Loss (InfoNCE) + CLIP Training Loop

Build the InfoNCE contrastive loss and a CLIP-style training loop that aligns image and text embeddings.

Medium

Implement Sliding Window Attention

Restrict attention to a fixed local window around each token, reducing memory from O(n²) to O(n·w) for long sequences.

Medium

Implement Top-k Sampling

Select the k highest-probability tokens, zero out the rest, renormalize, and sample for controlled text generation.

Medium

Implement Top-p (Nucleus) Sampling

Sort logits, compute cumulative probabilities, mask tokens below the nucleus threshold, and sample from the filtered distribution.

Medium

Implement DPO Loss from Scratch

Compute the Direct Preference Optimization loss that trains a policy directly from preference pairs without a reward model.

Hard

Implement Mixture of Experts Layer

Build a gated MoE with top-k routing, load balancing loss, and expert capacity constraints for sparse computation.

Hard

Implement PPO for RLHF

Build Proximal Policy Optimization with clipped surrogate objective, value function baseline, and KL penalty for RLHF.

Hard

Implement Selective State Space Model (Mamba)

Build Mamba's selective scan mechanism with input-dependent parameters, achieving linear-time sequence modeling without attention.

Hard

Implement Speculative Decoding

Draft tokens with a fast model, verify in parallel with the target model, and accept/reject to guarantee identical output distribution.

Hard

Implement GRPO (DeepSeek-R1 Algorithm)

Build Group Relative Policy Optimization that scores multiple completions per prompt and uses group-relative advantages.

Expert