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/wuyifan18/DeepLog

Pytorch Implementation of DeepLog.

anomaly-detection deeplearning pytorch sequence-prediction

Last synced: 05 Nov 2024

https://github.com/cezannec/capsule_net_pytorch

Readable implementation of a Capsule Network as described in "Dynamic Routing Between Capsules" [Hinton et. al.]

capsule-network pytorch pytorch-tutorial

Last synced: 02 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: 28 Oct 2024

https://github.com/vanderschaarlab/synthcity

A library for generating and evaluating synthetic tabular data for privacy, fairness and data augmentation.

data-augmentation fairness-ml generative-model machine-learning privacy pytorch synthetic-data tabular-data

Last synced: 11 Nov 2024

https://github.com/itailang/SampleNet

Differentiable Point Cloud Sampling (CVPR 2020 Oral)

cvpr2020 deep-learning geometry-processing neural-network point-cloud pytorch sampling tensorlow

Last synced: 28 Oct 2024

https://github.com/atcold/pytorch-cortexnet

PyTorch implementation of the CortexNet predictive model

deep-learning predictive-modeling pytorch self-supervised unsupervised-learning video

Last synced: 31 Oct 2024

https://github.com/Atcold/pytorch-CortexNet

PyTorch implementation of the CortexNet predictive model

deep-learning predictive-modeling pytorch self-supervised unsupervised-learning video

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

https://github.com/Owen-Liuyuxuan/visualDet3D

Official Repo for Ground-aware Monocular 3D Object Detection for Autonomous Driving / YOLOStereo3D: A Step Back to 2D for Efficient Stereo 3D Detection

computer-vision icra monocular-3d-detection object-detection pytorch stereo-3d-detection

Last synced: 28 Oct 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: 04 Nov 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: 11 Oct 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: 05 Nov 2024

https://github.com/shi-labs/natten

Neighborhood Attention Extension. Bringing attention to a neighborhood near you!

cuda neighborhood-attention pytorch

Last synced: 08 Nov 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: 05 Nov 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/kakaoenterprise/JORLDY

Repository for Open Source Reinforcement Learning Framework JORLDY

framework gym mlagents pytorch reinforcement-learning

Last synced: 08 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: 02 Nov 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/prigoyal/pytorch_memonger

Experimental ground for optimizing memory of pytorch models

memory-management pytorch

Last synced: 28 Oct 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: 11 Oct 2024

https://github.com/dvl-tum/mot_neural_solver

Official PyTorch implementation of "Learning a Neural Solver for Multiple Object Tracking" (CVPR 2020 Oral).

graph-neural-networks multi-object-tracking pytorch

Last synced: 28 Oct 2024

https://github.com/amanchadha/iseebetter

iSeeBetter: Spatio-Temporal Video Super Resolution using Recurrent-Generative Back-Projection Networks | Python3 | PyTorch | GANs | CNNs | ResNets | RNNs | Published in Springer Journal of Computational Visual Media, September 2020, Tsinghua University Press

cnn convolutional-neural-network convolutional-neural-networks gan gans generative-adversarial-network psnr pytorch recurrent-neural-networks residual-networks resnet resnets rnn self-supervised-learning srgan ssim super-resolution unsupervised-learning video-processing video-super-resolution

Last synced: 05 Nov 2024

https://github.com/pointcept/pointtransformerv2

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

3d-vision point-cloud pytorch

Last synced: 08 Nov 2024

https://github.com/lucidrains/axial-attention

Implementation of Axial attention - attending to multi-dimensional data efficiently

artificial-intelligence attention-mechanism deep-learning pytorch

Last synced: 30 Oct 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: 10 Nov 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: 28 Oct 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 Oct 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: 26 Oct 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: 04 Nov 2024

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

Evolution Strategies in PyTorch

machinelearning python pytorch

Last synced: 02 Nov 2024

https://github.com/utkuozbulak/pytorch-cnn-adversarial-attacks

Pytorch implementation of convolutional neural network adversarial attack techniques

adversarial-example fast-gradient-sign fooling-images gradient-ascent pytorch pytorch-tutorial

Last synced: 09 Nov 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: 28 Oct 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/talebolano/yolov3-network-slimming

yolov3 network slimming剪枝的一种实现

network pruning pytorch slimming yolo

Last synced: 09 Nov 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/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/francescosaveriozuppichini/pytorch-deep-learning-template

A Pytorch Computer Vision template to quick start your next project! 🚀🚀

computer-vision deep-learning python pytorch

Last synced: 31 Oct 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: 28 Oct 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: 31 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/Pointcept/PointTransformerV2

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

3d-vision point-cloud pytorch

Last synced: 28 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/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/quantylab/rltrader

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

deep-learning keras python pytorch reinforcement-learning tensorflow

Last synced: 11 Oct 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/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/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: 31 Oct 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/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/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: 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: 31 Oct 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/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/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/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/xplip/pixel

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

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

Last synced: 31 Oct 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/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/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/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: 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: 02 Aug 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/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/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/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/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/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/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/flagopen/flaggems

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

pytorch triton triton-kernels

Last synced: 07 Nov 2024

PyTorch Awesome Lists
Awesome-pytorch-list 704 awesome-semantic-segmentation-pytorch 35 Awesome-pytorch-list-CNVersion 692 awesome-colab-notebooks 525 CV-pretrained-model 103 awesome-AutoML-and-Lightweight-Models 150 openstl 43 Awesome-Denoise 164 ai4artists 286 arctic 34 awesome-cuda-and-hpc 427 awesome-yolov8-models 43 awesome-python-machine-learning-resources 1,884 awesome-adaptive-computation 173 awesome-NILM-with-code 47 awesome-dotnet-machine-learning 134 awesome-Multi-Document-Summarization 56 PyTorch-Guide 616 awesome-max-mojo 183 awesome-fastai 42 awesome-ai-tools 511 free-pytorch-resources 72
PyTorch Categories
Blogs 450 Frameworks 276 Pytorch & related libraries 231 Pytorch & related libraries｜Pytorch & 相关库 220 Research 198 Tutorials 192 文本数据和NLP 182 Uncategorized 161 数据可视化 131 机器学习框架 123 1.) Neural Architecture Search 122 Blogposts :sparkle: 119 图像数据与CV 106 Learning Resources 105 Learning 101 模型的可解释性 100 Tutorials, books, & examples 95 Interactive Learning & Tools 94 Tutorials & books & examples｜教程 & 书籍 & 示例 93 超参数优化和AutoML 93