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)
146Updated 8 months ago

Alternatives and similar repositories for grouped-query-attention-pytorch:

Users that are interested in grouped-query-attention-pytorch are comparing it to the libraries listed below