Skip to content

JAX practice problems

Problem of the day: Implement ReLU
Difficulty:
Category & tags
Category:
Tag:

Showing 50 of 171

Title Difficulty
Accuracy – Multiclass Easy
Adapter Layer Medium
ALiBi-Style Distance Bias Medium
Apply Along an Axis Medium
Attention Block with RoPE Hard
Attention with Online Softmax Hard
AUC-ROC Medium
Autoencoder Medium
Batched Matrix Multiply Medium
Beam Search Hard
Bidirectional RNN Medium
Binary Classifier Easy
BLEU-1gram Medium
BPE Encode Text Hard
BPE Merge Step Medium
Broadcasting Addition Easy
Causal Attention Lab Medium
Causal Attention Mask Easy
Causal Attention with Framework Operations Hard
Causal LM Forward Pass Medium
Causal LM with KV Cache Generation Hard
Causal RoPE and SwiGLU Block Hard
Causal Self-Attention Block Medium
Classifier-Free Guidance Medium
Compute Gradient Medium
Compute Jacobian Hard
Concatenate Tensors Easy
Confusion Matrix Easy
Contrastive Loss (InfoNCE) Medium
Create a Tensor from a List Easy
Cross Attention Medium
Cross-Attention Block Medium
Cumulative Sum Easy
Custom Activation with Gradient Hard
Data Augmentation Medium
Data Collator with Padding Medium
DDPM Denoising Step Hard
DDPM Forward Noising Medium
DDPM Noise Schedule Easy
DDPM Sampler Loop Hard
DDPM Training Step End-to-End Medium
Decision Boundary Lab Medium
Denoising Autoencoder Lab Medium
Depthwise Separable Convolution Hard
Distributed Training Step End-to-End Hard
Einstein Summation Medium
Element-wise Operations Easy
Encoder-Decoder Beam Search Hard
Encoder-Decoder Greedy Decode Medium
Encoder-Decoder Transformer Forward Pass Hard