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,694 results
[CVPR2023] MOTRv2: Bootstrapping End-to-End Multi-Object Tracking by Pretrained Object Detectors
Transformer implementation in PyTorch.
Transformer Network for Remaining Useful Life Prediction of Lithium-Ion Batteries
The official PyTorch implementation of the paper "SAITS: Self-Attention-based Imputation for Time Series". A fast and state-of-the-art (SOTA) deep-learning neural network model for efficient time-series imputation (impute multivariate incomplete time series containing NaN missing data/values with machine learning). https://arxiv.org/abs/2202.08516
Official Pytorch implementation of "OmniNet: A unified architecture for multi-modal multi-task learning" | Authors: Subhojeet Pramanik, Priyanka Agrawal, Aman Hussain
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
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
tfts: Time Series Deep Learning Models in TensorFlow
Papers about graph transformers.
SOTA Semantic Segmentation Models in PyTorch
allRank is a framework for training learning-to-rank neural models based on PyTorch.
24,524 repositories in the index in total.
