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-15 00:24:09 UTC
JSON Representation

https://github.com/vikparuchuri/zero_to_gpt

Go from no deep learning knowledge to implementing GPT.

deep-learning python pytorch tutorial

Last synced: 13 Nov 2024

https://github.com/shi-labs/neighborhood-attention-transformer

Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022

neighborhood-attention pytorch

Last synced: 15 Oct 2024

https://github.com/xl0/lovely-tensors

Tensors, ready for human consumption

deep-learning library pytorch statistics visualization

Last synced: 15 Oct 2024

https://github.com/ternaus/TernausNet

UNet model with VGG11 encoder pre-trained on Kaggle Carvana dataset

image-segmentation pytorch

Last synced: 07 Aug 2024

https://github.com/ternaus/ternausnet

UNet model with VGG11 encoder pre-trained on Kaggle Carvana dataset

image-segmentation pytorch

Last synced: 14 Oct 2024

https://github.com/ayoolaolafenwa/pixellib

Visit PixelLib's official documentation https://pixellib.readthedocs.io/en/latest/

artificial-intelligence computer-vision convolutional-neural-networks deep-learning deeplab deeplearning image-segmentation instance-segmentation machine-learning maskr-cnn object-detection pointrend pytorch segmentation semantic-segmentation tensorflow video-segmentation

Last synced: 14 Oct 2024

https://github.com/eduardoleao052/js-pytorch

A JavaScript library like PyTorch, with GPU acceleration.

automatic-differentiation deep-learning javascript-library neural-networks pytorch

Last synced: 15 Oct 2024

https://github.com/caoscott/srec

PyTorch Implementation of "Lossless Image Compression through Super-Resolution"

compression lossless neural-network pytorch

Last synced: 29 Oct 2024

https://github.com/cardwing/Codes-for-Lane-Detection

Learning Lightweight Lane Detection CNNs by Self Attention Distillation (ICCV 2019)

cnn deep-learning lane-detection lua pytorch tensorflow

Last synced: 26 Oct 2024

https://github.com/cardwing/codes-for-lane-detection

Learning Lightweight Lane Detection CNNs by Self Attention Distillation (ICCV 2019)

cnn deep-learning lane-detection lua pytorch tensorflow

Last synced: 30 Oct 2024

https://github.com/uvipen/super-mario-bros-a3c-pytorch

Asynchronous Advantage Actor-Critic (A3C) algorithm for Super Mario Bros

a3c ai deep-learning gym mario python pytorch reinforcement-learning super-mario-bros

Last synced: 14 Oct 2024

https://github.com/caoscott/SReC

PyTorch Implementation of "Lossless Image Compression through Super-Resolution"

compression lossless neural-network pytorch

Last synced: 05 Aug 2024

https://github.com/pbloem/former

Simple transformer implementation from scratch in pytorch.

machine-learning pytorch transformer

Last synced: 15 Oct 2024

https://github.com/maum-ai/voicefilter

Unofficial PyTorch implementation of Google AI's VoiceFilter system

audio-separation pytorch source-separation speech-separation voicefilter

Last synced: 29 Oct 2024

https://github.com/sicara/easy-few-shot-learning

Ready-to-use code and tutorial notebooks to boost your way into few-shot learning for image classification.

deep-learning few-shot-classifcation few-shot-learning few-shot-recognition image-classification machine-learning meta-learning pytorch

Last synced: 15 Oct 2024

https://github.com/unum-cloud/uform

Pocket-Sized Multimodal AI for content understanding and generation across multilingual texts, images, and 🔜 video, up to 5x faster than OpenAI CLIP and LLaVA 🖼️ & 🖋️

bert clip clustering contrastive-learning cross-attention huggingface-transformers image-search language-vision llava multi-lingual multimodal neural-network openai openclip pretrained-models pytorch representation-learning semantic-search transformer vector-search

Last synced: 14 Oct 2024

https://github.com/rixwew/pytorch-fm

Factorization Machine models in PyTorch

autoint avazu-dataset collaborative-filtering criteo-dataset ctr-prediction dcn deepfm factorization-machines ffm fm fnfm hofm movielens-dataset neural-collaborative-filtering nfm pnn pytorch xdeepfm

Last synced: 29 Oct 2024

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

[ICCV 2019] "DeblurGAN-v2: Deblurring (Orders-of-Magnitude) Faster and Better" by Orest Kupyn, Tetiana Martyniuk, Junru Wu, Zhangyang Wang

deblurgan deep-learning generative-adversarial-network iccv iccv2019 low-level-vision pytorch ukraine

Last synced: 29 Oct 2024

https://github.com/huggingface/pytorch-pretrained-biggan

🦋A PyTorch implementation of BigGAN with pretrained weights and conversion scripts.

artificial-intelligence biggan computer-vision gan generative-adversarial-network neural-network pytorch

Last synced: 15 Oct 2024

https://github.com/hendrycks/robustness

Corruption and Perturbation Robustness (ICLR 2019)

computer-vision convolutional-neural-networks deep-learning domain-generalization imagenet machine-learning ml-safety pytorch robustness

Last synced: 13 Nov 2024

https://github.com/shawn1993/cnn-text-classification-pytorch

CNNs for Sentence Classification in PyTorch

cnn-model pytorch

Last synced: 12 Nov 2024

https://github.com/Stonesjtu/pytorch_memlab

Profiling and inspecting memory in pytorch

cuda-memory memory-profiler pytorch

Last synced: 12 Nov 2024

https://github.com/huggingface/pytorch-pretrained-BigGAN

🦋A PyTorch implementation of BigGAN with pretrained weights and conversion scripts.

artificial-intelligence biggan computer-vision gan generative-adversarial-network neural-network pytorch

Last synced: 07 Aug 2024

https://github.com/microsoft/windows-machine-learning

Samples and Tools for Windows ML.

ai caffe2 coreml deep-learning keras machine-learning ml neural-network onnx pytorch scikit-learn tensorflow windows winmltools

Last synced: 07 Oct 2024

https://github.com/descriptinc/descript-audio-codec

State-of-the-art audio codec with 90x compression factor. Supports 44.1kHz, 24kHz, and 16kHz mono/stereo audio.

audio audio-compression codec compression-algorithm deep-learning gans generative-adversarial-network pytorch residual-vector-quantization

Last synced: 05 Aug 2024

https://github.com/stonesjtu/pytorch_memlab

Profiling and inspecting memory in pytorch

cuda-memory memory-profiler pytorch

Last synced: 15 Oct 2024

https://github.com/sniklaus/sepconv-slomo

an implementation of Video Frame Interpolation via Adaptive Separable Convolution using PyTorch

cuda cupy deep-learning python pytorch

Last synced: 29 Oct 2024

https://github.com/wpeebles/gangealing

Official PyTorch Implementation of "GAN-Supervised Dense Visual Alignment" (CVPR 2022 Oral, Best Paper Finalist)

alignment computer-vision correspondence cvpr cvpr2022 deep-learning gan generative image-manipulation pytorch stylegan stylegan2 tracking

Last synced: 30 Oct 2024

https://github.com/uvipen/Super-mario-bros-A3C-pytorch

Asynchronous Advantage Actor-Critic (A3C) algorithm for Super Mario Bros

a3c ai deep-learning gym mario python pytorch reinforcement-learning super-mario-bros

Last synced: 07 Aug 2024

https://github.com/ClementPinard/SfmLearner-Pytorch

Pytorch version of SfmLearner from Tinghui Zhou et al.

depth disparity kitti pytorch unsupervised

Last synced: 27 Oct 2024

https://github.com/VITA-Group/DeblurGANv2

[ICCV 2019] "DeblurGAN-v2: Deblurring (Orders-of-Magnitude) Faster and Better" by Orest Kupyn, Tetiana Martyniuk, Junru Wu, Zhangyang Wang

deblurgan deep-learning generative-adversarial-network iccv iccv2019 low-level-vision pytorch ukraine

Last synced: 07 Aug 2024

https://github.com/toshas/torch-fidelity

High-fidelity performance metrics for generative models in PyTorch

evaluation frechet-inception-distance gan generative-model inception-score kernel-inception-distance metrics perceptual-path-length precision pytorch reproducibility reproducible-research

Last synced: 16 Nov 2024

https://github.com/piddnad/DDColor

[ICCV 2023] DDColor: Towards Photo-Realistic Image Colorization via Dual Decoders

computer-vision image-colorization pytorch

Last synced: 29 Aug 2024

https://github.com/ermongroup/sdedit

PyTorch implementation for SDEdit: Image Synthesis and Editing with Stochastic Differential Equations

controllable-generation image-editing image-generation image-manipulation pytorch score-matching

Last synced: 18 Nov 2024

https://github.com/snap-research/efficientformer

EfficientFormerV2 [ICCV 2023] & EfficientFormer [NeurIPs 2022]

deep-learning detection efficient-inference efficient-neural-networks imagenet mobile-devices pytorch semantic-segmentation transformer transformers

Last synced: 15 Nov 2024

https://github.com/a-r-j/graphein

Protein Graph Library

bioinformatics computational-biology deep-learning dgl drug-discovery gene-regulatory-networks geometric-deep-learning graph-neural-networks interactome interactomics ppi-networks protein protein-data-bank protein-design protein-structure python pytorch pytorch-geometric rna structural-biology

Last synced: 13 Nov 2024

https://github.com/thudm/swissarmytransformer

SwissArmyTransformer is a flexible and powerful library to develop your own Transformer variants.

pretrained-models pytorch transformer

Last synced: 14 Nov 2024

https://github.com/nvidia/modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

deep-learning machine-learning nvidia-gpu physics pytorch

Last synced: 29 Oct 2024

https://github.com/NVIDIA/modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

deep-learning machine-learning nvidia-gpu physics pytorch

Last synced: 30 Oct 2024

https://github.com/orobix/prototypical-networks-for-few-shot-learning-pytorch

Implementation of Prototypical Networks for Few Shot Learning (https://arxiv.org/abs/1703.05175) in Pytorch

cnn prototypical-networks python pytorch

Last synced: 11 Nov 2024

https://github.com/tencent/tencentpretrain

Tencent Pre-training framework in PyTorch & Pre-trained Model Zoo

albert bart bert chinese classification clue elmo fine-tuning gpt gpt-2 model-zoo natural-language-processing ner pegasus pre-training pytorch roberta t5 unilm xlm-roberta

Last synced: 13 Nov 2024

https://github.com/THUDM/SwissArmyTransformer

SwissArmyTransformer is a flexible and powerful library to develop your own Transformer variants.

pretrained-models pytorch transformer

Last synced: 01 Nov 2024

https://github.com/Oldpan/Pytorch-Memory-Utils

pytorch memory track code

gpu memory pytorch

Last synced: 15 Nov 2024

https://github.com/WisconsinAIVision/MixNMatch

Pytorch implementation of MixNMatch

deep-learning disentangled-representations fine-grained gans image-generation image-manipulation pytorch

Last synced: 15 Nov 2024

https://github.com/wisconsinaivision/mixnmatch

Pytorch implementation of MixNMatch

deep-learning disentangled-representations fine-grained gans image-generation image-manipulation pytorch

Last synced: 17 Nov 2024

https://github.com/orobix/Prototypical-Networks-for-Few-shot-Learning-PyTorch

Implementation of Prototypical Networks for Few Shot Learning (https://arxiv.org/abs/1703.05175) in Pytorch

cnn prototypical-networks python pytorch

Last synced: 26 Oct 2024

https://github.com/graykode/gpt-2-pytorch

Simple Text-Generator with OpenAI gpt-2 Pytorch Implementation

gpt-2 gpt2 implementation natural-language-processing nlp pytorch story-telling text-generator

Last synced: 13 Nov 2024

https://github.com/nnaisense/evotorch

Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.

artificial-intelligence distributed evolutionary-algorithms evolutionary-computation gpu neural-networks neuroevolution optimization optimization-algorithms python pytorch reinforcement-learning

Last synced: 05 Nov 2024

https://github.com/anibali/docker-pytorch

A Docker image for PyTorch

cuda docker docker-image pytorch

Last synced: 11 Oct 2024

https://github.com/WenjieDu/PyPOTS

A Python toolkit/library for reality-centric machine/deep learning and data mining on partially-observed time series, including SOTA neural network models for scientific analysis tasks of imputation/classification/clustering/forecasting/anomaly detection/cleaning on incomplete industrial (irregularly-sampled) multivariate TS with NaN missing values

classification clustering data-mining data-science deep-learning forecasting healthcare imputation incomplete industrial interpolation machine-learning missing-values missingness neural-network partially-observed-time-series pytorch science-research time-series time-series-analysis

Last synced: 02 Nov 2024

https://github.com/jmtomczak/intro_dgm

"Deep Generative Modeling": Introductory Examples

autoregressive-modeling deep-generative-modelling deep-learning energy-based-model flow-based-modeling generative-adversarial-network generative-ai neural-compression pytorch score-based-generative-modeling variational-autoencoder

Last synced: 30 Oct 2024

https://github.com/graykode/gpt-2-Pytorch

Simple Text-Generator with OpenAI gpt-2 Pytorch Implementation

gpt-2 gpt2 implementation natural-language-processing nlp pytorch story-telling text-generator

Last synced: 26 Oct 2024

https://github.com/KaiyangZhou/pytorch-center-loss

Pytorch implementation of Center Loss

center-loss computer-vision deep-learning feature-learning loss-functions python pytorch

Last synced: 15 Nov 2024

https://github.com/ermongroup/SDEdit

PyTorch implementation for SDEdit: Image Synthesis and Editing with Stochastic Differential Equations

controllable-generation image-editing image-generation image-manipulation pytorch score-matching

Last synced: 31 Oct 2024

https://github.com/snap-research/EfficientFormer

EfficientFormerV2 [ICCV 2023] & EfficientFormer [NeurIPs 2022]

deep-learning detection efficient-inference efficient-neural-networks imagenet mobile-devices pytorch semantic-segmentation transformer transformers

Last synced: 15 Nov 2024

https://github.com/NVIDIA-AI-IOT/trt_pose

Real-time pose estimation accelerated with NVIDIA TensorRT

human-pose human-pose-estimation jetson jetson-nano jetson-xavier live-demo pretrained-models pytorch real-time tensorrt torch2trt

Last synced: 25 Oct 2024

https://github.com/VoltaML/voltaML-fast-stable-diffusion

Beautiful and Easy to use Stable Diffusion WebUI

ai-art aitemplate generative-art linux naive-ui python pytorch rust stable-diffusion text2image typescript vue webui windows

Last synced: 25 Oct 2024

https://github.com/open-mmlab/mmflow

OpenMMLab optical flow toolbox and benchmark

openmmlab optical-flow pytorch

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

https://github.com/kaiyangzhou/pytorch-center-loss

Pytorch implementation of Center Loss

center-loss computer-vision deep-learning feature-learning loss-functions python pytorch

Last synced: 29 Oct 2024

https://github.com/NATSpeech/NATSpeech

A Non-Autoregressive Text-to-Speech (NAR-TTS) framework, including official PyTorch implementation of PortaSpeech (NeurIPS 2021) and DiffSpeech (AAAI 2022)

diffsinger diffspeech huggingface portaspeech pytorch speech speech-synthesis tts

Last synced: 07 Aug 2024

https://github.com/kennymckormick/pyskl

A toolbox for skeleton-based action recognition.

action-recognition deep-learning gesture-recognition posec3d pytorch skeleton-based-action-recognition stgcn

Last synced: 29 Oct 2024

https://github.com/GAP-LAB-CUHK-SZ/gaustudio

A Modular Framework for 3D Gaussian Splatting and Beyond

3d-reconstruction 3dgs gaussian-splatting multi-view-reconstruction nerf pytorch surface-reconstruction

Last synced: 04 Aug 2024

https://github.com/huawei-noah/addernet

Code for paper " AdderNet: Do We Really Need Multiplications in Deep Learning?"

convolutional-neural-networks cvpr2020 efficient-inference imagenet pytorch

Last synced: 13 Nov 2024

https://github.com/thudm/cogview2

official code repo for paper "CogView2: Faster and Better Text-to-Image Generation via Hierarchical Transformers"

pretrained-models pytorch text-to-image transformer

Last synced: 14 Nov 2024

https://github.com/KinWaiCheuk/nnAudio

Audio processing by using pytorch 1D convolution network

1d-convolution audio-processing cqt-spectrogram melspectrogram neural-network preprocessing pytorch spectrogram spectrogram-conversion-toolbox stft

Last synced: 05 Aug 2024

https://github.com/kinwaicheuk/nnaudio

Audio processing by using pytorch 1D convolution network

1d-convolution audio-processing cqt-spectrogram melspectrogram neural-network preprocessing pytorch spectrogram spectrogram-conversion-toolbox stft

Last synced: 29 Oct 2024

https://github.com/kmkurn/pytorch-crf

(Linear-chain) Conditional random field in PyTorch.

conditional-random-fields neural-networks pytorch

Last synced: 30 Oct 2024

https://github.com/qubvel/ttach

Image Test Time Augmentation with PyTorch!

augmentation classification computer-vision deep-learning keypoint-detection pytorch segmentation test-time-augmentation tta tta-wrapper

Last synced: 21 Oct 2024

https://github.com/freewym/espresso

Espresso: A Fast End-to-End Neural Speech Recognition Toolkit

asr end-to-end fairseq kaldi python pytorch speech-recognition

Last synced: 05 Nov 2024

https://github.com/fbcotter/pytorch_wavelets

Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet

dtcwt filterbank pytorch wavelet wavelet-transform

Last synced: 30 Oct 2024

https://github.com/pku-alignment/omnisafe

JMLR: OmniSafe is an infrastructural framework for accelerating SafeRL research.

benchmark constraint-rl constraint-satisfaction-problem deep-learning deep-reinforcement-learning machine-learning pytorch reinforcement-learning safe-reinforcement-learning safe-rl saferl safety-critical safety-gym safety-gymnasium

Last synced: 14 Nov 2024

https://github.com/aws/deep-learning-containers

AWS Deep Learning Containers (DLCs) are a set of Docker images for training and serving models in TensorFlow, TensorFlow 2, PyTorch, and MXNet.

aws docker mxnet pytorch sagemaker tensorflow tensorflow2

Last synced: 07 Oct 2024

https://github.com/rusty1s/pytorch_sparse

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations

autograd pytorch sparse sparse-matrices

Last synced: 21 Oct 2024

https://github.com/oneapi-src/oneapi-samples

Samples for Intel® oneAPI Toolkits

ai cpu cuda fortran fpga gpu jupyter oneapi oneccl onedal onednn onedpl onemkl onetbb python pytorch rendering scikit-learn sycl tensorflow

Last synced: 10 Oct 2024

https://github.com/Sharpiless/Yolov5-Deepsort

最新版本yolov5+deepsort目标检测和追踪，能够显示目标类别，支持5.0版本可训练自己数据集

computer-vision deepsort object-detection object-tracking pytorch yolov5

Last synced: 09 Nov 2024

https://github.com/descriptinc/melgan-neurips

GAN-based Mel-Spectrogram Inversion Network for Text-to-Speech Synthesis

deep-learning gans pytorch speech-synthesis

Last synced: 05 Aug 2024

https://github.com/uber/neuropod

A uniform interface to run deep learning models from multiple frameworks

deep-learning deeplearning incubation inference keras machine-learning machinelearning pytorch tensorflow

Last synced: 13 Nov 2024

https://github.com/THUDM/CogView2

official code repo for paper "CogView2: Faster and Better Text-to-Image Generation via Hierarchical Transformers"

pretrained-models pytorch text-to-image transformer

Last synced: 04 Nov 2024

https://github.com/sukebenet/dreampower

DeepNude algorithm fork with improvements.

deepnude dreamnet dreampower fake-nudes gan image-processing opencv photo pytorch

Last synced: 26 Sep 2024

https://github.com/piesposito/blitz-bayesian-deep-learning

A simple and extensible library to create Bayesian Neural Network layers on PyTorch.

bayesian-deep-learning bayesian-layers bayesian-neural-networks pytorch pytorch-implementation pytorch-tutorial

Last synced: 13 Oct 2024

https://github.com/asteroid-team/torch-audiomentations

Fast audio data augmentation in PyTorch. Inspired by audiomentations. Useful for deep learning.

audio audio-data-augmentation audio-effects augmentation data-augmentation deep-learning differentiable-data-augmentation dsp machine-learning music python pytorch sound sound-processing waveform

Last synced: 05 Nov 2024

https://threedle.github.io/text2mesh/

3D mesh stylization driven by a text input in PyTorch

3d computer-graphics differentiable-rendering geometry-processing mesh-generation meshes neural-fields neural-style pytorch

Last synced: 01 Nov 2024

https://github.com/D-X-Y/landmark-detection

Four landmark detection algorithms, implemented in PyTorch.

facial-landmarks keypoint-detection landmark-detection pytorch style-transfer

Last synced: 28 Oct 2024

https://github.com/d-x-y/landmark-detection

Four landmark detection algorithms, implemented in PyTorch.

facial-landmarks keypoint-detection landmark-detection pytorch style-transfer

Last synced: 13 Nov 2024

https://github.com/alexandre01/deepsvg

[NeurIPS 2020] Official code for the paper "DeepSVG: A Hierarchical Generative Network for Vector Graphics Animation". Includes a PyTorch library for deep learning with SVG data.

autoencoder deep-learning deep-svg library machine-learning python pytorch sketch-rnn sketches svg svg-animations svg-vae transformer

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

https://github.com/threedle/text2mesh

3D mesh stylization driven by a text input in PyTorch

3d computer-graphics differentiable-rendering geometry-processing mesh-generation meshes neural-fields neural-style pytorch

Last synced: 27 Oct 2024

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

Gated-Shape CNN for Semantic Segmentation (ICCV 2019)

computer-vision deep-learning iccv2019 nv-tlabs pytorch semantic-boundaries semantic-segmentation

Last synced: 15 Nov 2024

https://github.com/mukosame/zooming-slow-mo-cvpr-2020

Fast and Accurate One-Stage Space-Time Video Super-Resolution (accepted in CVPR 2020)

cvpr cvpr2020 pytorch spatio-temporal super-resolution video video-frame-interpolation video-super-resolution

Last synced: 14 Nov 2024

https://github.com/lhotse-speech/lhotse

Tools for handling speech data in machine learning projects.

ai audio data deep-learning kaldi machine-learning python pytorch speech speech-recognition

Last synced: 08 Aug 2024

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

Gated-Shape CNN for Semantic Segmentation (ICCV 2019)

computer-vision deep-learning iccv2019 nv-tlabs pytorch semantic-boundaries semantic-segmentation

Last synced: 31 Oct 2024

https://github.com/pnnl/neuromancer

Pytorch-based framework for solving parametric constrained optimization problems, physics-informed system identification, and parametric model predictive control.

constrained-optimization control-systems deep-learning differentiable-control differentiable-optimization differentiable-programming dynamical-systems nonlinear-dynamics nonlinear-optimization physics-informed-ml pytorch