Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
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: 2025-02-16 00:24:26 UTC
- JSON Representation
https://github.com/marload/aquvitae
Knowledge Distillation Toolkit
deep-learning knowledge-distillation light-weight machine-learning model-compression pytorch tensorflow
Last synced: 02 Dec 2024
https://github.com/oaid/tengine-convert-tools
Tengine Convert Tool supports converting multi framworks' models into tmfile that suitable for Tengine-Lite AI framework.
artificial-intelligence cnn dnn mxnet onnx pytorch tensorflow
Last synced: 05 Nov 2024
https://github.com/sebgao/ctensor
A super light-weight deep learning library based on NumPy in PyTorch fashion.
deep-learning deep-neural-networks numpy pytorch
Last synced: 26 Dec 2024
https://github.com/sony/san
PyTorch implementation of slicing adversarial network (SAN)
deep-generative-model gan generative-model machine-learning optimal-transport pytorch sliced-wasserstein-distance wasserstein-distance
Last synced: 15 Nov 2024
https://github.com/Coderx7/SimpleNet
This repository contains the architectures, Models, logs, etc pertaining to the SimpleNet Paper (Lets keep it simple: Using simple architectures to outperform deeper architectures )
cnn-model cnn-pytorch convolutional-neural-networks deep-learning fast-net image-classification imagenet pytorch simplenet
Last synced: 05 Nov 2024
https://github.com/huangtinglin/MixGCF
MixGCF: An Improved Training Method for Graph Neural Network-based Recommender Systems, KDD2021
graph-neural-network information-retrieval negative-sampling network-embedding pytorch recommender-system
Last synced: 28 Nov 2024
https://github.com/autostronomy/astrophot
A fast, flexible, automated, and differentiable astronomical image 2D forward modelling tool for precise parallel multi-wavelength photometry
astronomy python pytorch science-research scientific-computing
Last synced: 15 Feb 2025
https://github.com/d-li14/ghostnet.pytorch
73.6% GhostNet 1.0x pre-trained model on ImageNet
ghostnet imagenet mobilenetv3 pretrained-models pytorch reproduction
Last synced: 07 Nov 2024
https://github.com/pcyin/pytorch_basic_nmt
A simple yet strong implementation of neural machine translation in pytorch
deep-learning natural-language-processing neural-machine-translation pytorch pytorch-implmention
Last synced: 01 Dec 2024
https://github.com/kevinzakka/ibc
A PyTorch implementation of Implicit Behavioral Cloning
behavioral-cloning deep-learning energy-based-model pytorch robotics
Last synced: 28 Oct 2024
https://github.com/lantiga/pytorch2c
A Python module for compiling PyTorch graphs to C
compiled-graphs deep-learning graph python pytorch tensor
Last synced: 06 Jan 2025
https://github.com/drivendataorg/zamba
A Python package for identifying 42 kinds of animals, training custom models, and estimating distance from camera trap videos
animals camera-traps chimps cli conservation deep-learning ecology gpu jungle machine-learning neural-network python pytorch pytorch-lightning video-processing videos
Last synced: 18 Feb 2025
https://github.com/wassname/attentive-neural-processes
implementing "recurrent attentive neural processes" to forecast power usage (w. LSTM baseline, MCDropout)
anp anp-rnn attention attentive-neural-processes neural-processes prediction pytorch rnn uncertainty
Last synced: 08 Nov 2024
https://github.com/avinashpaliwal/deep-slomo
Official PyTorch implementation of "Deep Slow Motion Video Reconstruction with Hybrid Imaging System" (TPAMI)
convolutional-neural-networks deep-learning deep-slomo dual-cameras frame-interpolation hybrid-imaging pytorch pytorch-implementation slomo slow-motion super-resolution video-frame-interpolation video-super-resolution
Last synced: 05 Dec 2024
https://github.com/alldbi/SuperMix
Pytorch implementation of CVPR2021 paper: SuperMix: Supervising the Mixing Data Augmentation
augmentation cvpr2021 deep-learning distillation pytorch saliency-detection supervised
Last synced: 13 Nov 2024
https://github.com/TuragaLab/DECODE
This is the official implementation of our publication "Deep learning enables fast and dense single-molecule localization with high accuracy" (Nature Methods)
deep-learning gpu high-density localization-microscopy microscopy pytorch smlm
Last synced: 14 Nov 2024
https://github.com/amdegroot/pytorch-containers
Torch Containers simplified in PyTorch
autograd complex-architectures pytorch pytorch-containers torch
Last synced: 14 Nov 2024
https://github.com/djsaunde/spiketorch
Experiments with spiking neural networks (SNNs) in PyTorch. See https://github.com/BINDS-LAB-UMASS/bindsnet for the successor to this project.
machine-learning neurons python pytorch snn spiking-neural-networks
Last synced: 24 Dec 2024
https://github.com/layerai-archive/sdk
Metadata store for Production ML
collaboration data-science data-versioning deep-learning experiment-tracking hyperparameter-optimization hyperparameter-tuning keras machine-learning mlops model-versioning python pytorch reinforcement-learning sklearn tensorflow
Last synced: 19 Jan 2025
https://github.com/rubenwiersma/hsn
Code for SIGGRAPH paper CNNs on Surfaces using Rotation-Equivariant Features
3d convolutional-neural-networks geometric-deep-learning pytorch pytorch-geometric siggraph
Last synced: 16 Feb 2025
https://github.com/ashishpatel26/vision-transformer-keras-tensorflow-pytorch-examples
Tensorflow implementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformer
example keras pytorch tensorflow tensorflow2 vision-transformer
Last synced: 19 Nov 2024
https://github.com/snap-research/graphless-neural-networks
[ICLR 2022] Code for Graph-less Neural Networks: Teaching Old MLPs New Tricks via Distillation (GLNN)
deep-learning distillation efficient-inference gnn graph-algorithm graph-neural-networks knowledge-distillation pytorch scalability
Last synced: 13 Feb 2025
https://github.com/interdigitalinc/compressai-vision
CompressAI-Vision helps you design, test and compare Video Compression for Machines pipelines. Compression methods can be either pulled from custom AI-based modules from CompressAI or traditional codecs such as H.266/VVC.
compression computer-vision deep-learning machine-to-machine-communication pytorch video-compression
Last synced: 14 Feb 2025
https://github.com/bluemirrors/cvu
Computer Vision deployment tools for dummies and experts. CVU aims at making CV pipelines easier to build and consistent around platforms, devices, and models.
computer-vision object-detection onnxruntime pytorch tensorflow2 tensorrt yolov5
Last synced: 10 Feb 2025
https://github.com/Haiyang-W/CAGroup3D
[NeurIPS2022] This is the official code of "CAGroup3D: Class-Aware Grouping for 3D Object Detection on Point Clouds".
3d-object-detection pytorch scannet-dataset sunrgbd
Last synced: 28 Oct 2024
https://github.com/buxihuo/OW-YOLO
Detect known and unknown objects in the open world(具有区分已知与未知能力的全新检测器))
autolabel deep-learning object-detection open-world open-world-detection pytorch unknown yolo yolov5 yolov8
Last synced: 09 Nov 2024
https://github.com/lartpang/hdfnet
(ECCV 2020) Hierarchical Dynamic Filtering Network for RGB-D Salient Object Detection
eccv eccv2020 image-processing imageprocessing imagesegmentation pytorch saliency saliency-detection salient-object-detection segmentation
Last synced: 17 Jan 2025
https://github.com/M-Nauta/ProtoTree
ProtoTrees: Neural Prototype Trees for Interpretable Fine-grained Image Recognition, published at CVPR2021
computer-vision cvpr2021 decision-trees deep-neural-networks explainability explainable-ai explainable-ml fine-grained-classification fine-grained-visual-categorization interpretability interpretable-deep-learning interpretable-machine-learning pytorch
Last synced: 15 Nov 2024
https://github.com/ailln/fuzi-nlu
🤖️ 聊天机器人——夫子的「自然语言理解」模块
chatbot intent-classification joint-nlu nlu pytorch question-answering seq2seq seq2seq-chatbot slot-filling socketio
Last synced: 18 Nov 2024
https://github.com/oxwhirl/treeqn
deep-learning deep-reinforcement-learning openai-gym pytorch reinforcement-learning
Last synced: 08 Nov 2024
https://github.com/chrischoy/open-ucn
The first fully convolutional metric learning for geometric/semantic image correspondences.
correspondence fully-convolutional-networks image-analysis metric-learning pytorch
Last synced: 07 Nov 2024
https://github.com/mberr/torch-ppr
(Personalized) Page-Rank computation using PyTorch
gpu pagerank personalized-pagerank pytorch
Last synced: 18 Nov 2024
https://github.com/graykode/modelsummary
All Model summary in PyTorch similar to `model.summary()` in Keras
Last synced: 20 Dec 2024
https://github.com/encounter1997/sfa
Official Implementation of "Exploring Sequence Feature Alignment for Domain Adaptive Detection Transformers"
detection-transformer domain-adaptation pytorch
Last synced: 14 Nov 2024
https://github.com/thunlp/prompt-transferability
On Transferability of Prompt Tuning for Natural Language Processing
nlp parameter-efficient-learning parameter-efficient-tuning pretrained-language-model pretrained-language-models pretrained-models prompt prompt-tuning pytorch transfer-learning
Last synced: 10 Nov 2024
https://github.com/lorenmt/minimal-isaac-gym
A Minimal Example of Isaac Gym with DQN and PPO.
Last synced: 03 Nov 2024
https://github.com/j-min/dropouts
PyTorch Implementations of Dropout Variants
bayesian-neural-networks dropout gaussian-dropout local-reparametrization-trick pytorch variational-dropout variational-inference
Last synced: 19 Nov 2024
https://github.com/necla-ml/Diff-JPEG
Official and maintained implementation of the paper "Differentiable JPEG: The Devil is in the Details" [WACV 2024].
differentiable-codec differentiable-jpeg image-codec image-compression jpeg pytorch ste surrogate wacv
Last synced: 04 Nov 2024
https://github.com/hzxie/rmnet
The official implementation of "Efficient Regional Memory Network for Video Object Segmentation". (Xie et al., CVPR 2021)
cvpr2021 pytorch segmentation video-object-segmentation
Last synced: 06 Dec 2024
https://github.com/r00tman/eventnerf
Neural Radiance Fields from a Single Colour Event Camera [CVPR 2023]
computer-vision cvpr2023 event-camera eventnerf mesh nerf neural-radiance-fields neural-rendering pytorch volume-rendering
Last synced: 09 Dec 2024
https://github.com/patrick-kidger/fasterneuraldiffeq
Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
controlled-differential-equations deep-learning deep-neural-networks differential-equations dynamical-systems machine-learning neural-differential-equations numerical-analysis numerical-methods ordinary-differential-equations pytorch
Last synced: 06 Dec 2024
https://github.com/cxhernandez/molencoder
Molecular AutoEncoder in PyTorch
autoencoder computational-chemistry deep-learning python pytorch
Last synced: 27 Oct 2024
https://github.com/HaochengWan/PVT
PVT: Point-Voxel Transformer for 3D Deep Learning
kitti modelnet40 point-cloud point-cloud-processing pvt pytorch s3dis shapenet sota
Last synced: 28 Oct 2024
https://github.com/bshall/acoustic-model
Acoustic models 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/traveller59/torch2trt
convert torch module to tensorrt network or tvm function
Last synced: 17 Nov 2024
https://github.com/naity/image_tabular
Integrate image and tabular data for deep learning
deep-learning fastai image-classification pytorch tabular-data
Last synced: 15 Nov 2024
https://github.com/edisonleeeee/greatx
A graph reliability toolbox based on PyTorch and PyTorch Geometric (PyG).
adversarial-attacks distribution-shift graph-convolutional-networks graph-neural-networks graph-reliability-toolbox inherent-noise pytorch pytorch-geometric
Last synced: 16 Feb 2025
https://github.com/hsankesara/voxelmorph-pytorch
An unofficial PyTorch implementation of VoxelMorph- An unsupervised 3D deformable image registration method
cvpr2018 deformable-image-registration ieeetran image-registration pytorch spatial-transformer-networks voxel-morph
Last synced: 27 Oct 2024
https://github.com/filapro/unidet3d
[AAAI2025] UniDet3D: Multi-dataset Indoor 3D Object Detection
3d-object-detection mmdetection3d pytorch s3dis scannet
Last synced: 17 Feb 2025
https://github.com/qq456cvb/PRIN
Pointwise Rotation-Invariant Network (AAAI 2020)
pytorch rotation-invariant-features segmentation
Last synced: 28 Oct 2024
https://github.com/ofnote/tryongan
TryOnGAN: Unofficial Implementation
deep-learning fashion gan generative-adversarial-network pytorch viton
Last synced: 16 Nov 2024
https://github.com/ai4iacts/hexagdly
Process hexagonally sampled data with PyTorch
deep-learning hexagonal-grids pytorch
Last synced: 01 Nov 2024
https://github.com/comet-ml/issue-tracking
Questions, Help, and Issues for Comet ML
comet-ml keras machine-learning pytorch scikit-learn tenserflow
Last synced: 14 Feb 2025
https://github.com/abap34/ml-lecture
東京工業大学 traP Kaggle班「機械学習講習会」の資料
deeplearning education machine-learning pytorch textbook
Last synced: 05 Nov 2024
https://github.com/hkchengrex/scribble-to-mask
[CVPR 2021] MiVOS - Scribble to Mask module
computer-vision cvpr2021 deep-learning interactive-segmentation pytorch segmentation
Last synced: 10 Nov 2024
https://github.com/d-li14/dhm
[CVPR 2020] Dynamic Hierarchical Mimicking Towards Consistent Optimization Objectives
cvpr2020 deeply-supervised imagenet knowledge-distillation pretrained-models pytorch
Last synced: 07 Nov 2024
https://github.com/rasbt/pytorch-memory-optim
This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog post.
deep-learning llm memory-optimization pytorch vision vit
Last synced: 01 Nov 2024
https://github.com/unity-technologies/datasetinsights
Synthetic Dataset Insights
computer-vision machine-learning pytorch synthetic-datasets unity-simulation
Last synced: 27 Dec 2024
https://github.com/MachineLearningLifeScience/stochman
Algorithms for computations on random manifolds made easier
machine-learning manifold python pytorch riemannian-geometry
Last synced: 12 Nov 2024
https://github.com/yukkyo/pytorch-filterresponsenormalizationlayer
PyTorch implementation of "Filter Response Normalization Layer: Eliminating Batch Dependence in the Training of Deep Neural Networks"
batchnorm2d catalyst deep-neural-networks frn python pytorch
Last synced: 27 Jan 2025
https://github.com/tohrusky/final2x-core
core for Final2x
artificial-intelligence ccrestoration computer-vision cross-platform image-processing low-level-vision ncnn opencv python3 pytorch super-resolution
Last synced: 15 Feb 2025
https://github.com/anthony-wang/CrabNet
Predict materials properties using only the composition information!
attention attention-mechanism machine-learning materials-informatics materials-science materials-screening neural-networks predict-materials-properties pytorch scikit-learn self-attention transformer
Last synced: 14 Nov 2024
https://github.com/scitator/run-skeleton-run
Reason8.ai PyTorch solution for NIPS RL 2017 challenge
actor-critic ddpg ddpg-agent nips nips-2017 physics-based ppo pytorch pytorch-solution reinforcement-learning tensorflow trpo
Last synced: 07 Jan 2025
https://github.com/Atharva-Phatak/torchflare
TorchFlare is a simple, beginner-friendly, and easy-to-use PyTorch Framework train your models effortlessly.
deep-learning deep-neural-networks gans generative-adversarial-network image-classfication image-classification image-processing image-segmentation infrastructure kaggle machine-learning natural-language-processing python pytorch pytorch-model self-supervised-learning text-classification
Last synced: 15 Nov 2024
https://github.com/dbaranchuk/memory-efficient-maml
Memory efficient MAML using gradient checkpointing
deep-learning gpu gradient-checkpointing maml meta-learning pytorch
Last synced: 07 Nov 2024
https://github.com/vitae-transformer/vitae-transformer-scene-text-detection
A comprehensive list [Hi-SAM@TPAMI'24, GoMatching@NeurIPS'24, DeepSolo(++)@ CVPR'23, DPText-DETR@AAAI'23, I3CL@IJCV'22] of our research works related to scene text detection, spotting, etc., including papers, codes.
deep-learning ocr pytorch scene-text-detection scene-text-spotting vision-transformer
Last synced: 17 Feb 2025
https://github.com/borhanMorphy/fastface
Light Face Detection using PyTorch Lightning
edge-ai facedetection pytorch pytorch-lightning
Last synced: 13 Jan 2025
https://github.com/neptune-ai/examples
📝 Examples of how to use Neptune for different use cases and with various MLOps tools
compare dashboard experiment fastai jupyter keras metadata ml mlops optuna pandas pipelines pytorch sklearn store tensorflow tracker tracking visualization xgboost
Last synced: 20 Feb 2025
https://github.com/daemon/pytorch-pcen
PyTorch reimplementation of per-channel energy normalization for audio.
Last synced: 09 Dec 2024
https://github.com/d4l3k/torchdrive
torch models and primitives for birdseye view and voxel models for self driving tasks.
Last synced: 18 Feb 2025
https://github.com/svishnu88/TGS-SaltNet
Kaggle | 21st place solution for TGS Salt Identification Challenge
computer-vision deep-learning fastai kaggle-competition pytorch
Last synced: 27 Nov 2024
https://github.com/yuanzhoulvpi2017/quick_sentence_transformers
sentence-transformers to onnx 让sbert模型推理效率更快
bert pytorch sentence-transformers transformer transformers
Last synced: 08 Nov 2024
https://github.com/hellloxiaotian/deep-learning-on-image-denoising-an-overview
Deep Learning on Image Denoising: An overview (Neural Networks, 2020)
blind-denoising cnn denoising-network denoising-survery image-denoising keras keras-cnn low-level low-level-vision matconvnet python pytorch pytorch-cnn real-noise survery tensorflow tensorflow-cnn theano
Last synced: 11 Nov 2024
https://github.com/borhanmorphy/fastface
Light Face Detection using PyTorch Lightning
edge-ai facedetection pytorch pytorch-lightning
Last synced: 21 Jan 2025
https://github.com/Erlemar/wheat
Wheat Detection challenge on Kaggle
deep-learning hydra kaggle-competition object-detection pytorch pytorch-lightning torchvision
Last synced: 15 Nov 2024
https://github.com/necla-ml/diff-jpeg
Official and maintained implementation of the paper "Differentiable JPEG: The Devil is in the Details" [WACV 2024].
differentiable-codec differentiable-jpeg image-codec image-compression jpeg pytorch ste surrogate wacv
Last synced: 09 Feb 2025
https://github.com/toshas/torch_truncnorm
Truncated Normal Distribution in PyTorch
distributions pytorch truncated-normal
Last synced: 25 Jan 2025
https://github.com/giotto-ai/giotto-deep
Deep learning made topological.
computational-geometry deep-learning image-processing nlp pytorch tda
Last synced: 19 Feb 2025
https://github.com/kefirski/pytorch_TDNN
Time Delayed NN implemented in pytorch
cnn deep-learning python pytorch tdnn
Last synced: 14 Nov 2024
https://github.com/lucasbotang/gradnorm
PyTorch implementation of the GradNorm
deep-learning gradnorm multitask-learning python pytorch
Last synced: 17 Feb 2025
https://github.com/huggingface/optimum-graphcore
Blazing fast training of 🤗 Transformers on Graphcore IPUs
fine-tuning graphcore machine-learning pytorch training transformers
Last synced: 02 Nov 2024
https://github.com/asteroid-team/asteroid-filterbanks
Asteroid's filterbanks :rocket:
asteroid-filterbanks audio audio-processing deep-learning filterbanks python3 pytorch
Last synced: 16 Feb 2025
https://github.com/theos-ai/easy-yolov7
This a clean and easy-to-use implementation of YOLOv7 in PyTorch, made with ❤️ by Theos AI.
computer-vision deep-learning machine-learning neural-networks object-detection python pytorch yolov5 yolov7
Last synced: 09 Nov 2024
https://github.com/writer/fitbert
Use BERT to Fill in the Blanks
bert fill-in-the-blank nlp python pytorch
Last synced: 23 Jan 2025
https://github.com/ajhamdi/mvtorch
a Pytorch library for multi-view 3D understanding and generation
3d 3d-deep-learning deep learning multi-view multi-view-geometry multi-view-learning nerf pytorch pytorch3d
Last synced: 17 Feb 2025
https://github.com/anshulpaigwar/Attentional-PointNet
Attentional-PointNet is Deep-Neural-Network architecture for 3D object detection in point clouds
3d-object-detection attention-mechanism autonomous-vehicles object-detection point-cloud pointnet pytorch pytorch-implmention
Last synced: 28 Oct 2024
https://github.com/acl21/deep-active-learning-pytorch
A PyTorch toolkit with 8 popular deep active learning query methods implemented.
deep-active-learning image-classification pytorch toolkit
Last synced: 05 Nov 2024
https://github.com/yusugomori/deeplearning-keras-tf2-torch
詳解ディープラーニング 第2版
deep-learning keras pytorch tensorflow2
Last synced: 07 Nov 2024
https://github.com/kevinzakka/form2fit
[ICRA 2020] Train generalizable policies for kit assembly with self-supervised dense correspondence learning.
deep-learning perception pytorch robotics self-supervised-learning
Last synced: 28 Oct 2024
https://github.com/erogol/fftnet
FFTNet vocoder implementation
deep-learning fftnet pytorch text2speech vocoder
Last synced: 06 Jan 2025
https://github.com/lyeoni/pretraining-for-language-understanding
Pre-training of Language Models for Language Understanding
language-model language-modeling language-understanding nlp pytorch pytorch-tutorial
Last synced: 06 Nov 2024
https://github.com/spcl/daceml
A Data-Centric Compiler for Machine Learning
compiler cuda deep-learning fpga high-performance-computing machine-learning pytorch
Last synced: 06 Nov 2024
https://github.com/sony/soundctm
Pytorch implementation of SoundCTM
audio-generation diffusion-models pytorch text-to-audio
Last synced: 17 Feb 2025
https://github.com/rose-stl-lab/torchts
Time series forecasting with PyTorch
deep-learning machine-learning pytorch pytorch-lightning time-series
Last synced: 15 Feb 2025