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
PyTorch based Probabilistic Time Series forecasting framework based on GluonTS backend
[ICLR'23 Spotlight🔥] The first successful BERT/MAE-style pretraining on any convolutional network; Pytorch impl. of "Designing BERT for Convolutional Networks: Sparse and Hierarchical Masked Modeling"
[pip install medmnist] 18x Standardized Datasets for 2D and 3D Biomedical Image Classification
Tensors, for human consumption
A best practice for deep learning project template architecture.
A flexible package for multimodal-deep-learning to combine tabular data with text and images using Wide and Deep models in Pytorch
Receptive Field Block Net for Accurate and Fast Object Detection, ECCV 2018
This repository provides the code and model checkpoints for AIMv1 and AIMv2 research projects.
MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.
Collective communications library with various primitives for multi-machine training.
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
24,523 repositories in the index in total.
