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
Efficient Deep Learning Systems course materials
Normalizing flows in PyTorch
an implementation of Video Frame Interpolation via Adaptive Separable Convolution using PyTorch
Deep learning toolkit-enabled VLSI placement
Pytorch version of SfmLearner from Tinghui Zhou et al.
Byted PyTorch Distributed for Hyperscale Training of LLMs and RLs
A unified ensemble framework for PyTorch to improve the performance and robustness of your deep learning model.
CNNs for Sentence Classification in PyTorch
PyTorch Implementation of "Lossless Image Compression through Super-Resolution"
OpenMMLab optical flow toolbox and benchmark
torchview: visualize pytorch models
Implementation of Prototypical Networks for Few Shot Learning (https://arxiv.org/abs/1703.05175) in Pytorch
Temporal Segment Networks (TSN) in PyTorch
Profiling and inspecting memory in pytorch
Factorization Machine models in PyTorch
From the Tensor to Stable Diffusion, a rough outline for a 10 week course.
Riemannian Adaptive Optimization Methods with pytorch optim
[IEEE TIP] "EnlightenGAN: Deep Light Enhancement without Paired Supervision" by Yifan Jiang, Xinyu Gong, Ding Liu, Yu Cheng, Chen Fang, Xiaohui Shen, Jianchao Yang, Pan Zhou, Zhangyang Wang
Tencent Pre-training framework in PyTorch & Pre-trained Model Zoo
PyTorch implementation of "Learning to Discover Cross-Domain Relations with Generative Adversarial Networks"
UNetFormer: A UNet-like transformer for efficient semantic segmentation of remote sensing urban scene imagery, ISPRS. Also, including other vision transformers and CNNs for satellite, aerial image and UAV image segmentation.
TextBox 2.0 is a text generation library with pre-trained language models
PyTorch re-implementation of DeepLab v2 on COCO-Stuff / PASCAL VOC datasets
PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
24,540 repositories in the index in total.
