Illustration dessinée par RepoMasters à partir de l’identifiant du dépôt et de la couleur de son langage. Crédits
AlphaPeu commune

lucidrains/memory-efficient-attention-pytorch

Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"

Propriétaire
lucidrains (compte personnel)
Étoiles
Étoiles : 395
Forks
Forks : 36
Langage
Python
Licence
MIT
Création
2022
Sujets
artificial-intelligencedeep-learningattention-mechanismmemory-efficient
Rang
Alpha (250 à 699 ★) Ce que veut dire ce rang
Voir sur GitHub

Autres dépôts de lucidrains

Les 175 dépôts de lucidrains

D’autres dépôts Python

Tous les dépôts Python