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.

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-11-15 00:24:09 UTC
JSON Representation

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

https://github.com/amrzhd/mriskullstripping

Developing a UNet3D model for accurate MRI skull stripping using the Calgary Campinas 359 dataset, enhancing neuroimaging preprocessing workflows.

biomedical-image-processing image-segmentation monai mri mri-images nibabel python pytorch unet unet-image-segmentation

Last synced: 10 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/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: 11 Oct 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/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: 30 Oct 2024

https://github.com/ahmedfgad/torchga

Train PyTorch Models using the Genetic Algorithm with PyGAD

cnn deep-learning evolutionary-algorithms genetic-algorithm machine-learning neural-networks neuroscience pygad python pytorch torch

Last synced: 17 Nov 2024

https://github.com/graykode/modelsummary

All Model summary in PyTorch similar to `model.summary()` in Keras

keras python pytorch summary

Last synced: 01 Nov 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: 06 Nov 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/fcakyon/balanced-loss

Easy to use class balanced cross entropy and focal loss implementation for Pytorch

balanced-loss binary-crossentropy class-balanced-loss computer-vision cross-entropy cvpr deep-learning focal-loss image-classification loss-functions machine-learning pip pypi python pytorch

Last synced: 15 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/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/lorenmt/minimal-isaac-gym

A Minimal Example of Isaac Gym with DQN and PPO.

dqn isaac-gym ppo pytorch

Last synced: 03 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/cxhernandez/molencoder

Molecular AutoEncoder in PyTorch

autoencoder computational-chemistry deep-learning python pytorch

Last synced: 27 Oct 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: 08 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/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/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/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 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/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/comet-ml/issue-tracking

Questions, Help, and Issues for Comet ML

comet-ml keras machine-learning pytorch scikit-learn tenserflow

Last synced: 05 Nov 2024

https://github.com/qq456cvb/PRIN

Pointwise Rotation-Invariant Network (AAAI 2020)

pytorch rotation-invariant-features segmentation

Last synced: 28 Oct 2024

https://github.com/mberr/torch-ppr

(Personalized) Page-Rank computation using PyTorch

gpu pagerank personalized-pagerank pytorch

Last synced: 26 Oct 2024

https://github.com/abap34/ml-lecture

東京工業大学 traP Kaggle班「機械学習講習会」の資料

deeplearning education machine-learning pytorch textbook

Last synced: 05 Nov 2024

https://github.com/rucaibox/recbole-cdr

This is a library built upon RecBole for cross-domain recommendation algorithms

cross-domain pytorch recbole recommendation

Last synced: 11 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/bjarten/computer-vision-nd

Projects and exercises for the Udacity Computer Vision Nanodegree

char cnn computer cvnd detection face facial image-captioning keypoints learning lstm machine nanodegree pytorch rnn slam tutorial udacity vision yolo

Last synced: 29 Oct 2024

https://github.com/eric-canas/qrdet

Robust QR Detector based on YOLOv8

easy-to-use object-detection pip pytorch qr-detection qrcode yolov8

Last synced: 13 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/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: 22 Oct 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: 07 Oct 2024

https://github.com/dusty-nv/jetson-voice

ASR/NLP/TTS deep learning inference library for NVIDIA Jetson using PyTorch and TensorRT

deep-learning jetson jetson-nano nlp pytorch speech-recognition tensorrt text-to-speech

Last synced: 07 Nov 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: 14 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/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: 11 Nov 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/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/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/yuanzhoulvpi2017/quick_sentence_transformers

sentence-transformers to onnx 让sbert模型推理效率更快

bert pytorch sentence-transformers transformer transformers

Last synced: 08 Nov 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: 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: 10 Oct 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: 13 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/project-monai/monai-deploy-app-sdk

MONAI Deploy App SDK offers a framework and associated tools to design, develop and verify AI-driven applications in the healthcare imaging domain.

ai deep-learning deploy dicom healthcare image-processing machine-learning medical-imaging ml ml-infrastructure ml-platform mlops model-deployment model-serving monai pipeline python pytorch workflow

Last synced: 10 Oct 2024

https://github.com/rockerboo/lora-inspector

LoRA (Low-Rank Adaptation) inspector for Stable Diffusion

analysis lora low-rank-approximation machine-learning pytorch

Last synced: 26 Oct 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/writer/fitbert

Use BERT to Fill in the Blanks

bert fill-in-the-blank nlp python pytorch

Last synced: 29 Sep 2024

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/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/dengfenglai321/YOLOv4-Hat-detection

基于YOLOv4的安全帽佩戴检测

hat-detection pytorch yolov4

Last synced: 09 Nov 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: 27 Oct 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: 09 Nov 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/rose-stl-lab/torchts

Time series forecasting with PyTorch

deep-learning machine-learning pytorch pytorch-lightning time-series

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

https://github.com/erogol/fftnet

FFTNet vocoder implementation

deep-learning fftnet pytorch text2speech vocoder

Last synced: 22 Oct 2024

https://github.com/intel/torch-ccl

oneCCL Bindings for Pytorch*

oneccl pytorch

Last synced: 30 Oct 2024

https://github.com/d4l3k/torchdrive

torch models and primitives for birdseye view and voxel models for self driving tasks.

pytorch self-driving

Last synced: 14 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/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/yusugomori/deeplearning-keras-tf2-torch

詳解ディープラーニング第2版

deep-learning keras pytorch tensorflow2

Last synced: 07 Nov 2024

https://github.com/rose-stl-lab/dyffusion

[NeurIPS 2023] A Dynamics-informed Diffusion Model for Spatiotemporal Forecasting

deep-learning diffusion diffusion-models ensemble-forecasts machine-learning neurips neurips-2023 probabilistic-forecasting pytorch pytorch-lightning spatiotemporal-forecasting

Last synced: 09 Nov 2024

https://github.com/shchur/ifl-tpp

Implementation of "Intensity-Free Learning of Temporal Point Processes" (Spotlight @ ICLR 2020)

normalizing-flows pytorch temporal-point-processes

Last synced: 09 Nov 2024

https://github.com/kefirski/pytorch_Highway

Highway network implemented in pytorch

highway-network python pytorch

Last synced: 14 Nov 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: 13 Nov 2024

https://github.com/dsaurus/tensor4d

This is the official implementation of Tensor4D: Efficient Neural 4D Decomposition for High-fidelity Dynamic Reconstruction and Rendering.

3d-reconstruction 4d machine-learning nerf pytorch rendering

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

https://github.com/yaox12/byol-pytorch

PyTorch implementation of "Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning" with DDP and Apex AMP

byol byol-pytorch pretrained-model pytorch pytorch-implementation self-supervised-learning

Last synced: 28 Oct 2024

https://github.com/huawei-noah/multi_hyp_cc

[CVPR2020] A Multi-Hypothesis Approach to Color Constancy

color-constancy convolutional-neural-networks cvpr cvpr2020 pytorch whitebalance

Last synced: 13 Nov 2024

https://github.com/lromul/argus

Lightweight library for training neural networks in PyTorch

deep-learning machine-learning neural-network python pytorch

Last synced: 10 Oct 2024

https://github.com/lucasjinreal/ssds_pytorch

Multiple basenet MobileNet v1,v2, ResNet combined with SSD detection method and it's variants such as RFB, FSSD etc.

mobilenet mobilenetv2 pytorch ssd

Last synced: 06 Nov 2024

https://github.com/ndrplz/differentiable-renderer

Rastering algorithm to approximate the rendering of a 3D model silhouette in a fully differentiable way.

differentiable-rendering eccv-2018 python pytorch rendering rendering-engine tensorflow tensorflow-models

Last synced: 12 Nov 2024

https://github.com/vitae-transformer/vitae-transformer-scene-text-detection

A comprehensive list [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: 14 Nov 2024

https://github.com/subeeshvasu/2018_subeesh_epsr_eccvw

Project page of the paper 'Analyzing Perception-Distortion Tradeoff using Enhanced Perceptual Super-resolution Network' (ECCVW 2018)

deep-learning epsr pirm-sr pytorch super-resolution

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

https://github.com/adamdad/knowledgefactor

[ECCV2022] Factorizing Knowledge in Neural Networks

deep-learning distillation eccv2022 kd knowldge-distillation knowledge-transfer multitask-learning pytorch

Last synced: 16 Nov 2024

https://github.com/QY-H00/attention-interpolation-diffusion

[NeurIPS 2024] Official Implementation of Attention Interpolation of Text-to-Image Diffusion

deep-learning diffusion image-interpolation pytorch stable-diffusion text-to-image-generation

Last synced: 31 Oct 2024

https://github.com/wovchena/text-detection-fots.pytorch

FOTS text detection branch reimplementation, hmean: 83.3%

cvpr2018 fots pytorch text-detection

Last synced: 22 Oct 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/hellloxiaotian/acnet

Asymmetric CNN for image super-resolution (IEEE Transactions on Systmes, Man, and Cybernetics: Systems 2021)

blind-super-resolution cnn cnn-pytorch deep-learning deep-network deep-neural-networks image-processing low-level-vision python2 python27 pytorch pytorch-cnn super-resolution

Last synced: 12 Oct 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/imdeepmind/neuralpy

NeuralPy: A Keras like deep learning library works on top of PyTorch

data-science deep-learning keras library machine-learning neural-network neuralpy neuralpy-torch python pytorch

Last synced: 13 Nov 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: 15 Nov 2024

https://github.com/mukosame/aoda

Official implementation of "Adversarial Open Domain Adaptation for Sketch-to-Photo Synthesis"(WACV 2022/CVPRW 2021)

computer-vision cvpr cvpr2021 deep-learning gan gans image-generation image-manipulation image2image open-domain pytorch sketch wacv wacv2022

Last synced: 07 Nov 2024

https://github.com/tommyip/mamba2-minimal

Minimal Mamba-2 implementation in PyTorch

llm machine-learning mamba pytorch ssm

Last synced: 10 Nov 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