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/DerryHub/BEVFormer_tensorrt

BEVFormer inference on TensorRT, including INT8 Quantization and Custom TensorRT Plugins (float/half/half2/int8).

bevformer cuda int8-inference pytorch quantization tensorrt-plugins

Last synced: 31 Jul 2024

https://github.com/wuyifan18/DeepLog

Pytorch Implementation of DeepLog.

anomaly-detection deeplearning pytorch sequence-prediction

Last synced: 01 Aug 2024

https://github.com/operatorai/modelstore

🏬 modelstore is a Python library that allows you to version, export, and save a machine learning model to your filesystem or a cloud storage provider.

data-science keras machine-learning mlops modelstore python-library pytorch s3-storage scikit-learn tensorflow transformer

Last synced: 02 Aug 2024

https://github.com/ReshotAI/gaussian-painters

Gaussian Painters using 3D Gaussian Splatting

3d ai art gaussian-splatting machine-learning nerf painting python pytorch

Last synced: 01 Aug 2024

https://github.com/NeuromorphicProcessorProject/snn_toolbox

Toolbox for converting analog to spiking neural networks (ANN to SNN), and running them in a spiking neuron simulator.

brian brian2 caffe deep-learning deep-neural-networks keras lasagne loihi nest pynn pytorch spiking-neural-networks spinnaker tensorflow

Last synced: 02 Aug 2024

https://github.com/BlackHC/toma

Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory

data-science gpu machine-learning python pytorch

Last synced: 03 Aug 2024

https://github.com/kefirski/pytorch_RVAE

Recurrent Variational Autoencoder that generates sequential data implemented with pytorch

deep-learning nlp python pytorch vae

Last synced: 01 Aug 2024

https://github.com/kakaoenterprise/JORLDY

Repository for Open Source Reinforcement Learning Framework JORLDY

framework gym mlagents pytorch reinforcement-learning

Last synced: 08 Aug 2024

https://github.com/dptech-corp/Uni-Fold

An open-source platform for developing protein models beyond AlphaFold.

alphafold alphafold-multimer deep-learning protein-structure pytorch

Last synced: 02 Aug 2024

https://github.com/kevinmusgrave/pytorch-adapt

Domain adaptation made easy. Fully featured, modular, and customizable.

computer-vision deep-learning domain-adaptation machine-learning pytorch transfer-learning

Last synced: 24 Sep 2024

https://github.com/uber-research/DeepPruner

DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch (ICCV 2019)

iccv2019 patchmatch pytorch real-time stereo-matching stereo-vision

Last synced: 31 Jul 2024

https://github.com/megvii-research/MOTRv2

[CVPR2023] MOTRv2: Bootstrapping End-to-End Multi-Object Tracking by Pretrained Object Detectors

end-to-end multi-object-tracking pytorch transformer

Last synced: 31 Jul 2024

https://github.com/HikariTJU/LD

Localization Distillation for Object Detection (CVPR 2022, TPAMI 2023)

deep-learning detection knowledge-distillation ld object-detection pytorch

Last synced: 01 Aug 2024

https://github.com/davidmascharka/tbd-nets

PyTorch implementation of "Transparency by Design: Closing the Gap Between Performance and Interpretability in Visual Reasoning"

deep-learning machine-learning neural-networks pytorch visual-question-answering visualization vqa

Last synced: 07 Aug 2024

https://github.com/Megvii-CSG/MegReader

A research project for text detection and recognition using PyTorch 1.2.

ctc deep-learning ocr pytorch text-detection text-detection-recognition text-recognition

Last synced: 01 Aug 2024

https://github.com/Haiyang-W/DSVT

[CVPR2023] Official Implementation of "DSVT: Dynamic Sparse Voxel Transformer with Rotated Sets"

3d-object-detection backbone cvpr2023 point-cloud pytorch transformer

Last synced: 31 Jul 2024

https://github.com/talebolano/yolov3-network-slimming

yolov3 network slimming剪枝的一种实现

network pruning pytorch slimming yolo

Last synced: 02 Aug 2024

https://github.com/prigoyal/pytorch_memonger

Experimental ground for optimizing memory of pytorch models

memory-management pytorch

Last synced: 01 Aug 2024

https://github.com/ymcui/PERT

PERT: Pre-training BERT with Permuted Language Model

bert nlp plm pre-trained-model pytorch tensorflow transformers

Last synced: 03 Aug 2024

https://github.com/emidan19/deep-tempest

Restoration for TEMPEST images using deep-learning

deep-learning gnu-radio image-restoration pytorch sdr tempest

Last synced: 08 Aug 2024

https://github.com/square/pysurvival

Open source package for Survival Analysis modeling

deep-learning machine-learning numpy python pytorch survival-analysis

Last synced: 02 Aug 2024

https://github.com/KevinMusgrave/pytorch-adapt

Domain adaptation made easy. Fully featured, modular, and customizable.

computer-vision deep-learning domain-adaptation machine-learning pytorch transfer-learning

Last synced: 01 Aug 2024

https://github.com/vballoli/nfnets-pytorch

NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/

adaptive-gradient-clipping deepmind image-classification nfnets paper pytorch sgd sota

Last synced: 31 Jul 2024

https://github.com/atgambardella/pytorch-es

Evolution Strategies in PyTorch

machinelearning python pytorch

Last synced: 01 Aug 2024

https://github.com/justusschock/shapenet

PyTorch implementation of "Super-Realtime Facial Landmark Detection and Shape Fitting by Deep Regression of Shape Model Parameters" predicting facial landmarks with up to 400 FPS

deep-learning delira facial-keypoints facial-keypoints-cnn facial-landmarks landmark-detection pytorch shape shape-models shape-prediction shape-predictor

Last synced: 01 Aug 2024

https://github.com/ShirAmir/dino-vit-features

Official implementation for the paper "Deep ViT Features as Dense Visual Descriptors".

co-segmentation computer-vision deep-learning dino part-segmentation pytorch semantic-correspondence vision-transformers

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

https://github.com/Pointcept/PointTransformerV2

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

3d-vision point-cloud pytorch

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

https://github.com/quantylab/rltrader

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

deep-learning keras python pytorch reinforcement-learning tensorflow

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

https://github.com/fendouai/pytorch1.0-cn

PyTorch 1.0 官方文档 中文版，欢迎关注微信公众号：磐创AI

pytorch pytorch-nlp pytorch-tutorial

Last synced: 03 Aug 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: 01 Aug 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: 03 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/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/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/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/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/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/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/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/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: 24 Sep 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/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/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/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/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/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: 24 Sep 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/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/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/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://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://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/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/WangFeng18/3d-gaussian-splatting

Implementation for 3d gaussian splatting

gaussian-splatting nerf pytorch real-time-rendering

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

PyTorch Awesome Lists
Awesome-pytorch-list 704 awesome-semantic-segmentation-pytorch 35 Awesome-pytorch-list-CNVersion 692 CV-pretrained-model 103 awesome-colab-notebooks 502 awesome-AutoML-and-Lightweight-Models 150 Awesome-Denoise 164 ai4artists 286 arctic 34 awesome-yolov8-models 43 awesome-cuda-and-hpc 424 awesome-python-machine-learning-resources 1,884 awesome-adaptive-computation 171 awesome-NILM-with-code 47 awesome-dotnet-machine-learning 134 awesome-Multi-Document-Summarization 56 PyTorch-Guide 616 awesome-fastai 42 awesome-ai-tools 499
PyTorch Categories
Pytorch & related libraries 228 Pytorch & related libraries｜Pytorch & 相关库 218 Research 197 Tutorials 192 文本数据和NLP 182 Blogs 164 Uncategorized 161 Frameworks 142 数据可视化 130 机器学习框架 123 1.) Neural Architecture Search 113 图像数据与CV 106 Blogposts :sparkle: 103 Learning 102 模型的可解释性 100 Interactive Learning & Tools 96 超参数优化和AutoML 92 Tutorials, books, & examples 82 Tutorials & books & examples｜教程 & 书籍 & 示例 81 工作流程和实验跟踪 80