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,672 results
Neural Machine Translation with Keras
Fast implementation of BERT inference directly on NVIDIA (CUDA, CUBLAS) and Intel MKL
Cross-view Transformers for real-time Map-view Semantic Segmentation (CVPR 2022 Oral)
臺灣大學 (NTU) 李宏毅教授「機器學習 (Machine Learning) 2021 Spring 」課程筆記
"Pop Music Transformer: Beat-based Modeling and Generation of Expressive Pop Piano Compositions", ACM Multimedia 2020
A PyTorch implementation of "Graph Wavelet Neural Network" (ICLR 2019)
TransMorph: Transformer for Unsupervised Medical Image Registration (PyTorch)
This code accompanies the the paper Trading with the Momentum Transformer: An Intelligent and Interpretable Architecture (https://arxiv.org/pdf/2112.08534.pdf).
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.
Official PyTorch implementation of Superpoint Transformer [ICCV'23], SuperCluster [3DV'24 Oral], and EZ-SP [ICRA'26]
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)
24,540 repositories in the index in total.
