#

multi-query-attention

Here are 2 public repositories matching this topic...

knotgrass / attention

several types of attention modules written in PyTorch for learning purposes

transformers pytorch transformer attention attention-mechanism softmax-layer multi-head-attention multi-query-attention grouped-query-attention scale-dot-product-attention

Updated Oct 1, 2024
Python

M-e-r-c-u-r-y / pytorch-transformers

Collection of different types of transformers for learning purposes

transformers pytorch multi-head-attention einsum-notation multi-query-attention

Updated Jan 30, 2020
Jupyter Notebook

Improve this page

Add a description, image, and links to the multi-query-attention topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the multi-query-attention topic, visit your repo's landing page and select "manage topics."