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.

GitHub: https://github.com/topics/pytorch
Wikipedia: https://en.wikipedia.org/wiki/PyTorch
Repo: https://github.com/pytorch/pytorch
Created by: Facebook's AI Research lab (FAIR)
Released: September 2016
Related Topics: python, pytorch-tutorial,
Last updated: 2024-11-10 00:23:23 UTC
JSON Representation

https://github.com/vietnh1009/quickdraw

Implementation of Quickdraw - an online game developed by Google

cnn computer-vision cv2 deep-learning deep-neural-networks deeplearning image-classification neural-network neural-networks opencv opencv-python python python3 pytorch quickdraw quickdraw-dataset

Last synced: 12 Nov 2024

https://github.com/uvipen/quickdraw

Implementation of Quickdraw - an online game developed by Google

cnn computer-vision cv2 deep-learning deep-neural-networks deeplearning image-classification neural-network neural-networks opencv opencv-python python python3 pytorch quickdraw quickdraw-dataset

Last synced: 15 Oct 2024

https://github.com/nyoki-mtl/pytorch-EverybodyDanceNow

Implementation of Everybody Dance Now by pytorch

everybody-dance-now pytorch

Last synced: 04 Nov 2024

https://github.com/sirius-ai/LPRNet_Pytorch

Pytorch Implementation For LPRNet, A High Performance And Lightweight License Plate Recognition Framework.

ctc-loss license-plate-recognition lprnet plate-detection plate-recognition pytorch

Last synced: 28 Oct 2024

https://github.com/Denys88/rl_games

RL implementations

deep-learning pytorch reinforcement-learning

Last synced: 03 Nov 2024

https://github.com/sberbank-ai-lab/LightAutoML

LAMA - automatic model creation framework

automated-machine-learning automl blackbox classification data-science ensembling feature-engineering gradient-boosting kaggle lama linear-model model-selection multiclass nlp parameter-tuning pipeline pytorch regression stacking whitebox

Last synced: 07 Aug 2024

https://github.com/nvidia/retinanet-examples

Fast and accurate object detection with end-to-end GPU optimization

deep-learning neural-network object-detection python pytorch retinanet tensorrt

Last synced: 29 Oct 2024

https://github.com/jankrepl/deepdow

Portfolio optimization with deep learning.

allocation convex-optimization deep-learning finance machine-learning markowitz portfolio-optimization pytorch stock-price-prediction timeseries trading wealth-management

Last synced: 28 Oct 2024

https://github.com/uvipen/QuickDraw

Implementation of Quickdraw - an online game developed by Google

cnn computer-vision cv2 deep-learning deep-neural-networks deeplearning image-classification neural-network neural-networks opencv opencv-python python python3 pytorch quickdraw quickdraw-dataset

Last synced: 07 Aug 2024

https://github.com/NVIDIA/retinanet-examples

Fast and accurate object detection with end-to-end GPU optimization

deep-learning neural-network object-detection python pytorch retinanet tensorrt

Last synced: 07 Aug 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: 13 Nov 2024

https://github.com/BaguaSys/bagua

Bagua Speeds up PyTorch

bagua deep-learning distributed distributed-computing distributed-systems machine-learning pytorch rust-lang

Last synced: 30 Oct 2024

https://github.com/baguasys/bagua

Bagua Speeds up PyTorch

bagua deep-learning distributed distributed-computing distributed-systems machine-learning pytorch rust-lang

Last synced: 12 Oct 2024

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: 13 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: 13 Nov 2024

https://github.com/denys88/rl_games

RL implementations

deep-learning pytorch reinforcement-learning

Last synced: 14 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: 12 Nov 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: 13 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: 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: 13 Nov 2024

https://github.com/amusi/PyTorch-From-Zero-To-One

PyTorch从入门到精通

deep-learning pytorch pytorch-tutorials

Last synced: 28 Oct 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: 13 Nov 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: 11 Nov 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/eric-mingjie/network-slimming

Network Slimming (Pytorch) (ICCV 2017)

channel-pruning convolutional-neural-networks deep-learning pytorch sparsity

Last synced: 08 Nov 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/ypwhs/dl-engineer-guidebook

深度学习工程师生存指南

deep-learning deeplearning gitbook guidebook hardware homebrew installation-notes linux macos pytorch tensorflow

Last synced: 14 Nov 2024

https://github.com/opendilab/lightzero

[NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios

alpha-beta-pruning alphazero atari board-game board-games continuous-control efficientzero gomoku gumbel-muzero gym mcts mcts-algorithm monte-carlo-tree-search muzero pytorch reinforcement-learning sampled-muzero self-play stochastic-muzero tictactoe

Last synced: 13 Oct 2024

https://github.com/bentrevett/pytorch-image-classification

Tutorials on how to implement a few key architectures for image classification using PyTorch and TorchVision.

alexnet cnn convolutional-networks convolutional-neural-network convolutional-neural-networks image-classification lenet pytorch pytorch-cnn pytorch-implementation pytorch-implmention pytorch-tutorial pytorch-tutorials resnet torchvision tutorial vgg

Last synced: 14 Nov 2024

https://github.com/gmalivenko/pytorch2keras

PyTorch to Keras model convertor

converter darts deep-learning deep-learning-models densenet imagenet keras keras-model-convertor keras-models keras-neural-networks keras-tensorflow machinelearning models-converter onnx pytorch pytorch-models resnet tensorflow tensorflow-models tensorflowjs

Last synced: 01 Nov 2024

https://github.com/NirAharon/BoT-SORT

BoT-SORT: Robust Associations Multi-Pedestrian Tracking

multi-object-tracker multi-object-tracking multi-pedestrian-tracking pytorch tracking-by-detection yolov7 yolox

Last synced: 27 Oct 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/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/microsoft/godel

Large-scale pretrained models for goal-directed dialog

conversational-ai data-processing dialogpt dialogue dialogue-systems grounded-generation language-grounding language-model machine-learning pretrained-model pytorch text-data text-generation transformer transformers

Last synced: 14 Nov 2024

https://github.com/mlmed/torchxrayvision

TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.

chest-radiographs chest-xray chest-xray-images cxr cxr-images dataset deep-learning foundation-models image-classification machine-learning medical medical-ai medical-application medical-image-analysis medical-image-processing medical-imaging pytorch torchxrayvision transfer-learning

Last synced: 03 Aug 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: 14 Nov 2024

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

Explainability for Vision Transformers

deep-learning explainable-ai pytorch transformer vision-transformer

Last synced: 13 Nov 2024

https://github.com/perone/medicaltorch

A medical imaging framework for Pytorch

computer-vision deep-learning machine-learning medical-imaging python pytorch

Last synced: 13 Nov 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/microsoft/GODEL

Large-scale pretrained models for goal-directed dialog

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/progamergov/neural-style-pt

PyTorch implementation of neural style transfer algorithm

deep-learning deep-style neural-style neural-style-pt neural-style-transfer nst pytorch pytorch-style-transfer style-transfer styletransfer

Last synced: 13 Nov 2024

https://github.com/NVlabs/ODISE

Official PyTorch implementation of ODISE: Open-Vocabulary Panoptic Segmentation with Text-to-Image Diffusion Models [CVPR 2023 Highlight]

deep-learning diffusion-models instance-segmentation open-vocabulary open-vocabulary-segmentation open-vocabulary-semantic-segmentation open-world-classification open-world-object-detection panoptic-segmentation pytorch semantic-segmentation text-image-retrieval zero-shot-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: 14 Nov 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/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/gordicaleksa/pytorch-original-transformer

My implementation of the original transformer model (Vaswani et al.). I've additionally included the playground.py file for visualizing otherwise seemingly hard concepts. Currently included IWSLT pretrained models.

attention attention-is-all-you-need attention-mechanism deep-learning deeplearning jupyter original-transformer python pytorch pytorch-transformer pytorch-transformers transformer transformer-tutorial transformers

Last synced: 11 Nov 2024

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

A PyTorch implementation of DenseNet.

deep-learning densenet pytorch

Last synced: 13 Nov 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/yzhangcs/parser

:rocket: State-of-the-art parsers for natural language.

constituency-parsing dependency-parsing pytorch semantic-dependency-parsing semiring state-of-the-art structured-prediction transformers

Last synced: 01 Oct 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/braindotai/Watermark-Removal-Pytorch

🔥 CNN for Watermark Removal using Deep Image Prior with Pytorch 🔥.

artefacts-removal deep-image-priors deep-learning paper pytorch restoration-tasks skip-connections watermark watermark-removal

Last synced: 08 Nov 2024

https://github.com/nyukat/breast_cancer_classifier

Deep Neural Networks Improve Radiologists' Performance in Breast Cancer Screening

breast-cancer breast-cancer-diagnosis classification deep-learning medical-image-analysis medical-imaging neural-network pretrained-models pytorch tensorflow

Last synced: 04 Aug 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/smilelight/lightNLP

基于Pytorch和torchtext的自然语言处理深度学习框架。

cbow chinese cws deep-learning dependency-parsing hierarchical-softmax language-model nature-language-process negative-sampling nlp pos python pytorch relation-extraction sentence-similarity skip-gram srl text-classification torchtext word2vec

Last synced: 03 Aug 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/kakaobrain/torchgpipe

A GPipe implementation in PyTorch

checkpointing deep-learning gpipe model-parallelism parallelism pipeline-parallelism pytorch

Last synced: 10 Nov 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/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/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/mimoralea/gdrl

Grokking Deep Reinforcement Learning

algorithms artificial-intelligence deep-learning deep-reinforcement-learning docker gpu machine-learning neural-networks numpy numpy-tutorial nvidia-docker pytorch pytorch-tutorials reinforcement-learning

Last synced: 13 Nov 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/haofanwang/controlnet-for-diffusers

Transfer the ControlNet with any basemodel in diffusers🔥

aigc controlnet diffusers pytorch tutorials

Last synced: 14 Nov 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/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/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/havakv/pycox

Survival analysis with PyTorch

deep-learning machine-learning neural-networks python pytorch survival-analysis

Last synced: 07 Nov 2024

https://github.com/mert-kurttutan/torchview

torchview: visualize pytorch models

deep-learning keras machine-learning neural-network python pytorch torchviz visualization

Last synced: 13 Oct 2024

https://github.com/bayesiains/nflows

Normalizing flows in PyTorch

density-estimation generative-model normalizing-flows pytorch

Last synced: 25 Oct 2024

https://github.com/datadreamer-dev/DataDreamer

DataDreamer: Prompt. Generate Synthetic Data. Train & Align Models. 🤖💤

alignment deep-learning fine-tuning gpt instruction-tuning llm llmops llms machine-learning natural-language-processing nlp nlp-library openai python pytorch synthetic-data synthetic-dataset-generation transformers

Last synced: 13 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/jingweiz/pytorch-rl

Deep Reinforcement Learning with pytorch & visdom

a3c acer actor-critic deep-learning deep-reinforcement-learning dqn pytorch pytorch-a3c reinforcement-learning trpo visdom

Last synced: 07 Aug 2024

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

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

gmm neural-network python pytorch zhihu

Last synced: 13 Nov 2024

https://github.com/polarisZhao/PFLD-pytorch

PFLD pytorch Implementation

face-landmark-detection ncnn pfld pfld-ncnn pfld-pytorch pytorch

Last synced: 28 Oct 2024

https://github.com/yanndubs/disentangling-vae

Experiments for understanding disentanglement in VAE latent representations

beta-tcvae beta-vae celeba chairs-dataset deep-learning disentangled-representations disentanglement dsprites factor-vae fashion-mnist mnist pytorch representation-learning reproducible-research unsupervised-learning vae variational-autoencoder

Last synced: 01 Nov 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/MushroomRL/mushroom-rl

Python library for Reinforcement Learning.

atari ddpg deep-learning deep-reinforcement-learning dqn mujoco openai-gym pybullet pytorch qlearning reinforcement-learning rl sac trpo

Last synced: 02 Nov 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/benedekrozemberczki/ClusterGCN

A PyTorch implementation of "Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networks" (KDD 2019).

clustering community-detection deep-learning deepwalk diff2vec gcn gemsec graph-clustering graph-convolution graph-convolutional-networks graph-neural-networks graph2vec graphsage louvain metis musae neural-network node-classification node2vec pytorch

Last synced: 12 Nov 2024

https://github.com/pranz24/pytorch-soft-actor-critic

PyTorch implementation of soft actor critic

deep-reinforcement-learning pytorch pytorch-implmention reinforcement-learning soft-actor-critic

Last synced: 12 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