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/BIGBALLON/PyTorch-CPP

PyTorch C++ inference with LibTorch

cpp demo imagenet inference libtorch opencv pytorch pytorch-cpp

Last synced: 31 Jul 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: 01 Aug 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/domainadaptation/salad

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

deep-learning domain-adaptation machine-learning pytorch

Last synced: 03 Aug 2024

https://github.com/iMoonLab/MeshNet

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

3d-shapes deep-learning mesh pytorch

Last synced: 31 Jul 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: 02 Aug 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: 31 Jul 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: 24 Sep 2024

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

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

augmentation pytorch pytorch-lightning tensorflow torchbearer

Last synced: 02 Aug 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: 01 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: 26 Sep 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: 02 Aug 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/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: 27 Sep 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/Sierkinhane/mtcnn-pytorch

A face detection algorithm

computer-vision facedetector pytorch

Last synced: 31 Jul 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/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/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: 01 Aug 2024

https://github.com/JUGGHM/PENet_ICRA2021

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

depth-completion pytorch

Last synced: 31 Jul 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: 27 Sep 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 Jul 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: 01 Aug 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: 01 Aug 2024

https://github.com/QUVA-Lab/escnn

Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/

deep-learning equivariance geometric-deep-learning geometry invariant isometries pytorch rotations symmetries

Last synced: 02 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: 02 Aug 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/AppleHolic/source_separation

Deep learning based speech source separation using Pytorch

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

Last synced: 01 Aug 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/bytedance/byteir

A model compilation solution for various hardware

llm llvm mlir mlsys onnx pytorch tensorflow

Last synced: 31 Jul 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

Last synced: 02 Aug 2024

https://github.com/zsef123/EfficientNets-PyTorch

A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."

deep-learning efficientnet pytorch

Last synced: 01 Aug 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: 31 Jul 2024

https://github.com/jw9730/tokengt

[NeurIPS'22] Tokenized Graph Transformer (TokenGT), in PyTorch

equivariance gnn graph hypergraph pytorch self-attention transformer

Last synced: 31 Jul 2024

https://github.com/pangwong/pytorch-multi-label-classifier

A pytorch implemented classifier for Multiple-Label classification

accuracy-curves classifier loss-curves multi-label-classification pytorch visdom

Last synced: 03 Aug 2024

https://github.com/TorchDrift/TorchDrift

Drift Detection for your PyTorch Models

ai drift-detection neural-networks pytorch

Last synced: 02 Aug 2024

https://github.com/frgfm/Holocron

PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne)

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

Last synced: 01 Aug 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: 01 Aug 2024

https://github.com/WangFeng18/3d-gaussian-splatting

Implementation for 3d gaussian splatting

gaussian-splatting nerf pytorch real-time-rendering

Last synced: 04 Aug 2024

https://github.com/tjddus9597/Proxy-Anchor-CVPR2020

Official PyTorch Implementation of Proxy Anchor Loss for Deep Metric Learning, CVPR 2020

cvpr2020 deep-metric-learning image-retrieval proxy-anchor-loss pytorch

Last synced: 12 Aug 2024

https://github.com/mjhydri/BeatNet

BeatNet is state-of-the-art (Real-Time) and Offline joint music beat, downbeat, tempo, and meter tracking system using CRNN and particle filtering. (ISMIR 2021's paper implementation).

beat-tracking beatnet crnn-network dnn-beat-tracking downbeat-tracking joint-model meter-detection online-time-signature-detection particle-filtering pytorch real-time real-time-beat-tracker real-time-downbeat-tracker real-time-tempo state-of-the-art

Last synced: 05 Aug 2024

https://github.com/r9y9/tacotron_pytorch

PyTorch implementation of Tacotron speech synthesis model.

python pytorch speech speech-synthesis tacotron

Last synced: 07 Aug 2024

https://github.com/alexandre01/UltimateLabeling

A multi-purpose Video Labeling GUI in Python with integrated SOTA detector and tracker

computer-vision labeling labeling-tool object-detection object-tracking python pytorch video

Last synced: 02 Aug 2024

https://github.com/alibaba-edu/simple-effective-text-matching-pytorch

A pytorch implementation of the ACL2019 paper "Simple and Effective Text Matching with Richer Alignment Features".

deep-learning nlp pytorch quora-question-pairs snli

Last synced: 01 Aug 2024

https://github.com/Tai-Wang/Depth-from-Motion

[ECCV 2022 oral] Monocular 3D Object Detection with Depth from Motion

3d-detection autonomous-driving monocular pytorch robotics structure-from-motion

Last synced: 31 Jul 2024

https://github.com/edouardoyallon/pyscatwave

Fast Scattering Transform with CuPy/PyTorch

deep-learning pytorch scattering-transform

Last synced: 01 Aug 2024

https://github.com/torch-js/torch-js

Node.js binding for PyTorch.

nodejs pytorch

Last synced: 02 Aug 2024

https://github.com/yinboc/prototypical-network-pytorch

A re-implementation of "Prototypical Networks for Few-shot Learning"

few-shot-learning machine-learning pytorch

Last synced: 30 Jul 2024

https://github.com/bamos/block

An intelligent block matrix library for numpy, PyTorch, and beyond.

deep-learning linear-algebra numpy pytorch

Last synced: 07 Aug 2024

https://github.com/theEricMa/OTAvatar

This is the official repository for OTAvatar: One-shot Talking Face Avatar with Controllable Tri-plane Rendering [CVPR2023].

cvpr2023 deep-learning deepfake face-animation face-reenactment image-animation motion-transfer pose-transfer pytorch talking-head

Last synced: 03 Aug 2024

https://github.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement

A minimum unofficial implementation of the "A Convolutional Recurrent Neural Network for Real-Time Speech Enhancement" (CRN) using PyTorch

cnn cnn-rnn pytorch real-time rnn speech-enhancement speech-processing

Last synced: 05 Aug 2024

https://github.com/megvii-research/FQ-ViT

[IJCAI 2022] FQ-ViT: Post-Training Quantization for Fully Quantized Vision Transformer

imagenet post-training-quantization pytorch quantization vision-transformer

Last synced: 31 Jul 2024

https://github.com/leaderj1001/MobileNetV3-Pytorch

Implementing Searching for MobileNetV3 paper using Pytorch

cifar-10 cifar-100 imagenet mobilenetv3 pytorch searching-for-mobilenetv3

Last synced: 31 Jul 2024

https://github.com/feedly/transfer-nlp

NLP library designed for reproducible experimentation management

framework language-model natural-language-understanding nlp playground pytorch transfer-learning

Last synced: 24 Sep 2024

https://github.com/Bluefog-Lib/bluefog

Distributed and decentralized training framework for PyTorch over graph

asynchronous decentralized deeplearning distributed-computing machine-learning mpi nccl one-sided pytorch

Last synced: 07 Aug 2024

https://github.com/wonjongg/StyleCariGAN

[SIGGRAPH 2021] Official implementation of "StyleCariGAN: Caricature Generation via StyleGAN Feature Map Modulation"

deep-learning pytorch

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

https://github.com/jaxony/ShuffleNet

ShuffleNet in PyTorch. Based on https://arxiv.org/abs/1707.01083

artificial-intelligence convolution deep-learning neural-network pytorch

Last synced: 31 Jul 2024

https://github.com/Luodian/Generalizable-Mixture-of-Experts

GMoE could be the next backbone model for many kinds of generalization task.

deep-learning domain-generalization mixture-of-experts pytorch pytorch-implementation

Last synced: 03 Aug 2024

https://github.com/ksw0306/ClariNet

A Pytorch Implementation of ClariNet

clarinet parallel-wavenet pytorch wavenet

Last synced: 01 Aug 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: 02 Aug 2024

https://github.com/Media-Smart/volksdep

volksdep is an open-source toolbox for deploying and accelerating PyTorch, ONNX and TensorFlow models with TensorRT.

acceleration deploy inference jetson-nano jetson-tx2 jetson-xavier keras onnx python pytorch tensorflow tensorrt

Last synced: 03 Aug 2024

https://github.com/dgkim5360/the-elements-of-statistical-learning-notebooks

Jupyter notebooks for summarizing and reproducing the textbook "The Elements of Statistical Learning" 2/E by Hastie, Tibshirani, and Friedman

jupyter-notebooks machine-learning matplotlib neural-network python3 pytorch scipy statistical-learning textbook

Last synced: 24 Sep 2024

https://github.com/quiver-team/torch-quiver

PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.

distributed-computing geometric-deep-learning gpu-acceleration graph-learning graph-neural-networks pytorch

Last synced: 08 Aug 2024

https://github.com/loeweX/Greedy_InfoMax

Code for the paper: Putting An End to End-to-End: Gradient-Isolated Learning of Representations

deep-learning local-learning pytorch

Last synced: 02 Aug 2024

https://github.com/outcastofmusic/quick-nlp

Pytorch NLP library based on FastAI

fastai nlp-library pytorch seq2seq

Last synced: 03 Aug 2024

https://github.com/aimagelab/show-control-and-tell

Show, Control and Tell: A Framework for Generating Controllable and Grounded Captions. CVPR 2019

caption-generation captioning-images cvpr2019 image-captioning pytorch visual-semantic

Last synced: 31 Jul 2024

https://github.com/DeNA/HandyRL

HandyRL is a handy and simple framework based on Python and PyTorch for distributed reinforcement learning that is applicable to your own environments.

deep-learning distributed-training games machine-learning policy-gradient pytorch reinforcement-learning

Last synced: 01 Aug 2024

https://github.com/lucastabelini/PolyLaneNet

Code for the paper entitled "PolyLaneNet: Lane Estimation via Deep Polynomial Regression" (ICPR 2020)

deep-learning lane-detection pytorch

Last synced: 31 Jul 2024

https://github.com/sunzeyeah/RLHF

Implementation of Chinese ChatGPT

chatgpt deep-learning deepspeed glm nlp pangu pytorch

Last synced: 31 Jul 2024

https://github.com/AssemblyAI-Community/MinImagen

MinImagen: A minimal implementation of the Imagen text-to-image model

deep-learning diffusion-models imagen pytorch super-resolution text-to-image

Last synced: 06 Aug 2024

https://github.com/extreme-bert/extreme-bert

ExtremeBERT is a toolkit that accelerates the pretraining of customized language models on customized datasets, described in the paper “ExtremeBERT: A Toolkit for Accelerating Pretraining of Customized BERT”.

bert deep-learning language-model language-models machine-learning natural-language-processing nlp python pytorch transformer

Last synced: 03 Aug 2024

https://github.com/suyukun666/UFO

Official PyTorch implementation of the “A Unified Transformer Framework for Co-Segmentation, Co-Saliency Detection and Video Salient Object Detection”. (TMM2023)

co-saliency-detection co-segmentation deep-learning inpainting matching pytorch video video-salient-object-detection

Last synced: 02 Aug 2024

https://github.com/merantix-momentum/squirrel-core

A Python library that enables ML teams to share, load, and transform data in a collaborative, flexible, and efficient way :chestnut:

ai cloud-computing collaboration computer-vision cv data-ingestion data-mesh data-science dataops datasets deep-learning distributed jax machine-learning ml natural-language-processing nlp python pytorch tensorflow

Last synced: 01 Aug 2024