Skip to content
Cracked
AI
Problems
Tracks
Learn JAX
Roadmap
Articles
Leaderboard
Search
Ctrl K
Mode:
JAX
PyTorch
JAX
Python
Rust
C++
PyTorch
JAX
Python
Rust
C++
Log in
Sign up
Problems
Tracks
Learn JAX
Articles
Roadmap
Leaderboard
Set up the runtime
Log in
Sign up
PyTorch
JAX
Python
Rust
C++
Radio
We can't find the internet
Attempting to reconnect
Something went wrong!
Attempting to reconnect
Vmap practice problems in JAX
Search problems
Difficulty:
Easy
Medium
Hard
Category:
Primitives
Framework
End-to-End
Research
Tag:
vmap
transformer
training
classification
attention
basics
cnn
generative
activation
metrics
embedding
diffusion
llm
loss
normalization
optimization
causal-lm
ddpm
decoding
neural-network
regression
regularization
vit
bert
decoder
fine-tuning
inference
linear-algebra
mlm
nlp
position-encoding
preprocessing
rnn
seq2seq
sequence-model
softmax
tokenization
autograd
convolution
cross-attention
encoder-decoder
gpt
jax.grad
parameter-efficient
recurrent
rope
self-attention
sgd
training-loop
vae
adam
beam-search
bpe
broadcasting
calibration
distributed
efficiency
ema
feature-engineering
gates
gating
Show more tags
Sort:
Title
Easiest first
Hardest first
Recently added
Clear filters
Showing 1 of 1
Status
Title
Difficulty
Category
Tags
Parallel Map with vmap
Hard
Framework
vmap
per-sample-gradient
jax.vmap
jax.grad
torch.vmap