1 dépôt
Specific patterns for constructing autoregressive GPT-style decoders with customizable depth and attention heads.
Distinct from Model Construction: Focuses specifically on the construction logic of GPT-style decoders, rather than general model construction.
Explore 1 awesome GitHub repository matching artificial intelligence & ml · GPT Decoder Construction. Refine with filters or upvote what's useful.
x-transformers est une bibliothèque PyTorch et un toolkit de recherche pour construire des architectures transformer. Il fournit un framework modulaire pour implémenter la recherche transformer expérimentale, incluant une suite de mécanismes d'attention avancés, des outils de modélisation de séquences longues et un framework pour les vision transformers. Le projet se distingue par son accent sur les composants haute performance et économes en mémoire, tels que Flash Attention avec des noyaux tuilés et l'attention multi-requêtes. Il implémente également des méthodes spécialisées pour étendre les fenêtres de contexte, incluant la récurrence de séquence et les plongements positionnels rotatifs. La bibliothèque couvre un large éventail de capacités architecturales, incluant divers schémas de normalisation pour stabiliser l'entraînement, des réseaux feedforward à portes et des topologies de couches personnalisées comme les réseaux Macaron. Elle prend en charge les constructions d'encodeur et de décodeur, fournissant des outils pour la génération de séquence autorégressive et les tâches vision-langage comme la légende d'image.
Implements patterns for constructing autoregressive GPT-style decoders with customizable depth and attention heads.