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/rf5/transfusion-asr

Transcribing Speech with Multinomial Diffusion, training code and models.

asr binomial-distribution diffusion discrete-diffusion pytorch speech-recognition

Last synced: 26 Sep 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/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/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/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/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/yang-0201/YOLOv6_pro

Make it easier for yolov6 to change the network structure

detection pytorch yolo yolov5 yolov6

Last synced: 02 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/moskomule/pytorch.rl.learning

for learning reinforcement learning using PyTorch.

pytorch reinforcement-learning

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/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/Fangyh09/Image2LMDB

Convert image folder to lmdb, adapted from Efficient-PyTorch

caffe caffe2 lmdb pytorch

Last synced: 03 Aug 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/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/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/lumo-tech/lumo

More light-weight pytorch experiment management library!

deep-learning experiment-management experiment-tracking pytorch version-control

Last synced: 27 Sep 2024

https://github.com/gngdb/pytorch-minimize

Use scipy.optimize.minimize as a PyTorch Optimizer.

python pytorch scipy

Last synced: 24 Sep 2024

https://github.com/hunto/image_classification_sota

Training ImageNet / CIFAR models with sota strategies and fancy techniques such as ViT, KD, Rep, etc.

cifar image-classification imagenet kd nas pruning pytorch rep transformer vit

Last synced: 03 Aug 2024

https://github.com/grimme-lab/dxtb

Efficient And Fully Differentiable Extended Tight-Binding

automatic-differentiation computational-chemistry pytorch quantum-chemistry tight-binding

Last synced: 27 Sep 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

https://github.com/EMUNES/Auto-Subtitle-File-Generation

Generate subtitle files with timelines in an automatic way.

automatic deep-learning generator pytorch subtitles

Last synced: 04 Aug 2024

https://github.com/kacky24/stylenet

A pytorch implemention of "StyleNet: Generating Attractive Visual Captions with Styles"

caption cnn dlhacks image-captioning lstm microsoft pytorch pytorch-implemention stylenet

Last synced: 01 Aug 2024

https://github.com/granularai/fabric

Urban change model designed to identify changes across 2 timestamps

change-detection cnn deep-learning pytorch satellite-imagery sentinel-2 unet unet-3d

Last synced: 01 Aug 2024

https://github.com/tonyduan/diffusion

From-scratch diffusion model implemented in PyTorch.

diffusion generative-models pytorch

Last synced: 31 Jul 2024

https://github.com/ducha-aiki/google-retrieval-challenge-2019-fastai-starter

fast.ai starter kit for Google Landmark Retrieval 2019 challenge

cnn fastai kaggle pytorch retrieval

Last synced: 02 Aug 2024

https://github.com/cthoyt/class-resolver

🔍 Lookup classes and instantiate them with style

optuna python pytorch utility

Last synced: 03 Aug 2024

https://github.com/kaituoxu/X-Punctuator

A PyTorch implementation of a punctuation prediction system using (B)LSTM, which automatically adds suitable punctuation into text without punctuation.

lstm punctuation pytorch

Last synced: 03 Aug 2024

https://github.com/jinglescode/python-signal-processing

splearn: package for signal processing and machine learning with Python. Contains tutorials on understanding and applying signal processing.

machine-learning numpy python python-tutorial pytorch pytorch-tutorial scipy sequence signal-processing

Last synced: 24 Sep 2024

https://github.com/ap229997/Conditional-Batch-Norm

Pytorch implementation of NIPS 2017 paper "Modulating early visual processing by language"

cbn modulated-resnet pytorch vqa

Last synced: 03 Aug 2024

https://github.com/lukasliebel/dfc2020_baseline

Simple Baseline for the IEEE GRSS Data Fusion Contest 2020

deep-learning earth-observation pytorch remote-sensing

Last synced: 07 Aug 2024

https://github.com/lwch/gotorch

golang libtorch binding

deep-learning go gpu libtorch mlp pytorch

Last synced: 27 Sep 2024

https://github.com/sjmikler/pytorch-symbolic

Provides symbolic API for model creation in PyTorch.

deep-learning neural-networks pytorch

Last synced: 26 Sep 2024

https://github.com/moskomule/anatome

Ἀνατομή is a PyTorch library to analyze representation of neural networks

neural-networks pytorch pytorch-library representation-learning

Last synced: 03 Aug 2024

https://github.com/Confusezius/ICCV2019_MIC

(ICCV 2019) This repo contains code for "MIC: Mining Interclass Characteristics for Improved Metric Learning", which proposes an auxiliary training task to explain away intra-class variations.

cars196 computer-vi cub200 deep-lear deep-metric-learning metric- pku-vehicle pytorch shop-clothes-retrieval

Last synced: 03 Aug 2024

https://github.com/LiyuanLucasLiu/Torch-Scope

A Toolkit for Training, Tracking, Saving Models and Syncing Results

gpu-memory logger pytorch tensorboard toolkit

Last synced: 03 Aug 2024

https://github.com/MrtnMndt/OpenVAE_ContinualLearning

Open-source code for our paper: Unified Probabilistic Deep Continual Learning through Generative Replay and Open Set Recognition

continual-learning deep-learning open-set-recognition pytorch variational-autoencoder

Last synced: 03 Aug 2024

https://github.com/fregu856/ebms_3dod

Official implementation of "Accurate 3D Object Detection using Energy-Based Models", CVPR Workshops 2021.

3d-object-detection computer-vision deep-learning energy-based-model machine-learning object-detection pytorch

Last synced: 31 Jul 2024

https://github.com/leftthomas/DGCNN

A PyTorch implementation of DGCNN based on AAAI 2018 paper "An End-to-End Deep Learning Architecture for Graph Classification"

gcn gnn graph graph-classification graph-convolutional-networks pytorch

Last synced: 31 Jul 2024

https://github.com/MzeroMiko/mamba-mini

An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivation. It is probably the code which is the most close to selective_scan_cuda in mamba.

efficient mamba pytorch selective-scan

Last synced: 31 Jul 2024

https://github.com/swyoon/pytorch-minimal-gaussian-process

A minimal implementation of Gaussian process regression in PyTorch

gaussian-process gaussian-process-regression gaussian-processes pytorch

Last synced: 02 Aug 2024

https://github.com/abdoelsayed2016/TNCR_Dataset

Deep learning, Convolutional neural networks, Image processing, Document processing, Table detection, Page object detection, Table classification. https://www.sciencedirect.com/science/article/pii/S0925231221018142

classification cnn convolutional-neural-networks deep-learning image-classification mmdetection object-detection object-recognition pytorch table table-detection table-recognition

Last synced: 03 Aug 2024

https://github.com/jw9730/hot

[NeurIPS'21] Higher-order Transformers for sets, graphs, and hypergraphs, in PyTorch

equivariance gnn graph hypergraph pytorch self-attention transformer

Last synced: 31 Jul 2024

https://github.com/Lightning-Universe/lightning-ColossalAI

Large Scale Distributed Model Training strategy with Colossal AI and Lightning AI

lightning python pytorch

Last synced: 02 Aug 2024

https://sirui-xu.github.io/STARS/

[ECCV 2022 oral] Official PyTorch implementation of the paper "Diverse Human Motion Prediction Guided by Multi-Level Spatial-Temporal Anchors"

deep-learning disentangled-representations eccv2022 generative-model graph-neural-networks human-motion-prediction motion-forecasting motion-prediction pytorch

Last synced: 02 Aug 2024

https://github.com/Sirui-Xu/STARS

[ECCV 2022 oral] Official PyTorch implementation of the paper "Diverse Human Motion Prediction Guided by Multi-Level Spatial-Temporal Anchors"

deep-learning disentangled-representations eccv2022 generative-model graph-neural-networks human-motion-prediction motion-forecasting motion-prediction pytorch

Last synced: 02 Aug 2024

https://github.com/baiksung/MeTAL

Official PyTorch implementation of "Meta-Learning with Task-Adaptive Loss Function for Few-Shot Learning" (ICCV2021 Oral)

computer-vision deep-learning few-shot-learning iccv machine-learning maml meta-learning pytorch

Last synced: 03 Aug 2024

https://github.com/speediedan/finetuning-scheduler

A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.

artificial-intelligence fine-tuning finetuning machine-learning neural-networks pytorch pytorch-lightning superglue transfer-learning

Last synced: 01 Aug 2024

https://github.com/KushajveerSingh/resize_network_cv

PyTorch implementation of the paper "Learning to Resize Images for Computer Vision Tasks" on Imagenette and Imagewoof datasets

computer-vision pytorch pytorch-implementation pytorch-lightning

Last synced: 03 Aug 2024

https://github.com/youngfly11/LCMCG-PyTorch

AAAI2020-The official implementation of "Learning Cross-modal Context Graph for Visual Grounding"

aaai2020 pytorch

Last synced: 04 Aug 2024

https://github.com/val-iisc/SDAT

[ICML 2022]Source code for "A Closer Look at Smoothness in Domain Adversarial Training",

adversarial-training dann domain-adaptation icml-2022 pytorch sharpness-aware-minimization

Last synced: 03 Aug 2024

https://github.com/searchableai/kitanaqa

KitanaQA: Adversarial training and data augmentation for neural question-answering models

adversarial-attacks adversarial-training bert data-augmentation ml-automation natural-language-processing nlp pytorch question-answering transformer

Last synced: 25 Sep 2024

https://github.com/lab-cosmo/sphericart

Multi-language library for the calculation of spherical harmonics in Cartesian coordinates

geometric-deep-learning python pytorch spherical-harmonics

Last synced: 25 Sep 2024

https://github.com/mboudiaf/pytorch-meta-dataset

A non-official 100% PyTorch implementation of META-DATASET benchmark for few-shot classification

few-shot-classifcation few-shot-learning meta-dataset pytorch tfrecorddataset

Last synced: 03 Aug 2024

https://github.com/unixwzrd/oobabooga-macOS

Optimizing performance, building and installing packages required for oobabooga, AI and Data Science on Apple Silicon GPU. The goal is to optimize wherever possible, from the ground up.

blas journal llama-cpp-python macos numpy oobabooga pytorch

Last synced: 09 Aug 2024

https://github.com/Lee-Gihun/MixCo-Mixup-Contrast

(NeurIPS 2020 Workshop on SSL) Official Implementation of "MixCo: Mix-up Contrastive Learning for Visual Representation"

deep-learning mix-up pytorch self-supervised-learning

Last synced: 02 Aug 2024

https://github.com/saturncloud/dask-pytorch-ddp

dask-pytorch-ddp is a Python package that makes it easy to train PyTorch models on dask clusters using distributed data parallel.

computer-vision dask deep-learning distributed-computing machine-learning nlp pytorch

Last synced: 03 Aug 2024

https://github.com/k-h-ismail/dilated-convolution-with-learnable-spacings-pytorch

[ICLR 2023] "Dilated convolution with learnable spacings" Ismail Khalfaoui Hassani, Thomas Pellegrini and Timothée Masquelier

pytorch

Last synced: 27 Sep 2024

https://github.com/aangelopoulos/conformal-risk

Conformal prediction for controlling monotonic risk functions. Simple accompanying PyTorch code for conformal risk control in computer vision and natural language processing.

computer-vision conformal conformal-prediction natural-language-processing python pytorch pytorch-implementation uncertainty-estimation uncertainty-quantification

Last synced: 01 Aug 2024

https://github.com/ztxtech/Time-Evidence-Fusion-Network

Official implementation of "Time Evidence Fusion Network: Multi-source View in Long-Term Time Series Forecasting" (https://arxiv.org/abs/2405.06419)

cuda deep-learning machine-learning macos neural-network neural-networks pytorch time-series time-series-analysis time-series-forecasting time-series-prediction uestc

Last synced: 01 Aug 2024

https://github.com/vexilligera/strokenet

a neural network that draws digits using strokes

deep-learning nodejs pytorch reinforcement-learning

Last synced: 02 Aug 2024

https://github.com/customdiffusion360/custom-diffusion360

CustomDiffusion360: Customizing Text-to-Image Diffusion with Camera Viewpoint Control

camera-pose-control computer-vision customization generative-model pytorch stable-diffusion text-to-image

Last synced: 31 Jul 2024

https://github.com/Toytiny/RaFlow

Self-Supervised Scene Flow Estimation with 4-D Automotive Radar (RA-L & IROS'22)

automotive-radar autonomous-driving deep-learning point-cloud pytorch scene-flow-estimation self-supervised

Last synced: 31 Jul 2024

https://github.com/byminji/SLTtrack

Official Implementation of Towards Sequence-Level Training for Visual Tracking (ECCV 2022)

eccv2022 pytorch reinforcement-learning single-object-tracking sot tracking visual-tracking vot

Last synced: 01 Aug 2024

https://github.com/xierc/Semi_Human_Pose

[ICCV-2021] An Empirical Study of the Collapsing Problem in Semi-Supervised 2D Human Pose Estimation

consistency-regularization human-pose-estimation pytorch semi-supervised-learning

Last synced: 08 Aug 2024

https://github.com/Novartis/torchsurv

Deep survival analysis made easy

deep-learning pytorch survival-analysis

Last synced: 26 Sep 2024

https://github.com/OIdiotLin/torchtracer

A python package for visualization and storage management in a pytorch AI task.

lab machine-learning pytorch tools

Last synced: 03 Aug 2024

https://github.com/dsgiitr/adversarial_lab

Web-based Tool for visualisation and generation of adversarial examples by attacking ImageNet Models like VGG, AlexNet, ResNet etc.

adversarial-attacks computer-vision flask html-css-javascript imagenet machine-learning python pytorch visualization

Last synced: 02 Aug 2024

https://github.com/angeluriot/Generative_adversarial_network

A deep learning model that can create high quality images by training on a dataset.

deep-learning gan generative-ai pytorch

Last synced: 01 Aug 2024

https://github.com/csiro-robotics/TCE

This repository contains the code implementation used in the paper Temporally Coherent Embeddings for Self-Supervised Video Representation Learning (TCE).

action-recognition computer-vision contrastive-learning contrastive-loss deep-learning embeddings hmdb51 kinetics-datasets metric-learning pytorch representation-learning self-supervised-learning tsne-visualisations ucf-101

Last synced: 02 Aug 2024

https://github.com/francois-rozet/torchist

NumPy-style histograms in PyTorch

histogram numpy pytorch

Last synced: 03 Aug 2024

https://github.com/Harry24k/CW-pytorch

A pytorch implementation of "Towards Evaluating the Robustness of Neural Networks"

adversarial-attacks deep-learning pytorch

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 openstl 43 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-max-mojo-mlir 182 awesome-ai-tools 500
PyTorch Categories
Blogs 294 Pytorch & related libraries 228 Pytorch & related libraries｜Pytorch & 相关库 218 Research 197 Tutorials 192 文本数据和NLP 182 Uncategorized 161 Frameworks 142 数据可视化 130 机器学习框架 123 1.) Neural Architecture Search 114 图像数据与CV 106 Blogposts :sparkle: 103 Learning 102 模型的可解释性 100 Interactive Learning & Tools 96 超参数优化和AutoML 92 Tutorials, books, & examples 83 Tutorials & books & examples｜教程 & 书籍 & 示例 81 工作流程和实验跟踪 80