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
CNN Image Retrieval in PyTorch: Training and evaluating CNNs for Image Retrieval in PyTorch
NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).
A Python Library for Graph Outlier Detection (Anomaly Detection)
Model analyzer in PyTorch
Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)
Toward Multimodal Image-to-Image Translation
A PyTorch Implementation of Federated Learning
Monocular, One-stage, Regression of Multiple 3D People and their 3D positions & trajectories in camera & global coordinates. ROMP[ICCV21], BEV[CVPR22], TRACE[CVPR2023]
A memory-efficient implementation of DenseNets
A Data Streaming Library for Efficient Neural Network Training
A very simple generative adversarial network (GAN) in PyTorch
An easy-to-use federated learning platform
An easy/swift-to-adapt PyTorch-Lighting template. 套壳模板，简单易用，稍改原来Pytorch代码，即可适配Lightning。You can translate your previous Pytorch code much easier using this template, and keep your freedom to edit all the functions as well. Big-project-friendly as well. No need to rewrite your config in hydra.
Pyramid Stereo Matching Network (CVPR2018)
A pytorch implementation of Paper "Improved Training of Wasserstein GANs"
World's first general purpose 3D object detection codebse.
Implementation of Non-local Block.
an implementation of 3D Ken Burns Effect from a Single Image using PyTorch
In-depth tutorials for implementing deep learning models on your own with PyTorch.
Measures and metrics for image2image tasks. PyTorch.
A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
Bayesian Convolutional Neural Network with Variational Inference based on Bayes by Backprop in PyTorch.
Winning Solution in NTIRE19 Challenges on Video Restoration and Enhancement (CVPR19 Workshops) - Video Restoration with Enhanced Deformable Convolutional Networks. EDVR has been merged into BasicSR and this repo is a mirror of BasicSR.
Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS
24,523 repositories in the index in total.
