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: 2024-07-29 13:55:27 UTC
- JSON Representation
https://github.com/MaxHalford/pytorch-resample
🎲 Iterable dataset resampling in PyTorch
imbalanced-learning oversampling pytorch resampling undersampling
Last synced: 03 Aug 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: 31 Jul 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: 02 Aug 2024
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: 31 Jul 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: 31 Jul 2024
https://github.com/lorenmt/minimal-isaac-gym
A Minimal Example of Isaac Gym with DQN and PPO.
Last synced: 01 Aug 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: 03 Aug 2024
https://github.com/cxhernandez/molencoder
Molecular AutoEncoder in PyTorch
autoencoder computational-chemistry deep-learning python pytorch
Last synced: 01 Aug 2024
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: 03 Aug 2024
https://github.com/qq456cvb/PRIN
Pointwise Rotation-Invariant Network (AAAI 2020)
pytorch rotation-invariant-features segmentation
Last synced: 31 Jul 2024
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: 03 Aug 2024
https://github.com/abap34/ml-lecture
東京工業大学 traP Kaggle班「機械学習講習会」の資料
deeplearning education machine-learning pytorch textbook
Last synced: 01 Aug 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: 03 Aug 2024
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: 07 Aug 2024
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: 03 Aug 2024
https://github.com/kefirski/pytorch_TDNN
Time Delayed NN implemented in pytorch
cnn deep-learning python pytorch tdnn
Last synced: 03 Aug 2024
https://github.com/huggingface/optimum-graphcore
Blazing fast training of 🤗 Transformers on Graphcore IPUs
fine-tuning graphcore machine-learning pytorch training transformers
Last synced: 01 Aug 2024
https://github.com/ai4iacts/hexagdly
Process hexagonally sampled data with PyTorch
deep-learning hexagonal-grids pytorch
Last synced: 01 Aug 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: 01 Aug 2024
https://github.com/PortPy-Project/PortPy
Opensource radiation treatment planning system in Python [AAPM'23]
artificial-intelligence cancer deep-learning dicom mathematical-modelling medical-physics optimization python pytorch radiation-oncology radiation-treatment-planning radiotherapy
Last synced: 02 Aug 2024
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: 31 Jul 2024
https://github.com/Erlemar/wheat
Wheat Detection challenge on Kaggle
deep-learning hydra kaggle-competition object-detection pytorch pytorch-lightning torchvision
Last synced: 03 Aug 2024
https://github.com/kefirski/pytorch_Highway
Highway network implemented in pytorch
highway-network python pytorch
Last synced: 03 Aug 2024
https://github.com/asteroid-team/asteroid-filterbanks
Asteroid's filterbanks :rocket:
asteroid-filterbanks audio audio-processing deep-learning filterbanks python3 pytorch
Last synced: 05 Aug 2024
https://github.com/MachineLearningLifeScience/stochman
Algorithms for computations on random manifolds made easier
machine-learning manifold python pytorch riemannian-geometry
Last synced: 02 Aug 2024
https://github.com/chongruo/pytorch-HED
An implementation of HED in pytorch
Last synced: 03 Aug 2024
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: 31 Jul 2024
https://github.com/val-iisc/sketch-parse
Code, demos and data for SketchParse (a neural network for sketch segmentation). Paper:
deep-learning pytorch semantic-segmentation sketches
Last synced: 12 Aug 2024
https://github.com/guyyariv/TempoTokens
This repo contains the official PyTorch implementation of: Diverse and Aligned Audio-to-Video Generation via Text-to-Video Model Adaptation
ai-art audio-to-video audio-visual deep-learning diffusion-models generative-ai modelscope pytorch video-synthesis
Last synced: 05 Aug 2024
https://github.com/li012589/NeuralRG
Pytorch source code for arXiv paper Neural Network Renormalization Group, a generative model using variational renormalization group and normalizing flow.
ising-model normalizing-flow pytorch renormalization-group
Last synced: 05 Aug 2024
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: 02 Aug 2024
https://github.com/dwromero/wavelet_networks
Code repository of the paper "Wavelet Networks: Scale-Translation Equivariant Learning From Raw Time-Series, TMLR" https://arxiv.org/abs/2006.05259
cnns equivariance group-convolution pytorch time-series
Last synced: 02 Aug 2024
https://github.com/eva-n27/BERT-for-Chinese-Question-Answering
bert pytorch question-answering
Last synced: 01 Aug 2024
https://github.com/2014gaokao/pedestrian-attribute-recognition-with-GCN
Pytorch implementation of pedestrian attribute recognition with graph convolutional network
Last synced: 01 Aug 2024
https://github.com/HolyWu/vs-realesrgan
Real-ESRGAN function for VapourSynth
pytorch real-esrgan tensorrt vapoursynth
Last synced: 01 Aug 2024
https://github.com/pietrolesci/energizer
An active learning library for Pytorch based on Pytorch-Lightning.
active-learning deep-learning python pytorch pytorch-lightning
Last synced: 03 Aug 2024
https://github.com/MaverickPeter/DiSCO-pytorch
Differentiable Scan Context with Orientation
icra2021 place-recognition pytorch robotics
Last synced: 31 Jul 2024
https://github.com/cogitare-ai/cogitare
🔥 Cogitare - A Modern, Fast, and Modular Deep Learning and Machine Learning framework for Python
deep-learning framework machine-learning modular-deep-learning pytorch toolbox toolchain
Last synced: 03 Aug 2024
https://github.com/alishdipani/Neural-Style-Transfer-Audio
This is PyTorch Implementation of Neural Style Transfer Algorithm which is modified for Audios.
audio-processing convolutional-neural-networks fourier-transform neural-style-transfer pytorch pytorch-implmention
Last synced: 07 Aug 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: 01 Aug 2024
https://github.com/inferno-pytorch/speedrun
Research code need not be ugly.
deep-learning neural-networks pytorch research
Last synced: 01 Aug 2024
https://github.com/m-tassano/dvdnet
DVDnet: A Simple and Fast Network for Deep Video Denoising
cnn cnn-architecture deep-learning denoising python3 pytorch video
Last synced: 01 Aug 2024
https://github.com/tingiskhan/pyfilter
Particle filtering and sequential parameter inference in Python
bayesian-inference particle-filter pytorch sequential-monte-carlo time-series
Last synced: 02 Aug 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: 01 Aug 2024
https://github.com/xxxnell/spatial-smoothing
(ICML 2022) Official PyTorch implementation of “Blurs Behave Like Ensembles: Spatial Smoothings to Improve Accuracy, Uncertainty, and Robustness”.
bayesian-deep-learning bayesian-neural-networks neural-network pytorch robustness uncertainty uncertainty-estimation uncertainty-quantification
Last synced: 03 Aug 2024
https://github.com/sahagobinda/GPM
Official [ICLR] Code Repository for "Gradient Projection Memory for Continual Learning"
computer-vision continual-learning deep-learning optimizaiton pytorch
Last synced: 03 Aug 2024
https://github.com/csinva/imodelsX
Scikit-learn friendly library to interpret, and prompt-engineer text datasets using large language models.
ai deep-learning explainability huggingface interpretability language-model machine-learning ml natural-language-processing natural-language-understanding neural-network pytorch scikit-learn text text-classification transformer-models xai
Last synced: 02 Aug 2024
https://github.com/wandering007/nasnet-pytorch
A neat pytorch implementation of nasnet and the ported weights from tensorflow
Last synced: 31 Jul 2024
https://github.com/LukasHedegaard/pytorch-benchmark
Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption
benchmark deep-learning flops gpu jetson python pytorch timing-analysis
Last synced: 09 Aug 2024
https://github.com/imirzadeh/stable-continual-learning
Towards increasing stability of neural networks for continual learning: https://arxiv.org/abs/2006.06958.pdf (NeurIPS'20)
catastrophic-forgetting continual-learning deep-learning lifelong-learning pytorch
Last synced: 03 Aug 2024
https://github.com/sevagh/demucs.cpp
C++17 port of Demucs v3 (hybrid) and v4 (hybrid transformer) models with ggml and Eigen3
blas demixing demucs eigen3 ggml guitar music-source-separation piano pytorch
Last synced: 31 Jul 2024
https://github.com/gmalivenko/gluon2pytorch
Gluon to PyTorch deep neural network model converter
converter darknet deep-learning deep-neural-networks densenet densenet-pytorch dual-path-networks dualpath-network gluon machine-learning mxnet nasnet onnx pytorch pytorch-converter resnet
Last synced: 01 Aug 2024
https://github.com/pyrates-neuroscience/PyRates
Open-source, graph-based Python code generator and analysis toolbox for dynamical systems (pre-implemented and custom models). Most pre-implemented models belong to the family of neural population models.
code-generation delayed-differential-equation differential-equations dynamical-systems fortran90 julia matlab network-simulator neural-networks numpy parameter-search python pytorch scientific-computing scientific-research simulations tensorflow
Last synced: 02 Aug 2024
https://github.com/nerox8664/gluon2pytorch
Gluon to PyTorch deep neural network model converter
converter darknet deep-learning deep-neural-networks densenet densenet-pytorch dual-path-networks dualpath-network gluon machine-learning mxnet nasnet onnx pytorch pytorch-converter resnet
Last synced: 28 Aug 2024
https://github.com/Project-DC/pygeneses
A PyTorch based DeepRL framework to train and study artificial species in bio-inspired environments.
artificial-agents artificial-intelligence deep-learning hacktoberfest pytorch reinforcement-learning
Last synced: 03 Aug 2024
https://github.com/BlackHC/batchbald_redux
Reusable BatchBALD implementation
active-learning batchbald machine-learning nbdev neurips-2019 pytorch
Last synced: 03 Aug 2024
https://github.com/vocalpy/vak
A neural network framework for researchers studying acoustic communication
animal-communication animal-vocalizations bioacoustic-analysis bioacoustics birdsong python python3 pytorch spectrograms speech-processing torch torchvision vocalizations
Last synced: 03 Aug 2024
https://github.com/desimone/pytorch-cat-vs-dogs
example kaggle machine-learning pytorch
Last synced: 03 Aug 2024
https://github.com/benedekrozemberczki/SINE
A PyTorch Implementation of "SINE: Scalable Incomplete Network Embedding" (ICDM 2018).
aane asne deep-learning deepwalk dimensionality-reduction gensim graph-embedding grarep machine-learning network-embedding networkx node-embedding node2vec pytorch sklearn tadw tensorflow torch unsupervised-learning walklets
Last synced: 01 Aug 2024
https://github.com/benedekrozemberczki/TENE
A sparsity aware implementation of "Enhanced Network Embedding with Text Information" (ICPR 2018).
asne attributed-embedding deep-learning deepwalk feature-extraction graph-convolution graph-embedding graph-neural-network implicit-factorization machine-learning matrix-decomposition matrix-factorization network-embedding node-classification node-embedding node2vec pattern-recognition pytorch tadw tensorflow
Last synced: 01 Aug 2024
https://github.com/Uehwan/3-D-Scene-Graph
3D scene graph generator implemented in Pytorch.
3d-models 3d-scene-graph computer-vision deep-learning deeplearning environmental-modelling intelligence pytorch robot robotics scene-graph scenegraph
Last synced: 01 Aug 2024
https://github.com/keonlee9420/VAENAR-TTS
PyTorch Implementation of VAENAR-TTS: Variational Auto-Encoder based Non-AutoRegressive Text-to-Speech Synthesis.
duration glow neural-tts non-ar non-autoregressive pytorch self-attention speech-synthesis text-to-speech transforer tts unsupervised-duration unsupervised-learning vae
Last synced: 08 Aug 2024
https://github.com/chenqi1126/SIPE
[CVPR 2022] Self-supervised Image-specific Prototype Exploration for Weakly Supervised Semantic Segmentation
cvpr cvpr2022 pytorch semantic-segmentation weakly-supervised-segmentation
Last synced: 03 Aug 2024
https://github.com/oalieno/asm2vec-pytorch
Unofficial implementation of asm2vec using pytorch ( with GPU acceleration )
asm2vec gpu-acceleration machine-learning neural-language-processing python pytorch unofficial
Last synced: 03 Aug 2024
https://github.com/kabouzeid/point2vec
Self-Supervised Representation Learning on Point Clouds (GCPR 2023 | T4V Workshop @ CVPR 2023)
lightning machine-learning point-cloud pytorch self-supervised-learning transformer vision vision-transformer
Last synced: 31 Jul 2024
https://github.com/nolanzzz/mtmct
Design and Implementation of a Multi-Target Multi-Camera Tracking Solution
cuda deep-learning detection machine-learning opencv python pytorch reidentification research-project resnet tracking
Last synced: 31 Jul 2024
https://github.com/prajjwal1/fluence
A deep learning library based on Pytorch focussed on low resource language research and robustness
attention deep-learning nlp pytorch transformers
Last synced: 07 Aug 2024
https://github.com/jcwleo/mario_rl
a2c actor-critic curiosity-driven deep-learning icm ppo pytorch reinforcement-learning supermariobros
Last synced: 01 Aug 2024
https://github.com/janosh/tensorboard-reducer
Reduce multiple PyTorch TensorBoard runs to new event (or CSV) files.
averaging csv machine-learning pytorch reducer tensorboard tensorboard-pytorch tensorboard-reducer
Last synced: 05 Aug 2024
https://github.com/d-li14/regnet.pytorch
PyTorch-style and human-readable RegNet with a spectrum of pre-trained models
imagenet neural-architecture-search pretrained-models pytorch regnet resnext
Last synced: 03 Aug 2024
https://github.com/NikolasEnt/Lyft-Perception-Challenge
The 4th place and the fastest solution of the Lyft Perception Challenge (Image semantic segmentation with PyTorch)
computer-vision deep-learning linknet pytorch sdcnd self-driving-car semantic-segmentation
Last synced: 10 Aug 2024
https://github.com/yassersouri/pytorch-deep-sets
PyTorch re-implementation of parts of "Deep Sets" (NIPS 2017)
deep-learning machine-learning nips-2017 pytorch
Last synced: 31 Jul 2024
https://github.com/h3lio5/linguistic-style-transfer-pytorch
Implementation of "Disentangled Representation Learning for Non-Parallel Text Style Transfer(ACL 2019)" in Pytorch
acl adversarial-learning disentangled-representations disentanglement latent-representations natural-language-generation pytorch style-transfer text-style-transfer variational-autoencoder
Last synced: 07 Aug 2024
https://github.com/ibaiGorordo/Ultrafast-Lane-Detection-Inference-Pytorch-
Example scripts for the detection of lanes using the ultra fast lane detection model in Pytorch.
lane-detection lane-lines-detection lane-segmentation python pytorch self-driving-car
Last synced: 31 Jul 2024
https://github.com/kiudee/cs-ranking
Context-sensitive ranking and choice in Python with PyTorch
choice-model context-aware deep-learning discrete-choice learning-to-rank machine-learning neural-networks object-ranking pytorch ranking tensorflow
Last synced: 05 Aug 2024
https://github.com/Little-Podi/GRM
[CVPR'23] The official PyTorch implementation of our CVPR 2023 paper: "Generalized Relation Modeling for Transformer Tracking".
attention-mechanism cvpr2023 object-tracking pytorch single-object-tracking tracking vision-transformer visual-tracking
Last synced: 01 Aug 2024
https://github.com/polyaxon/polyaxon-examples
Code for tutorials and examples
deep-learning deeplearning horovod jupyter-notebook keras kubeflow kubernetes machine-learning mlops mpi mpijob mxnet nerual-network polyaxon pytorch pytorchjob scikit-learn tensorflow tfjob
Last synced: 05 Aug 2024
https://github.com/DensoITLab/TeachAugment
Official Implementation of TeachAugment: Data Augmentation Optimization Using Teacher Knowledge (CVPR2022, Oral)
classification cvpr2022 deeplearning pytorch
Last synced: 31 Jul 2024
https://github.com/KaiyangZhou/ssdg-benchmark
Benchmarks for semi-supervised domain generalization.
deep-learning domain-generalization neural-networks pytorch semi-supervised-learning
Last synced: 03 Aug 2024
https://github.com/ziplab/EcoFormer
[NeurIPS 2022 Spotlight] This is the official PyTorch implementation of "EcoFormer: Energy-Saving Attention with Linear Complexity"
classification efficient-transformers neurips-2022 pytorch vision-transformer
Last synced: 02 Aug 2024
https://github.com/H2SO4T/ARES
Black-box tool that uses Deep Reinforcement Learning to test and explore Android applications
android android-testing appium appium-android apple-silicon applesilicon black-box-testing deep-learning deep-reinforcement-learning pytorch reinforcement-learning
Last synced: 31 Jul 2024
https://github.com/shashankvkt/AlignMixup_CVPR22
Official Implementation of AlignMixup - CVPR 2022
alignment augmentation cvpr2022 deep-learning mixup optimal-transport pytorch
Last synced: 02 Aug 2024
https://github.com/leehanchung/cs182
Berkeley CS182/282A Designing, Visualizing and Understanding Deep Neural Networks
berkeley cnn cs182 cs231n cs231n-assignment natural-language-processing nlp pytorch reinforcement-learning tensorflow transformer
Last synced: 31 Jul 2024
https://github.com/matifali/dockerdl
Deep Learning Docker Image
cuda deep-learning docker jupyter numpy python pytorch tensorflow
Last synced: 31 Jul 2024
https://github.com/moskomule/pytorch.rl.learning
for learning reinforcement learning using PyTorch.
pytorch reinforcement-learning
Last synced: 03 Aug 2024
https://github.com/TissueImageAnalytics/cerberus
One Model is All You Need: Multi-Task Learning Enables Simultaneous Histology Image Segmentation and Classification
computational-pathology deep-learning multi-task-learning python pytorch
Last synced: 02 Aug 2024
https://github.com/kadirnar/yolov5-strongsort
Minimal PyTorch implementation of YOLOv5 and StrongSort
artificial-intelligence computer-vision deep-learning deep-sort object-detection object-tracker pytorch strongsort
Last synced: 02 Aug 2024
https://github.com/Fangyh09/Image2LMDB
Convert image folder to lmdb, adapted from Efficient-PyTorch
Last synced: 03 Aug 2024
https://github.com/roysubhankar/dwt-domain-adaptation
Code for paper "Unsupervised Domain Adaptation using Feature-Whitening and Consensus Loss" (CVPR 2019)
cvpr2019 domain-adaptation pytorch whitening
Last synced: 03 Aug 2024
https://github.com/pytorch-labs/torchfix
TorchFix - a linter for PyTorch-using code with autofix support
flake8 flake8-plugin linter python pytorch static-analysis static-code-analysis
Last synced: 03 Aug 2024
https://github.com/saforem2/l2hmc-qcd
Application of the L2HMC algorithm to simulations in lattice QCD.
deep-learning deepspeed gauge-theory hamiltonian-monte-carlo hmc horovod hydra lattice lattice-qcd machine-learning mcmc monte-carlo pytorch tensorflow
Last synced: 31 Jul 2024
https://github.com/LaihoE/did-it-spill
Check if you have training samples in your test set
computer-vision data-science deep-learning pytorch semantic-similarity time-series
Last synced: 02 Aug 2024
https://github.com/sony/DiffRoll
PyTorch implementation of DiffRoll, a diffusion-based generative automatic music transcription (AMT) model
automatic-music-transcription deep-generative-model diffusion generative-model inpainting machine-learning music-generation pytorch
Last synced: 31 Jul 2024
https://github.com/XuyangBai/KPConv.pytorch
PyTorch reimplementation for "KPConv: Flexible and Deformable Convolution for Point Clouds" https://arxiv.org/abs/1904.08889
3d-point-clouds geometric-deep-learning kpconv pytorch
Last synced: 31 Jul 2024
https://github.com/rahulunair/stable_diffusion_arc
Stable Difussion inference on Intel Arc dGPUs
intel intel-arc intel-gpu-max intelgpu ipex itex pytorch stable-diffusion stable-diffusion-api tensorflow
Last synced: 01 Aug 2024
https://github.com/gugarosa/learnergy
💡 Learnergy is a Python library for energy-based machine learning models.
dbn energy machine-learning python pytorch rbm
Last synced: 05 Aug 2024
https://github.com/gatheluck/FourierHeatmap
Unofficial pytorch implementation of Fourier Heat Map proposed in 'A Fourier Perspective on Model Robustness in Computer Vision' [Yin+, NeurIPS2019]
fourier-analysis heatmap pytorch robustness
Last synced: 03 Aug 2024