PyTorch | Ecosyste.ms: Awesome

https://github.com/outcastofmusic/quick-nlp

Pytorch NLP library based on FastAI

Last synced: 14 Nov 2024

https://github.com/kwea123/casmvsnet_pl

Cascade Cost Volume for High-Resolution Multi-View Stereo and Stereo Matching using pytorch-lightning

3d-reconstruction blendedmvs cascade-cost-volume depth-prediction dtu-dataset multi-view-stereo mvsnet pytorch pytorch-lightning tanks-and-temples

Last synced: 15 Jan 2025

https://github.com/loeweX/Greedy_InfoMax

Code for the paper: Putting An End to End-to-End: Gradient-Isolated Learning of Representations

deep-learning local-learning pytorch

Last synced: 12 Nov 2024

https://github.com/idea-research/humansd

[ICCV 2023] The official implementation of paper "HumanSD: A Native Skeleton-Guided Diffusion Model for Human Image Generation"

conditional-image-generation deep-learning iccv iccv2023 image-generation pytorch

Last synced: 15 Jan 2025

https://github.com/fregu856/3DOD_thesis

3D Object Detection for Autonomous Driving in PyTorch, trained on the KITTI dataset.

3d-deep-learning 3d-detection autonomous-driving computer-vision deep-learning machine-learning object-detection pytorch

Last synced: 27 Nov 2024

https://github.com/pmj110119/YOLOX_deepsort_tracker

using yolox+deepsort for object-tracking

deep-sort deepsort object-detection object-tracker pytorch tracker yolo yolo-x yolov3 yolov5 yolox

Last synced: 09 Nov 2024

https://github.com/aimagelab/show-control-and-tell

Show, Control and Tell: A Framework for Generating Controllable and Grounded Captions. CVPR 2019

caption-generation captioning-images cvpr2019 image-captioning pytorch visual-semantic

Last synced: 14 Jan 2025

https://github.com/rballester/tntorch

Tensor Network Learning with PyTorch

cp-decomposition data-science learning pytorch tensor-decomposition tensor-networks tensor-train tensors tucker-decomposition

Last synced: 15 Nov 2024

https://github.com/merantix-momentum/squirrel-core

A Python library that enables ML teams to share, load, and transform data in a collaborative, flexible, and efficient way :chestnut:

ai cloud-computing collaboration computer-vision cv data-ingestion data-mesh data-science dataops datasets deep-learning distributed internal machine-learning ml natural-language-processing nlp python pytorch tensorflow

Last synced: 02 Nov 2024

https://github.com/kennethnym/infinifi

infinifi plays gentle lofi music in the background indefinitely

artificial-intelligence audiocraft lo-fi lo-fi-radio machine-learning musicgen pytorch

Last synced: 13 Jan 2025

https://github.com/sunzeyeah/RLHF

Implementation of Chinese ChatGPT

chatgpt deep-learning deepspeed glm nlp pangu pytorch

Last synced: 31 Oct 2024

https://github.com/DeNA/HandyRL

HandyRL is a handy and simple framework based on Python and PyTorch for distributed reinforcement learning that is applicable to your own environments.

deep-learning distributed-training games machine-learning policy-gradient pytorch reinforcement-learning

Last synced: 04 Nov 2024

https://github.com/shoufachen/cyclemlp

[ICLR'22 Oral] Implementation of "CycleMLP: A MLP-like Architecture for Dense Prediction"

backbone-models mlp pytorch

Last synced: 17 Jan 2025

https://github.com/maum-ai/nuwave

NU-Wave: A Diffusion Probabilistic Model for Neural Audio Upsampling @ INTERSPEECH 2021

deep-generative-model deep-learning neural-audio-upsampling pytorch super-resolution upsampling

Last synced: 16 Jan 2025

https://github.com/maum-ai/nuwave2

NU-Wave 2: A General Neural Audio Upsampling Model for Various Sampling Rates @ INTERSPEECH 2022

deep-learning neural-audio-upsampling pytorch super-resolution upsampling

Last synced: 15 Jan 2025

https://github.com/mantasu/cs231n

Shortest solutions for CS231n 2021-2024

computer-vision convolutional-neural-networks cs231n deep-learning pytorch tensorflow visual-recognition

Last synced: 15 Jan 2025

https://github.com/pooya-mohammadi/yolov5-gradcam

Visualizing Yolov5's layers using GradCam

convolutional-neural-networks deep-learning deeputils gradcam pytorch visualizing yolov5

Last synced: 09 Nov 2024

https://github.com/extreme-bert/extreme-bert

ExtremeBERT is a toolkit that accelerates the pretraining of customized language models on customized datasets, described in the paper “ExtremeBERT: A Toolkit for Accelerating Pretraining of Customized BERT”.

bert deep-learning language-model language-models machine-learning natural-language-processing nlp python pytorch transformer

Last synced: 16 Nov 2024

https://github.com/DocF/multispectral-object-detection

Multispectral Object Detection with Yolov5 and Transformer

deep-learning object-detection pytorch transformer yolov5

Last synced: 09 Nov 2024

https://github.com/showlab/all-in-one

[CVPR2023] All in One: Exploring Unified Video-Language Pre-training

codebase pre-training pytorch video-language

Last synced: 17 Jan 2025

https://github.com/numenta/nupic.torch

Numenta Platform for Intelligent Computing PyTorch libraries

htm numenta-platform pytorch

Last synced: 13 Jan 2025

https://github.com/sgrvinod/a-pytorch-tutorial-to-transformers

Attention Is All You Need | a PyTorch Tutorial to Transformers

attention-is-all-you-need pytorch pytorch-tutorial transformer transformer-architecture transformer-tutorial

Last synced: 07 Jan 2025

https://github.com/jacobgil/pytorch-tensor-decompositions

PyTorch implementation of [1412.6553] and [1511.06530] tensor decomposition methods for convolutional layers.

deep-learning pytorch tensorly

Last synced: 15 Jan 2025

https://github.com/jingweiz/pytorch-dnc

Neural Turing Machine (NTM) & Differentiable Neural Computer (DNC) with pytorch & visdom

deep-learning dnc external-memory ntm pytorch visdom

Last synced: 21 Nov 2024

https://github.com/iArunava/ENet-Real-Time-Semantic-Segmentation

ENet - A Neural Net Architecture for real time Semantic Segmentation

artificial-intelligence deep-learning enet paper python3 pytorch

Last synced: 27 Nov 2024

https://github.com/tugstugi/pytorch-saltnet

Kaggle | 9th place single model solution for TGS Salt Identification Challenge

cnn convolutional-neural-networks deep-learning kaggle-competition pytorch segmentation seismic-imaging unet unet-image-segmentation unet-pytorch

Last synced: 16 Jan 2025

https://github.com/awslabs/renate

Library for automatic retraining and continual learning

aws continual-learning hyperparameter-tuning hyperparameters-optimization machine-learning machine-learning-algorithms neural neural-network pytorch pytorch-lightning sagemaker

Last synced: 11 Jan 2025

https://github.com/kkanshul/finegan

FineGAN: Unsupervised Hierarchical Disentanglement for Fine-grained Object Generation and Discovery

computer-vision deep-learning disentangled-representations fine-grained gans generative-adversarial-network image-generation image-manipulation pytorch

Last synced: 15 Nov 2024

https://github.com/Lin-Yijie/Graph-Matching-Networks

PyTorch implementation of Graph Matching Networks, e.g., Graph Matching with Bi-level Noisy Correspondence (COMMON, ICCV 2023), Graph Matching Networks for Learning the Similarity of Graph Structured Objects (GMN, ICML 2019).

deep-learning gnn graph-neural-networks pytorch

Last synced: 04 Nov 2024

https://github.com/millionintegrals/vel

Velocity in deep-learning research

convolutional-neural-networks deep-learning python pytorch reinforcement-learning

Last synced: 12 Jan 2025

https://github.com/delijati/pytorch-siamese

Siamese Network implementation using Pytorch

contrastive pytorch siamese

Last synced: 15 Jan 2025

https://github.com/krishnap25/mauve

Package to compute Mauve, a similarity score between neural text and human text. Install with `pip install mauve-text`.

deep-learning huggingface-transformers nlp pytorch text-generation

Last synced: 11 Jan 2025

https://github.com/cure-lab/pnpinversion

[ICLR2024] Official repo for paper "PnP Inversion: Boosting Diffusion-based Editing with 3 Lines of Code"

deep-learning diffusion-inversion diffusion-model diffusion-models iclr2024 image-editing image-generation paper-implementation paper-implementations pytorch

Last synced: 15 Jan 2025

https://github.com/MillionIntegrals/vel

Velocity in deep-learning research

convolutional-neural-networks deep-learning python pytorch reinforcement-learning

Last synced: 02 Nov 2024

https://github.com/szagoruyko/pyinn

CuPy fused PyTorch neural networks ops

cupy pytorch

Last synced: 17 Jan 2025

https://github.com/deephyper/deephyper

DeepHyper: Scalable Asynchronous Neural Architecture and Hyperparameter Search for Deep Neural Networks

automl deep-learning hpc hyperparameter-optimization keras machine-learning mpi multi-fidelity neural-architecture-search python pytorch raylib scalability tensorflow uncertainty-quantification

Last synced: 17 Jan 2025

https://github.com/jaxony/unet-pytorch

U-Net implementation for PyTorch based on https://arxiv.org/abs/1505.04597

deep-learning neural-network python pytorch semantic-segmentation

Last synced: 17 Jan 2025

https://github.com/prodramp/DeepWorks

A collection of Deep Learning projects and resources

artificial-intelligence deep-learning generative-models llm machine-learning pytorch tensorflow

Last synced: 29 Oct 2024

https://github.com/dyhan0920/PyramidNet-PyTorch

A PyTorch implementation for PyramidNets (Deep Pyramidal Residual Networks, https://arxiv.org/abs/1610.02915)

cifar-10 cifar-100 imagenet pyramidnet pytorch residual-networks resnet

Last synced: 31 Oct 2024

https://github.com/rentainhe/pytorch-distributed-training

Simple tutorials on Pytorch DDP training

apex cuda ddp-training deep-learning pytorch

Last synced: 15 Jan 2025

https://github.com/sdsubhajitdas/brain-tumor-segmentation

Brain Tumor Segmentation done using U-Net Architecture.

brain-tumor-segmentation deep-learning image-segmentation pytorch tumor-segmentation unet unet-image-segmentation unet-pytorch

Last synced: 15 Jan 2025

https://github.com/neulab/knn-transformers

PyTorch + HuggingFace code for RetoMaton: "Neuro-Symbolic Language Modeling with Automaton-augmented Retrieval" (ICML 2022), including an implementation of kNN-LM and kNN-MT

huggingface knn knn-lm knn-mt knn-transformers knnlm knnmt language language-models machine models nearest nearest-neighbor neighbor neuro-symbolic pytorch retomaton transformers translation

Last synced: 16 Jan 2025

https://github.com/benedekrozemberczki/sgcn

A PyTorch implementation of "Signed Graph Convolutional Network" (ICDM 2018).

deep-learning deepwalk gcn gpt2 gpt3 graph-attention graph-convolution graph-embedding graph-neural-networks graphsage machine-learning network-embedding neural-network node-classification node2vec pytorch pytorch-geometric sgcn signed-network transformer

Last synced: 15 Jan 2025

https://github.com/AnjieCheng/MnasNet-PyTorch

A PyTorch implementation of Mnasnet: MnasNet: Platform-Aware Neural Architecture Search for Mobile.

architecture-search mnasnet mobilenetv2 pytorch

Last synced: 31 Oct 2024

https://github.com/WenjieDu/SAITS

The official PyTorch implementation of the paper "SAITS: Self-Attention-based Imputation for Time Series". A fast and state-of-the-art (SOTA) deep-learning neural network model for efficient time-series imputation (impute multivariate incomplete time series containing NaN missing data/values with machine learning). https://arxiv.org/abs/2202.08516

attention attention-mechanism deep-learning imputation imputation-model impute incomplete-data incomplete-time-series interpolation irregular-sampling machine-learning missing-values partially-observed partially-observed-data partially-observed-time-series pytorch self-attention time-series time-series-imputation transformer

Last synced: 02 Nov 2024

https://github.com/junyaohu/common_metrics_on_video_quality

You can easily calculate FVD, PSNR, SSIM, LPIPS for evaluating the quality of generated or predicted videos.

fvd metrics psnr pytorch ssim video video-generation video-prediction

Last synced: 16 Jan 2025

https://github.com/benedekrozemberczki/gam

A PyTorch implementation of "Graph Classification Using Structural Attention" (KDD 2018).

attention deep-reinforcement-learning deepwalk gcn graph-attention graph-classification graph-convolution graph2vec graphsage machine-learning neural-networks node2vec pytorch reinforcement-agent reinforcement-learning sklearn structural-attention transfer-learning

Last synced: 15 Jan 2025

https://github.com/xpixelgroup/ranksrgan

ICCV 2019 (oral) RankSRGAN: Generative Adversarial Networks with Ranker for Image Super-Resolution. PyTorch implementation

gan generative-adversarial-network iccv2019 learning-to-rank low-level-vision pytorch super-resolution

Last synced: 16 Jan 2025

https://github.com/SamsungLabs/imvoxelnet

[WACV2022] ImVoxelNet: Image to Voxels Projection for Monocular and Multi-View General-Purpose 3D Object Detection

3d-object-detection imvoxelnet kitti mmdetection nuscenes object-detection pytorch scannet sun-rgbd

Last synced: 28 Oct 2024

https://github.com/benedekrozemberczki/SGCN

A PyTorch implementation of "Signed Graph Convolutional Network" (ICDM 2018).

deep-learning deepwalk gcn gpt2 gpt3 graph-attention graph-convolution graph-embedding graph-neural-networks graphsage machine-learning network-embedding neural-network node-classification node2vec pytorch pytorch-geometric sgcn signed-network transformer

Last synced: 02 Nov 2024

https://github.com/affjljoo3581/gpt2

PyTorch Implementation of OpenAI GPT-2

gpt2 language-model natural-language-generation natural-language-processing nlp pytorch transformer

Last synced: 09 Nov 2024

https://github.com/yatenglg/retinanet-pytorch

Retinanet目标检测算法(简单,明了,易用,全中文注释,单机多卡训练,视频检测)(based on pytorch,Simple, Clear, Mutil GPU)

object-detection pytorch retinanet

Last synced: 15 Jan 2025

https://github.com/qinbinli/moon

Model-Contrastive Federated Learning (CVPR 2021)

contrastive-learning cvpr2021 federated-learning pytorch

Last synced: 16 Jan 2025

https://github.com/imskr/Plant_Disease_Detection

Plant Disease Detector Web Application

cnn computervision deeplearning fastai machinelearning machinelearning-python plantclassification plantdisease plantdiseasedetection pytorch

Last synced: 15 Nov 2024

https://github.com/ducha-aiki/affnet

Code and weights for local feature affine shape estimation paper "Repeatability Is Not Enough: Learning Discriminative Affine Regions via Discriminability"

affine-shape-estimator computer-vision convolutional-networks convolutional-neural-networks deep-learning hessian image-matching image-retrieval local-features pytorch

Last synced: 15 Jan 2025

https://github.com/maum-ai/univnet

Unofficial PyTorch Implementation of UnivNet Vocoder (https://arxiv.org/abs/2106.07889)

deep-learning gan pytorch speech-synthesis text-to-speech tts vocoder

Last synced: 16 Jan 2025

https://github.com/benedekrozemberczki/GAM

A PyTorch implementation of "Graph Classification Using Structural Attention" (KDD 2018).

attention deep-reinforcement-learning deepwalk gcn graph-attention graph-classification graph-convolution graph2vec graphsage machine-learning neural-networks node2vec pytorch reinforcement-agent reinforcement-learning sklearn structural-attention transfer-learning

Last synced: 30 Oct 2024

https://github.com/imskr/plant_disease_detection

Plant Disease Detector Web Application

cnn computervision deeplearning fastai machinelearning machinelearning-python plantclassification plantdisease plantdiseasedetection pytorch

Last synced: 13 Jan 2025

https://github.com/JeongJiHeon/ScoreDiffusionModel

The Pytorch Tutorial of Score-based and Diffusion Model

diffusion-models pytorch score-based score-based-generative-models

Last synced: 17 Nov 2024

https://github.com/durandtibo/wildcat.pytorch

PyTorch implementation of "WILDCAT: Weakly Supervised Learning of Deep ConvNets for Image Classification, Pointwise Localization and Segmentation", CVPR 2017

convnet cvpr-2017 cvpr17 cvpr2017 deep-learning image-classification pytorch weakly-supervised-learning

Last synced: 15 Jan 2025

https://github.com/Randl/MobileNetV2-pytorch

Impementation of MobileNetV2 in pytorch

cnn image-classification mobilenet2 pytorch

Last synced: 31 Oct 2024

https://github.com/aipredict/ai-deployment

关注AI模型上线、模型部署

deploy keras lightgbm mxnet onnx pmml pyspark pytorch scikit-learn spark-ml tensorflow xgboost

Last synced: 30 Oct 2024

https://github.com/DerrickXuNu/v2x-vit

[ECCV2022] Official Implementation of paper "V2X-ViT: Vehicle-to-Everything Cooperative Perception with Vision Transformer"

3d-object-detection autonomous-driving collaborative-perception computer-vision deep-learning machine-learning multi-agent-system pytorch simulation v2x vehicle-to-everything vision-transformer

Last synced: 28 Oct 2024

https://github.com/salesforce/etsformer

PyTorch code for ETSformer: Exponential Smoothing Transformers for Time-series Forecasting

deep-learning exponential-smoothing forecasting interpretable-machine-learning pytorch time-series time-series-decomposition time-series-forecasting transformers

Last synced: 14 Jan 2025

https://github.com/TorchSpatiotemporal/tsl

tsl: a PyTorch library for processing spatiotemporal data.

deep-learning gnn graph-neural-networks pytorch spatio-temporal spatio-temporal-analysis spatio-temporal-data spatio-temporal-graph spatio-temporal-prediction spatiotemporal spatiotemporal-data spatiotemporal-data-analysis spatiotemporal-forecasting temporal-data temporal-graphs

Last synced: 12 Nov 2024

https://github.com/layumi/person-reid-3d

TNNLS'22 :statue_of_liberty: Parameter-Efficient Person Re-identification in the 3D Space :statue_of_liberty:

3d dukemtmc-reid graph-neural-networks market-1501 msmt17 ognet person-reid person-reidentification pytorch re-id re-identification reid

Last synced: 15 Jan 2025

https://github.com/francescosaveriozuppichini/a-journey-into-convolutional-neural-network-visualization-

A journey into Convolutional Neural Network visualization

computer-vision convolutional-neural-networks deep-learning deep-learning-tutorial pytorch

Last synced: 17 Jan 2025

https://github.com/pystiche/pystiche

Framework for Neural Style Transfer (NST) built upon PyTorch

framework neural-style-transfer python pytorch

Last synced: 17 Jan 2025

https://github.com/lartpang/machine-deep-learning

:wave: ML/DL学习笔记（基础+论文）

cs231n deeplearning deepthinking learning linux machine-learning notes python pytorch summary tensorflow thinking

Last synced: 16 Jan 2025

https://github.com/IDEA-Research/HumanSD

[ICCV 2023] The official implementation of paper "HumanSD: A Native Skeleton-Guided Diffusion Model for Human Image Generation"

conditional-image-generation deep-learning iccv iccv2023 image-generation pytorch

Last synced: 31 Oct 2024

https://github.com/Sleepychord/CogLTX

The source code of NeurIPS 2020 paper "CogLTX: Applying BERT to Long Texts"

bert pytorch

Last synced: 16 Nov 2024

https://idea-research.github.io/HumanSD/

[ICCV 2023] The official implementation of paper "HumanSD: A Native Skeleton-Guided Diffusion Model for Human Image Generation"

conditional-image-generation deep-learning iccv iccv2023 image-generation pytorch

Last synced: 31 Oct 2024

https://github.com/patrick-kidger/signatory

Differentiable computations of the signature and logsignature transforms, on both CPU and GPU. (ICLR 2021)

deep-learning deep-neural-networks logsignature logsignatures machine-learning pytorch rough-paths signature signatures

Last synced: 11 Jan 2025

https://github.com/ysymyth/3d-sdn

[NeurIPS 2018] 3D-Aware Scene Manipulation via Inverse Graphics

3d-sdn 3d-vision deep-learning disentangled-representations gans generative-adversarial-networks pytorch

Last synced: 17 Jan 2025

https://github.com/maum-ai/assem-vc

Official Code for Assem-VC @ICASSP2022

deep-learning pytorch speech-synthesis voice-conversion

Last synced: 16 Jan 2025

https://github.com/wutianyirosun/cgnet

CGNet: A Light-weight Context Guided Network for Semantic Segmentation [IEEE Transactions on Image Processing 2020]

camvid cityscapes pytorch semantic-segmentation

Last synced: 16 Jan 2025

https://github.com/vita-group/slak

[ICLR 2023] "More ConvNets in the 2020s: Scaling up Kernels Beyond 51x51 using Sparsity"; [ICML 2023] "Are Large Kernels Better Teachers than Transformers for ConvNets?"

51x51 convnet deep-learning dynamic-sparsity knowledge-distillation large-kernels object-detection pytorch segmentation

Last synced: 16 Jan 2025

https://github.com/ofnote/tsalib

Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)

deep-learning keras numpy pytorch tensor-shape tensorflow

Last synced: 14 Jan 2025

https://github.com/asappresearch/flambe

An ML framework to accelerate research and its path to production.

deep-learning distributed machine-learning ml python pytorch research

Last synced: 11 Jan 2025

https://github.com/compvis/metric-learning-divide-and-conquer

Source code for the paper "Divide and Conquer the Embedding Space for Metric Learning", CVPR 2019

few-shot-learning metric-learning pytorch

Last synced: 17 Jan 2025

https://github.com/cyoon1729/deep-Q-networks

Implementations of algorithms from the Q-learning family. Implementations inlcude: DQN, DDQN, Dueling DQN, PER+DQN, Noisy DQN, C51

deep-q-network pytorch q-learning reinforcement-learning

Last synced: 27 Nov 2024

https://github.com/huggingface/optimum-benchmark

🏋️ A unified multi-backend utility for benchmarking Transformers, Timm, PEFT, Diffusers and Sentence-Transformers with full support of Optimum's hardware optimizations & quantization schemes.

benchmark neural-compressor onnxruntime openvino pytorch tensorrt-llm text-generation-inference

Last synced: 04 Dec 2024

https://github.com/baaihealth/OpenComplex

Trainable PyTorch framework for developing protein, RNA and complex models.

alphafold2 protein-structure pytorch rna-structure

Last synced: 01 Dec 2024

https://github.com/khrylx/agentformer

[ICCV 2021] Official PyTorch Implementation of "AgentFormer: Agent-Aware Transformers for Socio-Temporal Multi-Agent Forecasting".

autonomous-driving deep-learning iccv2021 multi-agent-forecasting pytorch trajectory-forecasting trajectory-prediction transformer

Last synced: 15 Jan 2025

https://github.com/mertyg/vision-language-models-are-bows

Experiments and data for the paper "When and why vision-language models behave like bags-of-words, and what to do about it?" Oral @ ICLR 2023

blip clip compositionality multimodal pytorch vision-language

Last synced: 16 Jan 2025

https://github.com/carla-recourse/carla

CARLA: A Python Library to Benchmark Algorithmic Recourse and Counterfactual Explanation Algorithms

artificial-intelligence benchmark benchmarking counterfactual counterfactual-explanations counterfactuals explainability explainable-ai explainable-ml machine-learning python pytorch recourse tensorflow tensorflow2

Last synced: 16 Jan 2025

https://github.com/dalmia/siren

PyTorch implementation of Sinusodial Representation networks (SIREN)

activation-functions deep-learning machine-learning pytorch

Last synced: 13 Jan 2025

https://github.com/speedinghzl/pytorch-deeplab

DeepLab-ResNet rebuilt in Pytorch

deeplab-resnet pascal-voc pytorch semantic-segmentation

Last synced: 16 Jan 2025

https://github.com/lonepatient/nezha_chinese_pytorch

NEZHA: Neural Contextualized Representation for Chinese Language Understanding

chinese language-model nezha pytorch

Last synced: 16 Jan 2025

https://github.com/BY571/Soft-Actor-Critic-and-Extensions

PyTorch implementation of Soft-Actor-Critic and Prioritized Experience Replay (PER) + Emphasizing Recent Experience (ERE) + Munchausen RL + D2RL and parallel Environments.

actor-critic-algorithm continuous d2rl emphasizing-recent-experience multi-environment munchausen munchausen-reinforcement-learning parallel-computing prioritized-experience-replay pytorch reinforcement-learning reinforcement-learning-algorithms sac soft-actor-critic