← All companies

Company

Apple

6 problems tagged for Apple.

Easy× 1Medium× 3Hard× 2How these tags are sourced

Implement Softmax from Scratch

Build numerically stable softmax using the log-sum-exp trick, handling overflow and underflow in raw tensor math.

Easy

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 Knowledge Distillation

Train a smaller student model to match a larger teacher's soft predictions using temperature-scaled KL divergence loss.

Medium

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 Vision Transformer + MAE Pretraining

Build ViT with masked autoencoder pretraining — randomly mask patches, encode visible ones, decode to reconstruct.

Hard