Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

PyTorch

PyTorch is an open source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing, primarily developed by Facebook’s AI Research lab.

https://github.com/chrischoy/pytorch-custom-cuda-tutorial

Tutorial for building a custom CUDA function for Pytorch

python pytorch pytorch-backend tutorial wrapper

Last synced: 14 Nov 2024

https://github.com/PennyLaneAI/qml

Introductions to key concepts in quantum programming, as well as tutorials and implementations from cutting-edge quantum computing research.

autograd automatic-differentiation demo key-concepts neural-networks pytorch qml quantum-chemistry quantum-computing quantum-machine-learning tensorflow tutorials

Last synced: 03 Nov 2024

https://github.com/ksw0306/FloWaveNet

A Pytorch implementation of "FloWaveNet: A Generative Flow for Raw Audio"

clarinet generative-flow glow pytorch wavenet

Last synced: 17 Nov 2024

https://github.com/harry24k/bayesian-neural-network-pytorch

PyTorch implementation of bayesian neural network [torchbnn]

bayesian deep-learning neural-network pytorch

Last synced: 14 Nov 2024

https://github.com/adambielski/capsnet-pytorch

PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules

capsnet capsules deep-learning dynamic-routing-between-capsules machine-learning mnist pytorch

Last synced: 13 Nov 2024

https://github.com/facebookresearch/NeuralCompression

A collection of tools for neural compression enthusiasts.

compression deep-learning jax machine-learning neural-compression python pytorch

Last synced: 05 Aug 2024

https://github.com/fartashf/vsepp

PyTorch Code for the paper "VSE++: Improving Visual-Semantic Embeddings with Hard Negatives"

bmvc negatives paper pytorch vse

Last synced: 15 Nov 2024

https://github.com/dongminlee94/deep_rl

PyTorch implementation of deep reinforcement learning algorithms

a2c ddpg ddqn deep-reinforcement-learning dqn model-free-rl npg ppo pytorch sac sac-aea td3 trpo vpg

Last synced: 11 Nov 2024

https://github.com/salesforce/matchbox

Write PyTorch code at the level of individual examples, then run it efficiently on minibatches.

deep-learning minibatch nlp pytorch

Last synced: 14 Nov 2024

https://github.com/shibing624/pytextclassifier

pytextclassifier is a toolkit for text classification. 文本分类，LR，Xgboost，TextCNN，FastText，TextRNN，BERT等分类模型实现，开箱即用。

bert classification focalloss-pytorch hierarchical machine-learning nlp pytextclassifier python pytorch softmax text-classification text-classifier

Last synced: 11 Oct 2024

https://github.com/jaketae/storyteller

Multimodal AI Story Teller, built with Stable Diffusion, GPT, and neural text-to-speech

ddpm diffusion-models gpt image-generation natural-language-generation pytorch stable-diffusion text-to-image text-to-speech text-to-video video-generation

Last synced: 04 Aug 2024

https://github.com/adambielski/CapsNet-pytorch

PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules

capsnet capsules deep-learning dynamic-routing-between-capsules machine-learning mnist pytorch

Last synced: 07 Aug 2024

https://github.com/wvangansbeke/sparse-depth-completion

Predict dense depth maps from sparse and noisy LiDAR frames guided by RGB images. (Ranked 1st place on KITTI) [MVA 2019]

computer-vision deep-learning depth-completion depth-prediction kitti lidar noisy-data pytorch sensor-fusion

Last synced: 14 Nov 2024

https://github.com/crowsonkb/style-transfer-pytorch

Neural style transfer in PyTorch.

neural-art pytorch style-transfer

Last synced: 13 Nov 2024

https://github.com/nv-tlabs/steal

STEAL - Learning Semantic Boundaries from Noisy Annotations (CVPR 2019)

annotation cvpr2019 deep-learning devil-is-in-the-edges nv-tlabs pytorch semantic-boundaries semantic-segmentation steal

Last synced: 15 Nov 2024

https://github.com/fkodom/fft-conv-pytorch

Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.

convolution image-processing neural-networks python3 pytorch

Last synced: 17 Nov 2024

https://github.com/cure-lab/MagicDrive

[ICLR24] Official implementation of the paper “MagicDrive: Street View Generation with Diverse 3D Geometry Control”

autonomous-vehicles deep-learning diffusion-models image-generation pytorch video-generation

Last synced: 28 Oct 2024

https://github.com/ma-xu/pointMLP-pytorch

[ICLR 2022 poster] Official PyTorch implementation of "Rethinking Network Design and Local Geometry in Point Cloud: A Simple Residual MLP Framework"

modelnet40 pointcloud pytorch scanobjectnn

Last synced: 28 Oct 2024

https://github.com/wildltr/ptranking

Learning to Rank in PyTorch

learning-to-rank pytorch ranking

Last synced: 11 Nov 2024

https://github.com/archinetai/surgeon-pytorch

A library to inspect and extract intermediate layers of PyTorch models.

artificial-intelligence deep-learning pytorch

Last synced: 05 Nov 2024

https://github.com/hkchengrex/mivos

[CVPR 2021] Modular Interactive Video Object Segmentation: Interaction-to-Mask, Propagation and Difference-Aware Fusion. Semi-supervised VOS as well!

computer-vision cvpr2021 deep-learning interactive-segmentation pytorch segmentation video-object-segmentation video-segmentation

Last synced: 17 Nov 2024

https://github.com/Neutone/neutone_sdk

Join the community on Discord for more discussions around Neutone! https://discord.gg/VHSMzb8Wqp

ai audio audio-plugin deep-learning machine-learning music python pytorch realtime-audio sdk vst

Last synced: 05 Nov 2024

https://github.com/kevinzakka/recurrent-visual-attention

A PyTorch Implementation of "Recurrent Models of Visual Attention"

attention pytorch ram recurrent-attention-model recurrent-models

Last synced: 30 Oct 2024

https://github.com/zwwwayne/k-net

[NeurIPS2021] Code Release of K-Net: Towards Unified Image Segmentation

image-segmentation mmdetection mmsegmentation panoptic-segmentation pytorch semantic-segmentation

Last synced: 17 Nov 2024

https://github.com/ZwwWayne/K-Net

[NeurIPS2021] Code Release of K-Net: Towards Unified Image Segmentation

image-segmentation mmdetection mmsegmentation panoptic-segmentation pytorch semantic-segmentation

Last synced: 15 Nov 2024

https://github.com/tri-ml/dd3d

Official PyTorch implementation of DD3D: Is Pseudo-Lidar needed for Monocular 3D Object detection? (ICCV 2021), Dennis Park*, Rares Ambrus*, Vitor Guizilini, Jie Li, and Adrien Gaidon.

computer-vision deep-learning pytorch

Last synced: 13 Nov 2024

https://github.com/criteo-research/reco-gym

Code for reco-gym: A Reinforcement Learning Environment for the problem of Product Recommendation in Online Advertising

criteo openai-gym pytorch recommendation-system

Last synced: 17 Nov 2024

https://github.com/backtime92/CRAFT-Reimplementation

CRAFT-Pyotorch：Character Region Awareness for Text Detection Reimplementation for Pytorch

craft pytorch text-detection

Last synced: 03 Nov 2024

https://github.com/JingyibySUTsoftware/Yolov5-deepsort-driverDistracted-driving-behavior-detection

基于深度学习的驾驶员分心驾驶行为（疲劳+危险行为）预警系统使用YOLOv5+Deepsort实现驾驶员的危险驾驶行为的预警监测

deeplearning deepsort detection dlib dlib-face-detection dlib-tracker driver-behavior object-detection opencv playsound pyqt5 python pytorch sort tired tracker yolov5 yolov5-deepsort

Last synced: 09 Nov 2024

https://github.com/microsoft/archai

Accelerate your Neural Architecture Search (NAS) through fast, reproducible and modular research.

automated-machine-learning automl darts deep-learning hyperparameter-optimization machine-learning model-compression nas neural-architecture-search petridish python pytorch

Last synced: 07 Oct 2024

https://github.com/chingyaoc/ggnn.pytorch

A PyTorch Implementation of Gated Graph Sequence Neural Networks (GGNN)

ggnn gnns graph pytorch

Last synced: 30 Oct 2024

https://github.com/ahangchen/torch_base

Quickly bring up your PyTorch project(a skeleton)

computer-vision deep-learning machine-learning pytorch

Last synced: 18 Nov 2024

https://github.com/mit-han-lab/mcunet

[NeurIPS 2020] MCUNet: Tiny Deep Learning on IoT Devices; [NeurIPS 2021] MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning

deep-learning microncontroller neural-architecture-search pytorch tinyml

Last synced: 28 Oct 2024

https://github.com/wenmuzhou/psenet.pytorch

A pytorch re-implementation of PSENet: Shape Robust Text Detection with Progressive Scale Expansion Network

ocr python3 pytorch text-detection

Last synced: 14 Nov 2024

https://github.com/vita-group/autogan

[ICCV 2019] "AutoGAN: Neural Architecture Search for Generative Adversarial Networks" by Xinyu Gong, Shiyu Chang, Yifan Jiang and Zhangyang Wang

autogan cifar10 gan neural-architecture-search pytorch

Last synced: 16 Nov 2024

https://github.com/davidefiocco/streamlit-fastapi-model-serving

Simple web app example serving a PyTorch model using streamlit and FastAPI

docker-compose fastapi pycones pytorch streamlit

Last synced: 13 Oct 2024

https://github.com/salesforce/warp-drive

Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)

cuda deep-learning gpu high-throughput multiagent-reinforcement-learning numba pytorch reinforcement-learning

Last synced: 14 Nov 2024

https://github.com/TRI-ML/dd3d

Official PyTorch implementation of DD3D: Is Pseudo-Lidar needed for Monocular 3D Object detection? (ICCV 2021), Dennis Park*, Rares Ambrus*, Vitor Guizilini, Jie Li, and Adrien Gaidon.

computer-vision deep-learning pytorch

Last synced: 28 Oct 2024

https://github.com/edisonleeeee/graphgallery

GraphGallery is a gallery for benchmarking Graph Neural Networks, From InplusLab.

adversarial-machine-learning dgl geometric-deep-learning graph graph-neural-networks graphgallery models pyg pytorch

Last synced: 13 Nov 2024

https://github.com/sinAshish/Multi-Scale-Attention

Code for our paper "Multi-scale Guided Attention for Medical Image Segmentation"

attention-mechanism biomedical-image-analysis deep-learning encoder-decoder pytorch

Last synced: 15 Nov 2024

https://github.com/likyoo/open-cd

A Change Detection Repo Standing on the Shoulders of Giants

bit change-detection changer deep-learning pytorch transformer

Last synced: 17 Nov 2024

https://github.com/EdisonLeeeee/GraphGallery

GraphGallery is a gallery for benchmarking Graph Neural Networks, From InplusLab.

adversarial-machine-learning dgl geometric-deep-learning graph graph-neural-networks graphgallery models pyg pytorch

Last synced: 12 Nov 2024

https://github.com/Schlumberger/joint-vae

Pytorch implementation of JointVAE, a framework for disentangling continuous and discrete factors of variation :star2:

disentangled-representations disentanglement generative-models gumbel-softmax pytorch vae

Last synced: 14 Nov 2024

https://github.com/yinchangchang/ocr_densenet

第一届西安交通大学人工智能实践大赛（2018AI实践大赛--图片文字识别）第一名；仅采用densenet识别图中文字

densenet ocr ocr-recognition python pytorch

Last synced: 03 Nov 2024

https://github.com/layumi/aicity-reid-2020

:red_car: The 1st Place Submission to AICity Challenge 2020 re-id track (Baidu-UTS submission)

aicity cityflow cvpr2020 paddlepaddle pytorch vehicle vehicle-reid veri-776

Last synced: 14 Nov 2024

https://github.com/layumi/AICIty-reID-2020

:red_car: The 1st Place Submission to AICity Challenge 2020 re-id track (Baidu-UTS submission)

aicity cityflow cvpr2020 paddlepaddle pytorch vehicle vehicle-reid veri-776

Last synced: 08 Nov 2024

https://github.com/farukalamai/advanced-machine-learning-engineer-roadmap-2024

A Full Stack ML (Machine Learning) Roadmap involves learning the necessary skills and technologies to become proficient in all aspects of machine learning, including data collection and preprocessing, model development, deployment, and maintenance.

aws computer-vision data-analysis data-science data-visualization deep-learning git-github machine-learning machine-learning-roadmap mlops natural-language-processing neural-network nlp opencv pandas python pytorch statistics tensorflow yolo

Last synced: 14 Nov 2024

https://github.com/dfdazac/wassdistance

Approximating Wasserstein distances with PyTorch

deep-learning machine-learning optimal-transport pytorch

Last synced: 12 Nov 2024

https://github.com/clovaai/rexnet

Official Pytorch implementation of ReXNet (Rank eXpansion Network) with pretrained models

deep-learning image-classification imagenet object-detection pretrained-model pytorch rexnet transfer-learning

Last synced: 12 Nov 2024

https://github.com/walktree/libtorch-yolov3

A Libtorch implementation of the YOLO v3 object detection algorithm

cpp libtorch pytorch yolov3

Last synced: 09 Nov 2024

https://github.com/huggingface/large_language_model_training_playbook

An open collection of implementation tips, tricks and resources for training large language models

cuda large-language-models llm nccl nlp performance python pytorch scalability troubleshooting

Last synced: 11 Nov 2024

https://github.com/cuiziteng/Illumination-Adaptive-Transformer

[BMVC 2022] You Only Need 90K Parameters to Adapt Light: A Light Weight Transformer for Image Enhancement and Exposure Correction. SOTA for low light enhancement, 0.004 seconds try this for pre-processing.

bmvc exposure-correction image-enhancement image-reconstruction image-restoration low-level-vision low-light-enhance low-light-image-enhancement pytorch transformer-architecture transformer-models

Last synced: 06 Nov 2024

https://github.com/ploomber/sklearn-evaluation

Machine learning model evaluation made easy: plots, tables, HTML reports, experiment tracking and Jupyter notebook analysis.

data-science deep-learning jupyter-notebook machine-learning pytorch scikit-learn sklearn tensorflow

Last synced: 13 Oct 2024

https://github.com/thudm/graphmae

GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22

deep-learning graph-neural-networks pytorch self-supervised-learning

Last synced: 14 Nov 2024

https://github.com/chaitjo/geometric-gnn-dojo

Geometric GNN Dojo provides unified implementations and experiments to explore the design space of Geometric Graph Neural Networks.

e3nn equivariance equivariant-networks geometric-deep-learning geometric-graphs graph-isomorphism graph-neural-networks pytorch pytorch-geometric

Last synced: 15 Nov 2024

https://github.com/loretoparisi/CapsNet

CapsNet (Capsules Net) in Geoffrey E Hinton paper "Dynamic Routing Between Capsules" - State Of the Art

capsnet capsule-networks chainer deeplearning keras lasagne machine-learning mnist mxnet pytorch tensorflow

Last synced: 02 Nov 2024

https://github.com/RaivoKoot/Video-Dataset-Loading-Pytorch

Generic PyTorch dataset implementation to load and augment VIDEOS for deep learning training loops.

action-recognition dataloader deep-learning machine-learning pytorch video-dataset videos

Last synced: 10 Nov 2024

https://github.com/mikoto10032/automaticweightedloss

Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning

auxiliary-tasks deep-learning multi-task multi-task-learning pytorch weigh-losses

Last synced: 11 Nov 2024

https://github.com/sacmehta/ESPNetv2

A light-weight, power efficient, and general purpose convolutional neural network

cnn convolutional-neural-networks deep-learning efficient imagenet lightweight machine-learning pytorch semantic-segmentation

Last synced: 31 Oct 2024

https://github.com/mv-lab/InstructIR

[ECCV 2024] InstructIR: High-Quality Image Restoration Following Human Instructions https://huggingface.co/spaces/marcosv/InstructIR

computer-vision deblurring deep-learning dehazing denoising image-enhancement image-restoration inverse-problems language-model low-light-image-enhancement multi-task multimodal neural-network photography prompt pytorch super-resolution

Last synced: 03 Nov 2024

https://github.com/nschaetti/EchoTorch

A Python toolkit for Reservoir Computing and Echo State Network experimentation based on pyTorch. EchoTorch is the only Python module available to easily create Deep Reservoir Computing models.

artificial-intelligence artificial-neural-networks echo-state-networks machine-learning machine-learning-algorithms machinelearning neural-networks python python-toolkit pytorch recurrent-networks recurrent-neural-networks reservoir-computing torch

Last synced: 16 Nov 2024

https://github.com/iBelieveCJM/Tricks-of-Semi-supervisedDeepLeanring-Pytorch

PseudoLabel 2013, VAT, PI model, Tempens, MeanTeacher, ICT, MixMatch, FixMatch

fixmatch ict meanteacher mixmatch pi pseudo-label pytorch semi-supervised-learning ssl tempens vat

Last synced: 15 Nov 2024

https://github.com/sthalles/pytorch-byol

PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning

byol deep-learning pytorch representation-learning self-supervised-learning simclr unsupervised-learning

Last synced: 06 Nov 2024

https://github.com/THUDM/GraphMAE

GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22

deep-learning graph-neural-networks pytorch self-supervised-learning

Last synced: 08 Aug 2024

https://github.com/pykale/pykale

Knowledge-Aware machine LEarning (KALE): accessible machine learning from multiple sources for interdisciplinary research, part of the 🔥PyTorch ecosystem. ⭐ Star to support our work!

computer-vision data-science deep-learning domain-adaptation graph-analysis knowledge-aware-learning machine-learning medical-image-analysis meta-learning multimodal multimodal-learning python pytorch transfer-learning

Last synced: 11 Oct 2024

https://github.com/ruiminshen/yolo2-pytorch

PyTorch implementation of the YOLO (You Only Look Once) v2

caffe2 computer-vision deep-learning deep-neural-networks object-detection onnx python pytorch yolo2

Last synced: 04 Nov 2024

https://github.com/cfzd/Ultra-Fast-Lane-Detection-v2

Ultra Fast Deep Lane Detection With Hybrid Anchor Driven Ordinal Classification (TPAMI 2022)

deep-learning lane-detection pytorch

Last synced: 11 Nov 2024

https://github.com/lemonhu/ner-bert-pytorch

PyTorch solution of named entity recognition task Using Google AI's pre-trained BERT model.

chinese-ner entity-extraction google-bert information-extraction msra named-entity-recognition ner pytorch transformer

Last synced: 16 Nov 2024

https://github.com/um-arm-lab/pytorch_mppi

Model Predictive Path Integral (MPPI) with approximate dynamics implemented in pytorch

approximate-dynamics controls model-predictive-control mppi pytorch

Last synced: 16 Nov 2024

https://github.com/lemonhu/NER-BERT-pytorch

PyTorch solution of named entity recognition task Using Google AI's pre-trained BERT model.

chinese-ner entity-extraction google-bert information-extraction msra named-entity-recognition ner pytorch transformer

Last synced: 02 Nov 2024

PyTorch Awesome Lists
Awesome-pytorch-list 704 awesome-semantic-segmentation-pytorch 35 Awesome-pytorch-list-CNVersion 692 awesome-colab-notebooks 525 CV-pretrained-model 103 awesome-AutoML-and-Lightweight-Models 150 openstl 43 Awesome-Denoise 164 ai4artists 286 arctic 34 awesome-cuda-and-hpc 427 awesome-yolov8-models 43 awesome-python-machine-learning-resources 1,884 awesome-adaptive-computation 173 awesome-NILM-with-code 49 awesome-dotnet-machine-learning 134 awesome-Multi-Document-Summarization 56 PyTorch-Guide 616 awesome-max-mojo 183 awesome-fastai 42 awesome-ai-tools 526 free-pytorch-resources 72
PyTorch Categories
Blogs 450 Frameworks 277 Pytorch & related libraries 227 Pytorch & related libraries｜Pytorch & 相关库 216 Research 198 Tutorials 192 文本数据和NLP 182 Uncategorized 161 数据可视化 131 机器学习框架 123 1.) Neural Architecture Search 123 Blogposts :sparkle: 121 图像数据与CV 106 Learning Resources 103 Learning 101 模型的可解释性 100 Tutorials, books, & examples 96 Tutorials & books & examples｜教程 & 书籍 & 示例 94 Interactive Learning & Tools 94 超参数优化和AutoML 93