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/Pointcept/PointTransformerV2

[NeurIPS'22] An official PyTorch implementation of PTv2.

3d-vision point-cloud pytorch

Last synced: 28 Oct 2024

https://github.com/apple/ml-gmpi

[ECCV 2022, Oral Presentation] Official PyTorch implementation of GMPI

3d eccv2022 generative-adversarial-network generative-model multiplane-images neural-rendering pytorch

Last synced: 07 Oct 2024

https://github.com/ZhangGongjie/Meta-DETR

[T-PAMI 2022] Meta-DETR for Few-Shot Object Detection: Official PyTorch Implementation

computer-vision few-shot-object-detection meta-learning object-detection pytorch

Last synced: 15 Oct 2024

https://github.com/gordicaleksa/pytorch-deepdream

PyTorch implementation of DeepDream algorithm (Mordvintsev et al.). Additionally I've included playground.py to help you better understand basic concepts behind the algo.

deep-dream-tutorial deep-learning deep-learning-tutorial deepdream machine-learning python pytorch

Last synced: 11 Nov 2024

https://github.com/Algolzw/image-restoration-sde

Image Restoration with Mean-Reverting Stochastic Differential Equations, ICML 2023. Winning solution of the NTIRE 2023 Image Shadow Removal Challenge.

deblurring dehazing denoising deraining diffusion-models image-restoration inpainting pytorch score-based stochastic-differential-equations super-resolution

Last synced: 30 Oct 2024

https://github.com/chokkan/mlnote

機械学習帳

matplotlib numpy python pytorch scikit-learn

Last synced: 11 Nov 2024

https://github.com/ekzhang/fastseg

📸 PyTorch implementation of MobileNetV3 for real-time semantic segmentation, with pretrained weights & state-of-the-art performance

aspp cityscapes computer-vision deep-learning deeplabv3 edge-computing efficientnet kitti-dataset mapillary-vistas-dataset mobilenetv3 pytorch semantic-segmentation

Last synced: 26 Oct 2024

https://github.com/dmmiller612/sparktorch

Train and run Pytorch models on Apache Spark.

apache-spark deep-learning distributed-computing inference pipelines pytorch sparktorch

Last synced: 12 Oct 2024

https://github.com/quantylab/rltrader

파이썬과 케라스를 이용한 딥러닝/강화학습 주식투자 - 퀀트 투자, 알고리즘 트레이딩을 위한 최첨단 해법 입문 (개정판)

deep-learning keras python pytorch reinforcement-learning tensorflow

Last synced: 11 Oct 2024

https://github.com/golsun/DialogRPT

EMNLP 2020: "Dialogue Response Ranking Training with Large-Scale Human Feedback Data"

conversational-ai dataset dialog dialog-datasets dialogpt gpt-2 human-feedback-data pretrained-models pytorch transformers

Last synced: 30 Oct 2024

https://github.com/v-iashin/SpecVQGAN

Source code for "Taming Visually Guided Sound Generation" (Oral at the BMVC 2021)

audio audio-generation bmvc evaluation-metrics gan melgan multi-modal pytorch transformer vas vggsound video video-features video-understanding vqvae

Last synced: 06 Nov 2024

https://github.com/modularml/max

A collection of sample programs, notebooks, and tools which highlight the power of the MAX Platform

ai genai llama2 llm ml onnx pytorch

Last synced: 10 Oct 2024

https://github.com/SamsungLabs/adaptis

[ICCV19] AdaptIS: Adaptive Instance Selection Network, https://arxiv.org/abs/1909.07829

adaptis cityscapes instance-segmentation mapillary-vistas-dataset ms-coco mxnet panoptic-segmentation pytorch

Last synced: 08 Nov 2024

https://github.com/jacobgil/pytorch-explain-black-box

PyTorch implementation of Interpretable Explanations of Black Boxes by Meaningful Perturbation

blackbox deep-learning deeplearning pytorch visualization

Last synced: 14 Nov 2024

https://github.com/amirbar/detreg

Official implementation of the CVPR 2022 paper "DETReg: Unsupervised Pretraining with Region Priors for Object Detection".

deep-learning object-detection pytorch unsupervised-learning

Last synced: 14 Nov 2024

https://github.com/zhangmozhe/Deep-Exemplar-based-Video-Colorization

The source code of CVPR 2019 paper "Deep Exemplar-based Video Colorization".

colorization computer-vision deep-learning gan generative-adversarial-network image-generation image-processing old-photo pytorch

Last synced: 08 Aug 2024

https://github.com/domainadaptation/salad

A toolbox for domain adaptation and semi-supervised learning. Contributions welcome.

deep-learning domain-adaptation machine-learning pytorch

Last synced: 14 Nov 2024

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

Unofficial PyTorch Implementation for HifiFace (https://arxiv.org/abs/2106.09965)

face-swapping pytorch pytorch-lightning wandb

Last synced: 11 Nov 2024

https://github.com/leopard-ai/betty

Betty: an automatic differentiation library for generalized meta-learning and multilevel optimization

artificial-intelligence autodiff automatic-differentiation bilevel-optimization deep-learning hyperparameter-optimization implicit-differentiation machine-learning meta-learning multilevel-optimization neural-architecture-search pytorch reinforcement-learning

Last synced: 10 Nov 2024

https://github.com/BIGBALLON/PyTorch-CPP

PyTorch C++ inference with LibTorch

cpp demo imagenet inference libtorch opencv pytorch pytorch-cpp

Last synced: 27 Oct 2024

https://github.com/lukecavabarrett/pna

Implementation of Principal Neighbourhood Aggregation for Graph Neural Networks in PyTorch, DGL and PyTorch Geometric

dgl graph-machine-learning graph-neural-networks pytorch pytorch-geometric

Last synced: 08 Aug 2024

https://github.com/flagopen/flaggems

FlagGems is an operator library for large language models implemented in Triton Language.

pytorch triton triton-kernels

Last synced: 14 Nov 2024

https://github.com/xplip/pixel

Research code for pixel-based encoders of language (PIXEL)

deep-learning deep-neural-networks language-model machine-learning nlp pytorch

Last synced: 14 Nov 2024

https://github.com/iMoonLab/MeshNet

MeshNet: Mesh Neural Network for 3D Shape Representation (AAAI 2019)

3d-shapes deep-learning mesh pytorch

Last synced: 28 Oct 2024

https://github.com/bshall/hubert

HuBERT content encoders for: A Comparison of Discrete and Soft Speech Units for Improved Voice Conversion

pytorch representation-learning speech voice-conversion

Last synced: 13 Nov 2024

https://github.com/keonlee9420/PortaSpeech

PyTorch Implementation of PortaSpeech: Portable and High-Quality Generative Text-to-Speech

deep-neural-networks fastspeech generative-model hifi-gan high-quality mel-gan neural-tts non-ar non-autoregressive normalizing-flows portable-tts pytorch speech-synthesis text-to-speech tts vae

Last synced: 07 Aug 2024

https://github.com/bshall/zerospeech

VQ-VAE for Acoustic Unit Discovery and Voice Conversion

acoustic-features pytorch speech-synthesis voice-conversion vq-vae zerospeech

Last synced: 13 Nov 2024

https://github.com/stared/thinking-in-tensors-writing-in-pytorch

Thinking in tensors, writing in PyTorch (a hands-on deep learning intro)

deep-learning pytorch pytorch-tutorial tensor

Last synced: 09 Nov 2024

https://github.com/hancyran/RepSurf

[CVPR 2022 Oral] Official implementation for "Surface Representation for Point Clouds"

cvpr2022 modelnet40 point-cloud pytorch s3dis scannet scanobjectnn sunrgbd

Last synced: 28 Oct 2024

https://github.com/szymonmaszke/torchdatasets

PyTorch dataset extended with map, cache etc. (tensorflow.data like)

cache concatenate dataset disk filter library map pytorch tensorflow tf-data torch

Last synced: 01 Nov 2024

https://github.com/pytorch/torchx

TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.

airflow aws-batch components deep-learning distributed-training kubernetes machine-learning pipelines python pytorch ray slurm

Last synced: 07 Oct 2024

https://github.com/xiuyu-li/q-diffusion

[ICCV 2023] Q-Diffusion: Quantizing Diffusion Models.

ddim diffusion-models model-compression post-training-quantization pytorch quantization stable-diffusion

Last synced: 14 Nov 2024

https://github.com/ecs-vlc/FMix

Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'

augmentation pytorch pytorch-lightning tensorflow torchbearer

Last synced: 13 Nov 2024

https://github.com/vinhkhuc/PyTorch-Mini-Tutorials

Minimal tutorials for PyTorch

deep-learning machine-learning python pytorch

Last synced: 02 Nov 2024

https://github.com/AntixK/PyTorch-Model-Compare

Compare neural networks by their feature similarity

cka deep-learning feature-extraction imagenet neural-networks pip pytorch torch-cka transformers

Last synced: 03 Aug 2024

https://github.com/microsoft/nn-meter

A DNN inference latency prediction toolkit for accurately modeling and predicting the latency on diverse edge devices.

deep-learning deep-neural-networks edge-ai edge-computing efficient-model inference latency machine-learning neural-architecture-search onnx-models python pytorch tensorflow-models

Last synced: 10 Oct 2024

https://github.com/yuanyuanli85/Fast_Human_Pose_Estimation_Pytorch

Pytorch Code for CVPR2019 paper "Fast Human Pose Estimation" https://arxiv.org/abs/1811.05419

knowledge-distillation pose-estimation pytorch stacked-hourglass-networks

Last synced: 05 Nov 2024

https://github.com/bknyaz/graph_nn

Graph Classification with Graph Convolutional Networks in PyTorch (NeurIPS 2018 Workshop)

gcn graph graph-classification graph-convolutional-networks graph-unet multigraph pooling pytorch

Last synced: 12 Nov 2024

https://github.com/asyml/vision-transformer-pytorch

Pytorch version of Vision Transformer (ViT) with pretrained models. This is part of CASL (https://casl-project.github.io/) and ASYML project.

pytorch vision-transformer

Last synced: 12 Nov 2024

https://github.com/kevinnan-teen/intelligent-traffic-based-on-cv

本项目是基于计算机视觉的端到端交通路口智能监控系统．采用的设计架构由SRS流媒体服务器、GPU服务器、Local客户端三部分组成．可将远端视频流通过rtmp协议传输到流媒体服务器，然后经过目标检测等一些列算法对视频进行分析，最后在本地客户端查看分析结果．项目主要用Python实现，流媒体服务器采用开源的SRS实时视频服务器搭建，GPU服务器使用YOLO模型实现道路目标如人、车、交通灯等物体的识别，同时还支持并发连接．客户端基于PyQt5进行开发．

python pytorch srs yolov3

Last synced: 14 Nov 2024

https://github.com/pmixer/SASRec.pytorch

PyTorch(1.6+) implementation of https://github.com/kang205/SASRec

pytorch recommender-system sasrec sequential-models

Last synced: 05 Nov 2024

https://github.com/gordicaleksa/pytorch-neural-style-transfer

Reconstruction of the original paper on neural style transfer (Gatys et al.). I've additionally included reconstruction scripts which allow you to reconstruct only the content or the style of the image - for better understanding of how NST works.

deep-learning deep-learning-tutorial machine-learning neural-style-transfer neural-style-transfer-pytorch neural-style-transfer-tutorial non-photorealistic-rendering python pytorch style-transfer

Last synced: 11 Nov 2024

https://github.com/zjunlp/openue

[EMNLP 2020] OpenUE: An Open Toolkit of Universal Extraction from Text

bert event-extraction intent-classification named-entity-recognition natural-language-processing nlp nlp-extraction-tasks openue pytorch relation-extraction slot-filling triple-extraction

Last synced: 03 Aug 2024

https://github.com/sony/model_optimization

Model Compression Toolkit (MCT) is an open source project for neural network model optimization under efficient, constrained hardware. This project provides researchers, developers, and engineers advanced quantization and compression tools for deploying state-of-the-art neural networks.

deep-learning deep-neural-networks edge-ai machine-learning network-compression network-quantization neural-network optimizer ptq pytorch qat quantization tensorflow

Last synced: 08 Nov 2024

https://github.com/wohlert/generative-query-network-pytorch

Generative Query Network (GQN) in PyTorch as described in "Neural Scene Representation and Rendering"

deepmind generative-models gqn pytorch

Last synced: 07 Aug 2024

https://github.com/frgfm/Holocron

PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Other additions: AdEMAMix

ademamix computer-vision cspdarknet53 darknet deep-learning object-detection pytorch resnet rexnet tridentnet unet-image-segmentation yolo yolov4

Last synced: 05 Nov 2024

https://github.com/bethgelab/model-vs-human

Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)

benchmark pytorch robustness tensorflow toolbox

Last synced: 03 Aug 2024

https://github.com/benedekrozemberczki/AttentionWalk

A PyTorch Implementation of "Watch Your Step: Learning Node Embeddings via Graph Attention" (NeurIPS 2018).

attention deep-learning deepwalk graph-attention graph-neural-networks graph-representation-learning implicit-factorization machine-learning matrix-factorization neurips neurips-2018 nips node2vec pytorch sklearn structural-attention tensorflow torch walklet word2vec

Last synced: 04 Nov 2024

https://github.com/frgfm/holocron

PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Other additions: AdEMAMix

ademamix computer-vision cspdarknet53 darknet deep-learning object-detection pytorch resnet rexnet tridentnet unet-image-segmentation yolo yolov4

Last synced: 11 Oct 2024

https://github.com/Sierkinhane/mtcnn-pytorch

A face detection algorithm

computer-vision facedetector pytorch

Last synced: 28 Oct 2024

https://github.com/choyingw/synergynet

3DV 2021: Synergy between 3DMM and 3D Landmarks for Accurate 3D Facial Geometry

2d-3d 3d 3d-face-alignment 3d-face-reconstruction 3dv2021 3dvision computer-graphics computer-vision deep-neural-networks facial-keypoints facial-landmarks head-pose-estimation pytorch

Last synced: 06 Nov 2024

https://github.com/sciml/scimlbenchmarks.jl

Scientific machine learning (SciML) benchmarks, AI for science, and (differential) equation solvers. Covers Julia, Python (PyTorch, Jax), MATLAB, R

ai ai-for-science benchmark dae differential-equations differentialequations jax julia matlab nerual-differential-equations neural-ode ode partial-differential-equations pde python pytorch scientific-machine-learning sciml sde

Last synced: 14 Nov 2024

https://github.com/benedekrozemberczki/attentionwalk

A PyTorch Implementation of "Watch Your Step: Learning Node Embeddings via Graph Attention" (NeurIPS 2018).

Last synced: 10 Oct 2024

https://github.com/neuralmagic/sparsify

ML model optimization product to accelerate inference.

automl computer-vision deep-learning-accelerator image-classification inference-performance keras object-detection onnx pruning pytorch quantization smaller-models sparsification-recipe sparsify tensorflow

Last synced: 08 Nov 2024

https://github.com/damitkwr/ESRNN-GPU

PyTorch GPU implementation of the ES-RNN model for time series forecasting

deep-forecasting deep-learning es-rnn forecasting pytorch time-series-forecasting

Last synced: 02 Nov 2024

https://github.com/stonesjtu/pytorch-nce

The Noise Contrastive Estimation for softmax output written in Pytorch

importance-sampling language-model nce nce-criterion pytorch softmax speedup

Last synced: 09 Nov 2024

https://github.com/Junjue-Wang/LoveDA

[NeurIPS 2021] LoveDA: A Remote Sensing Land-Cover Dataset for Domain Adaptive Semantic Segmentation

domain-adaptation pytorch remote-sensing semantic-segmentation

Last synced: 03 Aug 2024

https://github.com/JUGGHM/PENet_ICRA2021

ICRA 2021 "Towards Precise and Efficient Image Guided Depth Completion"

depth-completion pytorch

Last synced: 28 Oct 2024

https://github.com/yqyao/FCOS_PLUS

Some improvements (center sample) about FCOS (FCOS: Fully Convolutional One-Stage Object Detection).

anchor-free computer-vision fcos object-detection one-stage pytorch

Last synced: 05 Aug 2024

https://github.com/bfortuner/pytorch-kaggle-starter

Pytorch starter kit for Kaggle competitions

deep-learning jupyter-notebook kaggle kaggle-competition machine-learning python3 pytorch pytorch-tutorial

Last synced: 08 Nov 2024

https://github.com/Kyubyong/pytorch_exercises

exercises neural-networks pytorch

Last synced: 12 Nov 2024

https://github.com/gasteigerjo/ppnp

PPNP & APPNP models from "Predict then Propagate: Graph Neural Networks meet Personalized PageRank" (ICLR 2019)

deep-learning gcn gnn graph-algorithms graph-classification graph-neural-networks machine-learning pagerank pytorch tensorflow

Last synced: 08 Nov 2024

https://github.com/nimarb/pytorch_influence_functions

This is a PyTorch reimplementation of Influence Functions from the ICML2017 best paper: Understanding Black-box Predictions via Influence Functions by Pang Wei Koh and Percy Liang.

deep-learning influence-functions pytorch pytorch-implementation

Last synced: 13 Nov 2024

https://github.com/rlgraph/rlgraph

RLgraph: Modular computation graphs for deep reinforcement learning

deep-learning deep-reinforcement-learning dqn machine-learning neural-networks ppo pytorch reinforcement-learning tensorflow

Last synced: 02 Nov 2024

https://github.com/chenxi116/PNASNet.pytorch

PyTorch implementation of PNASNet-5 on ImageNet

automl deep-learning imagenet neural-architecture-search pytorch

Last synced: 31 Oct 2024

https://github.com/bshall/ZeroSpeech

VQ-VAE for Acoustic Unit Discovery and Voice Conversion

acoustic-features pytorch speech-synthesis voice-conversion vq-vae zerospeech

Last synced: 11 Nov 2024

https://github.com/krasserm/fairseq-image-captioning

Transformer-based image captioning extension for pytorch/fairseq

fairseq image-captioning pytorch transformer

Last synced: 14 Nov 2024

https://github.com/kyubyong/pytorch_exercises

exercises neural-networks pytorch

Last synced: 10 Nov 2024

https://github.com/lromul/argus-freesound

Kaggle | 1st place solution for Freesound Audio Tagging 2019

audio classification deep-learning kaggle pytorch tagging

Last synced: 14 Nov 2024

https://github.com/toodef/neural-pipeline

Neural networks training pipeline based on PyTorch

deep-learning image-classification image-segmentation neural-networks object-detection pipeline pytorch supervised-learning training-pipeline

Last synced: 04 Nov 2024

https://github.com/asahi417/lm-question-generation

Multilingual/multidomain question generation datasets, models, and python library for question generation.

bart nlp pytorch question-answering question-generation t5

Last synced: 04 Nov 2024

https://github.com/lightning-ai/dl-fundamentals

Deep Learning Fundamentals -- Code material and exercises

ai deep-learning machine-learning python pytorch

Last synced: 14 Nov 2024

https://github.com/twitter-research/graph-neural-pde

Graph Neural PDEs

deep-learning graph-convolutional-networks graph-neural-networks partial-differential-equations pytorch

Last synced: 30 Oct 2024

https://vita-group.github.io/NeuralLift-360/

[CVPR 2023, Highlight] "NeuralLift-360: Lifting An In-the-wild 2D Photo to A 3D Object with 360° Views", Dejia Xu, Yifan Jiang, Peihao Wang, Zhiwen Fan, Yi Wang, Zhangyang Wang

image-to-3d nerf neural-radiance-fields pytorch stable-diffusion text-to-3d

Last synced: 30 Oct 2024

https://github.com/myungsub/CAIN

Source code for AAAI 2020 paper "Channel Attention Is All You Need for Video Frame Interpolation"

aaai2020 channel-attention dataset deep-convolutional-networks frame-interpolation pytorch video-frame-interpolation

Last synced: 08 Aug 2024

https://github.com/liuruoze/mini-alphastar

(JAIR'2022) A mini-scale reproduction code of the AlphaStar program. Note: the original AlphaStar is the AI proposed by DeepMind to play StarCraft II. JAIR = Journal of Artificial Intelligence Research.

deep-learning deep-neural-networks deep-reinforcement-learning gaming imitation-learning mini-alphastar multi-agent-systems pytorch reinforcement-learning sc2replay self-playing-bot starcraft-ii-bot starcraft2 starcraft2-ai supervised-learning

Last synced: 14 Nov 2024

https://github.com/AppleHolic/source_separation

Deep learning based speech source separation using Pytorch

audio deep-learning pytorch source-separation speech speech-separation

Last synced: 02 Nov 2024

https://github.com/IBM/aihwkit

IBM Analog Hardware Acceleration Kit

ai analog-devices cuda neural-networks pytorch

Last synced: 03 Aug 2024

https://github.com/jzi040941/PercepNet

Unofficial implementation of PercepNet: A Perceptually-Motivated Approach for Low-Complexity, Real-Time Enhancement of Fullband Speech

pytorch speech-enhancement

Last synced: 03 Aug 2024

https://github.com/yinboc/dgp

Rethinking Knowledge Graph Propagation for Zero-Shot Learning, in CVPR 2019

knowledge-graph machine-learning pytorch zero-shot-learning

Last synced: 08 Nov 2024

https://github.com/bytedance/byteir

A model compilation solution for various hardware

llm llvm mlir mlsys onnx pytorch tensorflow

Last synced: 27 Oct 2024

https://github.com/dionhaefner/pyhpc-benchmarks

A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python :rocket:

benchmarks cupy gpu high-performance-computing jax parallel-computing python pytorch tensorflow

Last synced: 29 Oct 2024

https://github.com/lrjconan/LanczosNetwork

Lanczos Network, Graph Neural Networks, Deep Graph Convolutional Networks, Deep Learning on Graph Structured Data, QM8 Quantum Chemistry Benchmark, ICLR 2019

benchmark graph-convolutional-networks graph-neural-networks iclr2019 lanczos-network pytorch