Topic: attention-mechanism

2,745 repositories

rasbt/LLMs-from-scratch

Implement a ChatGPT-like LLM in PyTorch from scratch, step by step

★ 106,664Jupyter NotebookForks 16,418

lucidrains/vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

★ 25,550PythonForks 3,496

nndl/nndl

邱锡鹏《神经网络与深度学习》第二版与通识版:电子书、章节目录、学习资源与勘误。

★ 19,435Forks 3,689

BlinkDL/RWKV-LM

RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable). We are at RWKV-7 "Goose". So it's combining the best of RNN and transformer - great performance, linear time, constant space (no kv-cache), fast training, infinite ctx_len, and free sentence embedding.

★ 14,750PythonForks 1,025

lucidrains/x-transformers

A concise but complete full-attention transformer with a set of promising experimental features from various papers

★ 5,959PythonForks 524

lucidrains/DALLE-pytorch

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

★ 5,627PythonForks 639

ruvnet/RuVector

RuVector provides High Performance, Real-Time decisions and agent memory , Self-Learning Ai, Vector GNN DB built in Rust.

★ 4,570RustForks 611

Kyubyong/transformer

A TensorFlow Implementation of the Transformer: Attention Is All You Need

★ 4,477PythonForks 1,304

PetarV-/GAT

Graph Attention Networks (https://arxiv.org/abs/1710.10903)

★ 3,556PythonForks 671

diegoantognini/pyGAT

Pytorch implementation of the Graph Attention Network model by Veličković et. al (2017, https://arxiv.org/abs/1710.10903)

★ 3,125PythonForks 701

gordicaleksa/pytorch-GAT

My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entropy histograms. I've supported both Cora (transductive) and PPI (inductive) examples!

★ 2,688Jupyter NotebookForks 357

lucidrains/alphafold2

To eventually become an unofficial Pytorch implementation / replication of Alphafold2, as details of the architecture get released

★ 1,640PythonForks 264

lucidrains/soundstorm-pytorch

Implementation of SoundStorm, Efficient Parallel Audio Generation from Google Deepmind, in Pytorch

★ 1,545PythonForks 94

lucidrains/lambda-networks

Implementation of LambdaNetworks, a new approach to image recognition that reaches SOTA with less compute

★ 1,530PythonForks 155

charlesXu86/Chatbot_CN

基于金融-司法领域(兼有闲聊性质)的聊天机器人,其中的主要模块有信息抽取、NLU、NLG、知识图谱等,并且利用Django整合了前端展示,目前已经封装了nlp和kg的restful接口

★ 1,291Forks 423

lucidrains/flamingo-pytorch

Implementation of 🦩 Flamingo, state-of-the-art few-shot visual question answering attention net out of Deepmind, in Pytorch

★ 1,267PythonForks 66

awslabs/sockeye

Sequence-to-sequence framework with a focus on Neural Machine Translation based on PyTorch

★ 1,216PythonForks 319

The-AI-Summer/self-attention-cv

Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.

★ 1,214PythonForks 152

lucidrains/CoCa-pytorch

Implementation of CoCa, Contrastive Captioners are Image-Text Foundation Models, in Pytorch

★ 1,201PythonForks 90