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,686 results
A learning rate range test implementation in PyTorch
Introduction to PyTorch, covering tensor initialization, operations, indexing, and reshaping.
Efficient Deep Learning Systems course materials
Normalizing flows in PyTorch
an implementation of Video Frame Interpolation via Adaptive Separable Convolution using PyTorch
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"
[NeurIPS'22] PointNeXt: Revisiting PointNet++ with Improved Training and Scaling Strategies
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
24,523 repositories in the index in total.
