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/jacobgil/pytorch-pruning

PyTorch Implementation of [1611.06440] Pruning Convolutional Neural Networks for Resource Efficient Inference

deep-learning pruning pytorch

Last synced: 30 Oct 2024

https://github.com/moemen95/PyTorch-Project-Template

A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.

condense-net dcgan deep-learning dqn erfnet machine-learning pytorch pytorch-project-template pytorch-template pytorch-tutorial scemantic semantic-segmentation-models

Last synced: 02 Nov 2024

https://github.com/moemen95/pytorch-project-template

A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.

condense-net dcgan deep-learning dqn erfnet machine-learning pytorch pytorch-project-template pytorch-template pytorch-tutorial scemantic semantic-segmentation-models

Last synced: 30 Oct 2024

https://github.com/dvgodoy/pytorchstepbystep

Official repository of my book: "Deep Learning with PyTorch Step-by-Step: A Beginner's Guide"

cnn-pytorch deep-learning python pytorch pytorch-tutorial rnn-pytorch

Last synced: 30 Oct 2024

https://github.com/moemen95/Pytorch-Project-Template

A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.

condense-net dcgan deep-learning dqn erfnet machine-learning pytorch pytorch-project-template pytorch-template pytorch-tutorial scemantic semantic-segmentation-models

Last synced: 03 Aug 2024

https://github.com/switchablenorms/Switchable-Normalization

Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10779

convolutional-neural-networks deeplearning imagenet normalization pytorch

Last synced: 02 Aug 2024

https://github.com/frotms/paddleocr2pytorch

PaddleOCR inference in PyTorch. Converted from [PaddleOCR](https://github.com/PaddlePaddle/PaddleOCR)

crnn db multilingual-models ocr paddleocr paddlepaddle pytorch pytorchocr

Last synced: 15 Oct 2024

https://github.com/dvgodoy/PyTorchStepByStep

Official repository of my book: "Deep Learning with PyTorch Step-by-Step: A Beginner's Guide"

cnn-pytorch deep-learning python pytorch pytorch-tutorial rnn-pytorch

Last synced: 28 Oct 2024

https://github.com/explosion/curated-transformers

🤖 A PyTorch library of curated Transformer models and their composable components

albert bert camembert dolly2 falcon gptneox llama llm llms nlp pytorch roberta transformer transformers xlm-roberta

Last synced: 07 Oct 2024

https://github.com/allegro/allrank

allRank is a framework for training learning-to-rank neural models based on PyTorch.

click-model deep-learning information-retrieval learning-to-rank machine-learning ndcg python pytorch ranking transformer

Last synced: 06 Nov 2024

https://github.com/philipperemy/n-beats

Keras/Pytorch implementation of N-BEATS: Neural basis expansion analysis for interpretable time series forecasting.

deep-learning neural-networks pytorch series-forecasting

Last synced: 21 Oct 2024

https://github.com/lonepatient/bert-multi-label-text-classification

This repo contains a PyTorch implementation of a pretrained BERT model for multi-label text classification.

albert bert fine-tuning multi-label-classification nlp pytorch pytorch-implmention text-classification transformers xlnet

Last synced: 06 Nov 2024

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

A pytorch re-implementation of Real-time Scene Text Detection with Differentiable Binarization

ocr python3 pytorch text-detection

Last synced: 07 Nov 2024

https://github.com/ZhiqingXiao/rl-book

Source codes for the book "Reinforcement Learning: Theory and Python Implementation"

deep-reinforcement-learning gym openai-gym python pytorch reinforcement-learning tensorflow tensorflow2

Last synced: 03 Nov 2024

https://github.com/lonePatient/Bert-Multi-Label-Text-Classification

This repo contains a PyTorch implementation of a pretrained BERT model for multi-label text classification.

albert bert fine-tuning multi-label-classification nlp pytorch pytorch-implmention text-classification transformers xlnet

Last synced: 04 Aug 2024

https://github.com/airctic/icevision

An Agnostic Computer Vision Framework - Pluggable to any Training Library: Fastai, Pytorch-Lightning with more to come

ai annotation-parsers coco-dataset coco-parser computer-vision deep-learning effecientdet fastai faster-rcnn mask-rcnn object-detection pycocotools python pytorch pytorch-lightning tutorials voc-dataset voc-parser

Last synced: 31 Oct 2024

https://github.com/liyuanlucasliu/lm-lstm-crf

Empower Sequence Labeling with Task-Aware Language Model

crf language-model ner pytorch sequence-labeling

Last synced: 09 Nov 2024

https://github.com/vahidk/tfrecord

Standalone TFRecord reader/writer with PyTorch data loaders

dataset loader pytorch tensorflow tfrecord

Last synced: 03 Aug 2024

https://github.com/zisianw/FaceBoxes.PyTorch

A PyTorch Implementation of FaceBoxes

face-detection faceboxes pytorch

Last synced: 28 Oct 2024

https://github.com/jacobgil/vit-explain

Explainability for Vision Transformers

deep-learning explainable-ai pytorch transformer vision-transformer

Last synced: 30 Oct 2024

https://github.com/wasidennis/AdaptSegNet

Learning to Adapt Structured Output Space for Semantic Segmentation, CVPR 2018 (spotlight)

adversarial-learning computer-vision deep-learning domain-adaptation generative-adversarial-network pytorch semantic-segmentation

Last synced: 05 Aug 2024

https://github.com/easezyc/deep-transfer-learning

A collection of implementations of deep domain adaptation algorithms

deep-transfer-learning domain-adaptation pytorch transfer-learning

Last synced: 02 Aug 2024

https://github.com/RL-VIG/LibFewShot

LibFewShot: A Comprehensive Library for Few-shot Learning. TPAMI 2023.

few-shot-learning fine-tuning image-classification meta-learning pytorch

Last synced: 03 Aug 2024

https://github.com/allegro/allRank

allRank is a framework for training learning-to-rank neural models based on PyTorch.

click-model deep-learning information-retrieval learning-to-rank machine-learning ndcg python pytorch ranking transformer

Last synced: 02 Aug 2024

https://github.com/minzhang-whu/change-detection-review

A review of change detection methods, including codes and open data sets for deep learning. From paper: change detection based on artificial intelligence: state-of-the-art and challenges.

artificial-intelligence caffe change-detection code deep-learning keras machine-learning matlab open-datasets python pytorch remote-sensing review sar streetview tensorflow unsupervised-learning

Last synced: 31 Oct 2024

https://github.com/MinZHANG-WHU/Change-Detection-Review

A review of change detection methods, including codes and open data sets for deep learning. From paper: change detection based on artificial intelligence: state-of-the-art and challenges.

artificial-intelligence caffe change-detection code deep-learning keras machine-learning matlab open-datasets python pytorch remote-sensing review sar streetview tensorflow unsupervised-learning

Last synced: 01 Nov 2024

https://github.com/baal-org/baal

Bayesian active learning library for research and industrial usecases.

active-learning ai bayesian-active-learning deep-learning machine-learning python pytorch

Last synced: 03 Aug 2024

https://github.com/hkchengrex/cascadepsp

[CVPR 2020] CascadePSP: Toward Class-Agnostic and Very High-Resolution Segmentation via Global and Local Refinement

computer-vision cvpr2020 deep-learning high-resolution pytorch refinement-network segmentation segmentation-refinement

Last synced: 10 Nov 2024

https://github.com/BUPT-GAMMA/OpenHGNN

This is an open-source toolkit for Heterogeneous Graph Neural Network(OpenHGNN) based on DGL.

dgl graph-neural-networks heterogeneous pytorch

Last synced: 08 Aug 2024

https://github.com/voldemortX/pytorch-auto-drive

PytorchAutoDrive: Segmentation models (ERFNet, ENet, DeepLab, FCN...) and Lane detection models (SCNN, RESA, LSTR, LaneATT, BézierLaneNet...) based on PyTorch with fast training, visualization, benchmarking & deployment help

apex cityscapes culane deeplab erfnet fcn gtav lane-detection lstr mobilenet onnx pascal-voc pytorch resa scnn semantic-segmentation tensorboard tensorrt tusimple

Last synced: 28 Oct 2024

https://github.com/bamos/densenet.pytorch

A PyTorch implementation of DenseNet.

deep-learning densenet pytorch

Last synced: 30 Oct 2024

https://github.com/pytorch/translate

Translate - a PyTorch Language Library

artificial-intelligence machine-learning onnx pytorch

Last synced: 26 Sep 2024

https://github.com/RedisAI/RedisAI

A Redis module for serving tensors and executing deep learning graphs

machine-learning onnxruntime pytorch redisai serving-tensors tensorflow

Last synced: 30 Oct 2024

https://github.com/diegoalejogm/gans

Generative Adversarial Networks implemented in PyTorch and Tensorflow

cyclegan dcgans gans gans-collections paper-implementations pytorch tensorflow

Last synced: 07 Aug 2024

https://github.com/nashory/pggan-pytorch

:fire::fire: PyTorch implementation of "Progressive growing of GANs (PGGAN)" :fire::fire:

celeba-hq-dataset gan generative-adversarial-network progressive-gan progressively-growing-gan pytorch tensorboard

Last synced: 05 Nov 2024

https://github.com/geoopt/geoopt

Riemannian Adaptive Optimization Methods with pytorch optim

optimization pytorch riemannian-geometry riemannian-manifold riemannian-optimization

Last synced: 30 Oct 2024

https://github.com/MontaEllis/Pytorch-Medical-Segmentation

This repository is an unoffical PyTorch implementation of Medical segmentation in 2D and 3D.

medical-image-processing pytorch segmentation torchio

Last synced: 07 Aug 2024

https://github.com/agrimgupta92/sgan

Code for "Social GAN: Socially Acceptable Trajectories with Generative Adversarial Networks", Gupta et al, CVPR 2018

deep-learning generative-adversarial-network human-trajectory-prediction pytorch social-navigation trajectory-prediction

Last synced: 10 Nov 2024

https://github.com/cloneofsimo/minDiffusion

Self-contained, minimalistic implementation of diffusion models with Pytorch.

diffusion pytorch

Last synced: 30 Oct 2024

https://github.com/doragd/chinese-chatbot-pytorch-implementation

:four_leaf_clover: Another Chinese chatbot implemented in PyTorch, which is the sub-module of intelligent work order processing robot. 👩‍🔧

chatbot chinese-nlp deep-learning pytorch pytorch-nlp

Last synced: 29 Oct 2024

https://github.com/kimhc6028/relational-networks

Pytorch implementation of "A simple neural network module for relational reasoning" (Relational Networks)

deep-learning pytorch relational-networks

Last synced: 02 Nov 2024

https://github.com/Replicable-MARL/MARLlib

One repository is all that is necessary for Multi-agent Reinforcement Learning (MARL)

deep-reinforcement-learning multi-agent-reinforcement-learning pytorch ray rllib

Last synced: 30 Oct 2024

https://github.com/wujiyang/Face_Pytorch

face recognition algorithms in pytorch framework, including arcface, cosface, sphereface and so on

arcface cosface face-recognition pytorch sphereface

Last synced: 28 Oct 2024

https://github.com/alibaba/BladeDISC

BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.

compiler deep-learning inference-optimization machine-learning mlir neural-network pytorch tensorflow

Last synced: 27 Oct 2024

https://github.com/megvii-research/IJCAI2023-CoNR

IJCAI2023 - Collaborative Neural Rendering using Anime Character Sheets

anime computer-graphics computer-vision deep-learning pytorch

Last synced: 08 Nov 2024

https://github.com/denis2054/transformers-for-nlp-2nd-edition

Transformer models from BERT to GPT-4, environments from Hugging Face to OpenAI. Fine-tuning, training, and prompt engineering examples. A bonus section with ChatGPT, GPT-3.5-turbo, GPT-4, and DALL-E including jump starting GPT-4, speech-to-text, text-to-speech, text to image generation with DALL-E, Google Cloud AI,HuggingGPT, and more

bert chatgpt chatgpt-api dall-e dall-e-api deep-learning gpt-3-5-turbo gpt-4 gpt-4-api huggingface-transformers machine-learning natural-language-processing nlp openai python pytorch roberta-model transformers trax

Last synced: 09 Nov 2024

https://github.com/shadow2496/VITON-HD

Official PyTorch implementation of "VITON-HD: High-Resolution Virtual Try-On via Misalignment-Aware Normalization" (CVPR 2021)

deep-learning pytorch

Last synced: 08 Nov 2024

https://github.com/996refuse/zheye

者也 - 知乎 倒立的文字 汉字验证码识别程序

gmm neural-network python pytorch zhihu

Last synced: 30 Oct 2024

https://github.com/xxxnell/how-do-vits-work

(ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"

loss-landscape pytorch self-attention transformer vision-transformer

Last synced: 03 Aug 2024

https://github.com/kazuto1011/grad-cam-pytorch

PyTorch re-implementation of Grad-CAM (+ vanilla/guided backpropagation, deconvnet, and occlusion sensitivity maps)

deconvnet grad-cam guided-backpropagation occlusion-sensitivity pytorch visualization

Last synced: 03 Aug 2024

https://github.com/mosecorg/mosec

A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine

cv deep-learning gpu hacktoberfest jax llm llm-serving machine-learning machine-learning-platform mlops model-serving mxnet nerual-network python pytorch rust tensorflow tts

Last synced: 09 Nov 2024

https://github.com/rahulschand/gpu_poor

Calculate token/s & GPU memory requirement for any LLM. Supports llama.cpp/ggml/bnb/QLoRA quantization

ggml gpu huggingface language-model llama llama2 llamacpp llm pytorch quantization

Last synced: 10 Oct 2024

https://github.com/yeyupiaoling/voiceprintrecognition-pytorch

This project uses a variety of advanced voiceprint recognition models such as EcapaTdnn, ResNetSE, ERes2Net, CAM++, etc. It is not excluded that more models will be supported in the future. At the same time, this project also supports MelSpectrogram, Spectrogram data preprocessing methods

arcface ecapa-tdnn pytorch speaker-recognition voice-recognition

Last synced: 31 Oct 2024

https://github.com/torrvision/crayon

A language-agnostic interface to TensorBoard

crayon data-visualization deep-learning docker pytorch tensorboard tensorflow torch7

Last synced: 30 Sep 2024

https://github.com/kwea123/VTuber_Unity

Use Unity 3D character and Python deep learning algorithms to stream as a VTuber!

dlib face-detection face-landmark-detection face-pose gaze-estimation python pytorch unity unity-chan vrm vtuber

Last synced: 10 Nov 2024

https://github.com/pytorch/benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

benchmark pytorch

Last synced: 07 Oct 2024

https://github.com/oaid/autokernel

AutoKernel 是一个简单易用，低门槛的自动算子优化工具，提高深度学习算法部署效率。

auto autosearch deep-learning halide optimization pytorch reinforcement-learning tengine tensor tensorflow tvm

Last synced: 05 Nov 2024

https://github.com/youngwanLEE/centermask2

[CVPR 2020] CenterMask : Real-time Anchor-Free Instance Segmentation

anchor-free centermask cvpr2020 detectron2 instance-segmentation object-detection pytorch real-time vovnet vovnetv2

Last synced: 26 Oct 2024

https://github.com/MediaBrain-SJTU/MING

明医 (MING)：中文医疗问诊大模型

consultation huggingface llm medical pytorch transformers

Last synced: 03 Aug 2024

https://github.com/parrt/tensor-sensor

The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.

debugging deep-learning jax matrix numpy python pytorch tensorflow tracing vector

Last synced: 15 Oct 2024

https://github.com/jeya-maria-jose/Medical-Transformer

Official Pytorch Code for "Medical Transformer: Gated Axial-Attention for Medical Image Segmentation" - MICCAI 2021

attention deep-learning medical-image-analysis medical-imaging pytorch segmentation transformer transformers

Last synced: 03 Aug 2024

https://github.com/tusen-ai/SST

Code for a series of work in LiDAR perception, including SST (CVPR 22), FSD (NeurIPS 22), FSD++ (TPAMI 23), FSDv2, and CTRL (ICCV 23, oral).

3d-object-detection autonomous-driving pytorch

Last synced: 28 Oct 2024

https://github.com/tusen-ai/sst

Code for a series of work in LiDAR perception, including SST (CVPR 22), FSD (NeurIPS 22), FSD++ (TPAMI 23), FSDv2, and CTRL (ICCV 23, oral).

3d-object-detection autonomous-driving pytorch

Last synced: 08 Nov 2024

https://github.com/vandit15/Class-balanced-loss-pytorch

Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"

computer-vision cvpr2019 deep-learning loss-functions pytorch

Last synced: 03 Aug 2024

https://github.com/microsoft/cameratraps

PyTorch Wildlife: a Collaborative Deep Learning Framework for Conservation.

camera-traps computer-vision conservation machine-learning megadetector pytorch pytorch-wildlife wildlife

Last synced: 30 Oct 2024

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

[NeurIPS 2020] MCUNet: Tiny Deep Learning on IoT Devices; [NeurIPS 2021] MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning; [NeurIPS 2022] MCUNetV3: On-Device Training Under 256KB Memory

c codegenerator cpp deep-learning edge-computing microcontroller neural-architecture-search pytorch quantization tinyml

Last synced: 27 Oct 2024

https://github.com/kuan-wang/pytorch-mobilenet-v3

MobileNetV3 in pytorch and ImageNet pretrained models

classification imagenet mobilenet mobilenetv2 mobilenetv3 pytorch

Last synced: 31 Oct 2024

https://github.com/Spijkervet/SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.

contrastive-learning pytorch representation-learning simclr unsupervised-learning

Last synced: 05 Nov 2024

https://github.com/kaituoxu/Speech-Transformer

A PyTorch implementation of Speech Transformer, an End-to-End ASR with Transformer network on Mandarin Chinese.

asr attention attention-is-all-you-need end-to-end pytorch self-attention transformer

Last synced: 07 Aug 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 47 awesome-dotnet-machine-learning 134 awesome-Multi-Document-Summarization 56 PyTorch-Guide 616 awesome-max-mojo 183 awesome-fastai 42 awesome-ai-tools 502
PyTorch Categories
Blogs 450 Frameworks 278 Pytorch & related libraries 231 Pytorch & related libraries｜Pytorch & 相关库 220 Research 198 Tutorials 192 文本数据和NLP 182 Uncategorized 161 数据可视化 131 机器学习框架 123 1.) Neural Architecture Search 122 Blogposts :sparkle: 119 图像数据与CV 106 Learning Resources 105 Learning 101 模型的可解释性 100 Tutorials, books, & examples 95 Interactive Learning & Tools 94 Tutorials & books & examples｜教程 & 书籍 & 示例 93 超参数优化和AutoML 93