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
Implementation of Communication-Efficient Learning of Deep Networks from Decentralized Data
In PyTorch Learing Neural Networks Likes CNN、BiLSTM
Easy to use extractive text summarization with BERT
AMD-SHARK Studio -- Web UI for SHARK+IREE High Performance Machine Learning Distribution
A General NeRF Acceleration Toolbox in PyTorch.
A configurable, tunable, and reproducible library for CTR prediction https://fuxictr.github.io
Improving Convolutional Networks via Attention Transfer (ICLR 2017)
[MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.
Official repository of my book: "Deep Learning with PyTorch Step-by-Step: A Beginner's Guide"
fire-smoke-detect-yolov4-yolov5 and fire-smoke-detection-dataset 火灾检测，烟雾检测
PyTorch tutorials, examples and some books I found 【不定期更新】整理的PyTorch 最新版教程、例子和书籍
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).
BoT-SORT: Robust Associations Multi-Pedestrian Tracking
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
24,540 repositories in the index in total.
