Illustration drawn by RepoMasters from the repo’s ID and its language color. Credits
PrototypeCommon

fkodom/grouped-query-attention-pytorch

(Unofficial) PyTorch implementation of grouped-query attention (GQA) from "GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpoints" (https://arxiv.org/pdf/2305.13245.pdf)

Owner
fkodom (personal account)
Stars
Stars: 196
Forks
Forks: 14
Language
Python
License
MIT
Created
2023
Tier
Prototype (100 to 249 ★) What this tier means
View on GitHub

More repos by fkodom

All 4 repos by fkodom

Nearby in Python

All Python repos