Queries, Keys, and Values

English Version jax
1

https://d2l.ai/chapter_attention-mechanisms-and-transformers/queries-keys-values.html