Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Core deep-learning frameworks and libraries for pretraining and distributed training.
Signals: deep-learning, neural-network, pytorch, tensorflow, jax, distributed-training, training, deepspeed
2,663 results
Export Hugging Face models to Core ML and TensorFlow Lite
Vision-Centric BEV Perception: A Survey
Tevatron - Unified Document Retrieval Toolkit across Scale, Language, and Modality. Demo in SIGIR 2023, SIGIR 2025.
A Keras TensorFlow 2.0 implementation of BERT, ALBERT and adapter-BERT.
[ECCV2022] MOTR: End-to-End Multiple-Object Tracking with TRansformer
(ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"
Transformer based on a variant of attention that is linear complexity in respect to sequence length
Official Pytorch Code for "Medical Transformer: Gated Axial-Attention for Medical Image Segmentation" - MICCAI 2021
A Change Detection Repo Standing on the Shoulders of Giants
Papers about graph transformers.
SOTA Semantic Segmentation Models in PyTorch
allRank is a framework for training learning-to-rank neural models based on PyTorch.
Implementation of TabTransformer, attention network for tabular data, in Pytorch
Explainability for Vision Transformers
Simple transformer implementation from scratch in pytorch. (archival, latest version on codeberg)
My implementation of the original transformer model (Vaswani et al.). I've additionally included the playground.py file for visualizing otherwise seemingly hard concepts. Currently included IWSLT pretrained models.
EfficientFormerV2 [ICCV 2023] & EfficientFormer [NeurIPs 2022]
SwissArmyTransformer is a flexible and powerful library to develop your own Transformer variants.
[NeurIPS 2020] Official code for the paper "DeepSVG: A Hierarchical Generative Network for Vector Graphics Animation". Includes a PyTorch library for deep learning with SVG data.
A Deep Learning library for EEG Tasks (Signals) Classification, based on TensorFlow.
💁 Awesome Treasure of Transformers Models for Natural Language processing contains papers, videos, blogs, official repo along with colab Notebooks. 🛫☑️
Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.
Sequence-to-sequence framework with a focus on Neural Machine Translation based on PyTorch
Attention is all you need implementation
24,545 repositories in the index in total.
