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: 2025-02-16 00:24:26 UTC
JSON Representation

https://github.com/yuanze-lin/revive

[NeurIPS 2022] Official Code for REVIVE: Regional Visual Representation Matters in Knowledge-Based Visual Question Answering

computer-vision deep-learning gpt-3 knowledge-based multimodal-deep-learning neurips2022 ok-vqa pytorch question-answering vision-and-languge vqa

Last synced: 13 Feb 2025

https://github.com/google-deepmind/dm_nevis

NEVIS'22: Benchmarking the next generation of never-ending learners

benchmark continual-learning efficient-learning jax pytorch

Last synced: 24 Jan 2025

https://github.com/anibali/margipose

3D monocular human pose estimation

deep-learning pose-estimation pytorch

Last synced: 12 Feb 2025

https://github.com/RBirkeland/MVCNN-PyTorch

3d deep-learning neural-network pytorch resnet

Last synced: 28 Oct 2024

https://github.com/TalSchuster/CrossLingualContextualEmb

Cross-Lingual Alignment of Contextual Word Embeddings

allennlp bert contextual-embeddings crosslingual elmo nlp pytorch wordembeddings zeroshot-learning

Last synced: 17 Nov 2024

https://github.com/g-research/fasttrackml

Experiment tracking server focused on speed and scalability

ai apache-spark data-science data-visualization experiment-tracking machine-learning metadata metadata-tracking metrics ml mlflow mlflow-tracking-server mlops pytorch tensorboard tensorflow visualization

Last synced: 17 Feb 2025

https://github.com/talschuster/crosslingualcontextualemb

Cross-Lingual Alignment of Contextual Word Embeddings

allennlp bert contextual-embeddings crosslingual elmo nlp pytorch wordembeddings zeroshot-learning

Last synced: 08 Nov 2024

https://github.com/monk1337/various-attention-mechanisms

This repository contain various types of attention mechanism like Bahdanau , Soft attention , Additive Attention , Hierarchical Attention etc in Pytorch, Tensorflow, Keras

attention attention-lstm attention-mechanism attention-mechanisms attention-model attention-network bahdanau-attention hierarchical-attention keras luong-attention multi-head-attention pytorch scaled-dot-product-attention self-attention sentence-attention

Last synced: 24 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: 17 Feb 2025

https://github.com/xettrisomeman/machine-leaarning-engineer-roadmap

A simple roadmap to become a machine learning engineer.

deep-learning machine-learning numpy pytorch roadmap

Last synced: 31 Dec 2024

https://github.com/Evgeneus/Graph-Domain-Adaptaion

PyTorch code for the paper "Curriculum Graph Co-Teaching for Multi-target Domain Adaptation" (CVPR2021)

cdan computer-vision coteaching curriculum-learning domain-adaptation domainnet graph-convolutional-networks graph-domain-adaptation graph-neural-networks mtdap office pacs pytorch

Last synced: 15 Nov 2024

https://github.com/denisyarats/exorl

ExORL: Exploratory Data for Offline Reinforcement Learning

control datasets deep-learning exporation model-free mujoco off-policy offline-rl python pytorch reinforcement-learning unsupevised

Last synced: 17 Nov 2024

https://github.com/G-Research/fasttrackml

Experiment tracking server focused on speed and scalability

Last synced: 02 Dec 2024

https://github.com/outerport/paramit

Parameterize Python scripts/notebooks all from the command line and run on cloud GPUs

ai aiops config experiment-tracking machine-learning metadata metadata-extraction metadata-tracking ml mlops python pytorch

Last synced: 13 Feb 2025

https://github.com/mzaradzki/factorization-machine-for-prediction

Factorization Machine for regression and classification

keras machine-learning python pytorch theano

Last synced: 01 Dec 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: 14 Feb 2025

https://github.com/JinheonBaek/GMT

Official Code Repository for the paper "Accurate Learning of Graph Representations with Graph Multiset Pooling" (ICLR 2021)

graph graph-classification graph-multiset-pooling graph-multiset-transformer graph-neural-networks graph-pooling graph-reconstruction graph-representation graph-representation-learning graph-transformer pytorch

Last synced: 28 Nov 2024

https://github.com/omegaml/omegaml

MLOps simplified. One platform, all the functionality you need. Swiss made

artificial-intelligence celery data-science deploy docker-image jupyter-notebook machine-learning mlops productized pytorch scikit-learn tensorflow

Last synced: 12 Feb 2025

https://github.com/clearhanhui/learnlibtorch

LibTorch 中文教程。

libtorch pytorch pytorch-cpp toturial

Last synced: 13 Feb 2025

https://github.com/slydg/gru_to_predict_timeseries

使用pytorch搭建的循环神经网络在股票数据时间序列上的应用

gru python3 pytorch sklearn

Last synced: 12 Nov 2024

https://github.com/openvinotoolkit/openvino_contrib

Repository for OpenVINO's extra modules

arm inference-engine java nvidia-gpu openvino pytorch

Last synced: 27 Oct 2024

https://github.com/takuseno/minerva

An out-of-the-box GUI tool for offline deep reinforcement learning

deep-learning deep-reinforcement-learning offline-rl pytorch

Last synced: 20 Dec 2024

https://github.com/tri-ml/dgp

ML Dataset Governance Policy for Autonomous Vehicle Datasets

autonomous-driving data-management deep-learning machine-learning pytorch

Last synced: 16 Feb 2025

https://github.com/haofanwang/accurate-head-pose

Pytorch code for Hybrid Coarse-fine Classification for Head Pose Estimation

head-pose-estimation headpose pytorch

Last synced: 07 Nov 2024

https://github.com/shyamsn97/hyper-nn

Easy Hypernetworks in Pytorch and Jax

flax hypernetworks jax machine-learning neural-networks pytorch

Last synced: 19 Dec 2024

https://github.com/tjiiv-cprg/monorun

[CVPR'21] MonoRUn: Monocular 3D Object Detection by Reconstruction and Uncertainty Propagation

3d-object-detection cvpr detection monocular pytorch reconstruction uncertainty

Last synced: 09 Nov 2024

https://github.com/harry24k/pgd-pytorch

A pytorch implementation of "Towards Deep Learning Models Resistant to Adversarial Attacks"

adversarial-attacks deep-learning pytorch

Last synced: 06 Nov 2024

https://github.com/chenyuntc/scene-baseline

PyTorch baseline for AI challenger Scene classification

image-classification pytorch

Last synced: 09 Nov 2024

https://github.com/chail/gan-ensembling

Invert and perturb GAN images for test-time ensembling

generative-adversarial-network image-classfication image-synthesis pytorch

Last synced: 15 Nov 2024

https://github.com/notai-tech/fastdeploy

Deploy DL/ ML inference pipelines with minimal extra code.

deep-learning docker falcon gevent gunicorn http-server inference-server model-deployment model-serving python pytorch serving streaming-audio tensorflow-serving tf-serving torchserve triton triton-inference-server triton-server websocket

Last synced: 12 Feb 2025

https://github.com/gengyanlei/segmentation_pytorch

semantic segmentation pytorch 语义分割

deeplab-v3-plus hed hf-fcn pspnet pytorch unet

Last synced: 29 Oct 2024

https://github.com/yinboc/trans-inr

Transformers as Meta-Learners for Implicit Neural Representations, in ECCV 2022

implicit-neural-representation machine-learning meta-learning pytorch transformer

Last synced: 08 Nov 2024

https://github.com/toshas/torch-discounted-cumsum

Fast Discounted Cumulative Sums in PyTorch

discounted-cumulative-sum pytorch reinforce reinforcement-learning rl

Last synced: 25 Jan 2025

https://github.com/zeta36/muzero

A simple implementation of MuZero algorithm for connect4 game

deepmind jupyter-notebook muzero python pytorch

Last synced: 02 Nov 2024

https://github.com/xuyxu/soft-decision-tree

PyTorch Implementation of "Distilling a Neural Network Into a Soft Decision Tree." Nicholas Frosst, Geoffrey Hinton., 2017.

classification-trees decision-tree deep-learning pytorch tree

Last synced: 08 Nov 2024

https://github.com/Torsion-Audio/nn-inference-template

Neural network inference template for real-time cricital audio environments - presented at ADC23

audio audio-plugin deep-learning inference inference-engine juce libtorch machine-learning neural-network onnx onnxruntime pytorch tensorflow tensorflow-lite

Last synced: 14 Nov 2024

https://github.com/openpifpaf/openpifpafwebdemo

Web browser based demo of OpenPifPaf.

ml-web onnx onnx-torch pose-estimation pytorch typescript

Last synced: 03 Jan 2025

https://github.com/myscience/x-lstm

Pytorch implementation of the xLSTM model by Beck et al. (2024)

lightning llm lstm pytorch pytorch-implementation xlstm

Last synced: 10 Dec 2024

https://github.com/doocs/deep-learning

🙃 深度学习实践与知识总结

ai deep-learning deep-neural-networks doocs pytorch tensorflow

Last synced: 30 Oct 2024

https://github.com/wchliao/multi-task-image-classification

Multi-task learning for image classification implemented in PyTorch.

cifar-100 cifar10 deep-learning machine-learning multi-task-learning omniglot pytorch

Last synced: 15 Nov 2024

https://github.com/ajhamdi/MVTN

pytorch implementation of the ICCV'21 paper "MVTN: Multi-View Transformation Network for 3D Shape Recognition"

3d 3d-models classification deep-learning iccv2021 point-cloud pytorch

Last synced: 28 Oct 2024

https://github.com/kunhe/FastAP-metric-learning

Code for CVPR 2019 paper "Deep Metric Learning to Rank"

cvpr deep-metric-learning learning-to-rank matconvnet metric-learning pytorch

Last synced: 15 Nov 2024

https://github.com/glefundes/mobile-face-gaze

Lightweight gaze estimation with PyTorch.

deep-learning gaze gaze-estimation pytorch

Last synced: 13 Feb 2025

https://github.com/eveningglow/multitask-CycleGAN

Pytorch implementation of multitask CycleGAN with auxiliary classification loss

computer-vision cyclegan deep-learning gan generative-adversarial-network image-translation pytorch

Last synced: 15 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: 29 Jan 2025

https://github.com/lyeoni/pytorch-mnist-vae

autoencoder deep-learning generative-model mnist pytorch vae

Last synced: 06 Nov 2024

https://github.com/ice-tong/pytorch-captcha

使用pytorch搭建cnn识别验证码

captcha-breaking pytorch pytorch-cnn pytorch-tutorial

Last synced: 30 Oct 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: 01 Nov 2024

https://github.com/1ytic/pytorch-edit-distance

Levenshtein edit-distance on PyTorch and CUDA

asr cuda edit-distance levenshtein nlp pytorch

Last synced: 17 Feb 2025

https://github.com/aws-samples/aws-lambda-docker-serverless-inference

Serve scikit-learn, XGBoost, TensorFlow, and PyTorch models with AWS Lambda container images support.

aws aws-lambda deep-java-library djl docker huggingface huggingface-transformers inference java machine-learning pytorch pytorch-models sagemaker scikit-learn serverless serverless-application-model tensor-flow tensorflow tensorflow-java xgboost

Last synced: 27 Oct 2024

https://github.com/eric-ai-lab/PEViT

Official implementation of AAAI 2023 paper "Parameter-efficient Model Adaptation for Vision Transformers"

fine-tuning image-classification parameter-efficient-tuning pytorch vision-transformer

Last synced: 05 Nov 2024

https://github.com/dagshub/client

DagsHub client libraries

ai data data-science data-streaming dvc hacktoberfest hacktoberfest2023 keras machine-learning machinelearning mlops python pytorch tensorflow

Last synced: 16 Feb 2025

https://github.com/mratsim/Arch-Data-Science

Archlinux PKGBUILDs for Data Science, Machine Learning, Deep Learning, NLP and Computer Vision

archlinux cuda cudnn data-science deep-learning lightgbm machine-learning mkl mxnet natural-language-processing natural-language-understanding nervana opencv package pandas pytorch scikit-learn spacy tensorflow xgboost

Last synced: 27 Nov 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: 14 Feb 2025

https://github.com/jkoutsikakis/pytorch-wrapper

Provides a systematic and extensible way to build, train, evaluate, and tune deep learning models using PyTorch.

data-science deep-learning machine-learning neural-network python pytorch pytorch-wrapper tensor

Last synced: 27 Nov 2024

https://github.com/xunge/pytorch_lmdb_imagenet

Use lmdb to speed up imagenet dataset

imagenet lmdb pytorch

Last synced: 15 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: 14 Feb 2025

https://github.com/orobix/visual-feature-attribution-using-wasserstein-gans-pytorch

Implementation of Visual Feature Attribution using Wasserstein GANs (VAGANs, https://arxiv.org/abs/1711.08998) in PyTorch

anomaly-detection cnn gan python pytorch vgan

Last synced: 11 Nov 2024

https://github.com/mratsim/arch-data-science

Archlinux PKGBUILDs for Data Science, Machine Learning, Deep Learning, NLP and Computer Vision

Last synced: 09 Nov 2024

https://github.com/nitin-rathi/hybrid-snn-conversion

Training spiking networks with hybrid ann-snn conversion and spike-based backpropagation

ann-snn-conversion backpropagation-algorithm deep-learning pytorch snn spiking-neural-networks

Last synced: 12 Nov 2024

https://github.com/lromul/ball-action-spotting

SoccerNet@CVPR | 1st place solution for Ball Action Spotting Challenge 2023

action-recognition action-spotting deep-learning football pytorch soccer soccernet video video-processing

Last synced: 01 Nov 2024

https://github.com/polyaxon/mloperator

Machine learning operator & controller for Kubernetes

dask deep-learning k8s keras kubernetes kubernetes-operator machine-learning mlops mpi mxnet notebook pytorch scikit-learn spark tensorboard tensorflow xgboost

Last synced: 17 Feb 2025

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 Feb 2025

https://github.com/masadcv/fastgeodis

Fast Implementation of Generalised Geodesic Distance Transform for CPU (OpenMP) and GPU (CUDA)

deep-learning gpu pytorch

Last synced: 19 Dec 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: 14 Feb 2025

https://github.com/roman-vygon/triplet_loss_kws

Learning Efficient Representations for Keyword Spotting with Triplet Loss

deep-learning keyword-spotting pytorch speech-recognition

Last synced: 03 Nov 2024

https://github.com/LBM-EPFL/PeSTo

Geometric deep learning method to predict protein binding interfaces from a protein structure.

deep-learning geometric-deep-learning protein-binding-site protein-protein-interface protein-structure pytorch

Last synced: 12 Nov 2024

https://github.com/senya-ashukha/quantile-regression-dqn-pytorch

A short and easy implementation of Quantile Regression DQN | Distributional Reinforcement Learning

pytorch qr-dqn quantile-regression-dqn reinforcement-learning

Last synced: 19 Nov 2024

https://github.com/hpcaitech/CachedEmbedding

A memory efficient DLRM training solution using ColossalAI

colossal-ai deep-learning dlrm embeddings nlp pytorch recommandation-system

Last synced: 07 Nov 2024

https://github.com/nishnik/capsnet-pytorch

My attempt at implementing CapsNet from the paper Dynamic Routing Between Capsules

convolutional-neural-networks deep-learning pytorch

Last synced: 18 Nov 2024

https://github.com/gtbluesky/onnxruntime_flutter

A flutter plugin for OnnxRuntime provides an easy, flexible, and fast Dart API to integrate Onnx models in flutter apps across mobile and desktop platforms.

dart ffi flutter onnx onnxruntime pytorch tensorflow

Last synced: 12 Feb 2025

https://github.com/soulteary/docker-stable-diffusion-taiyi

带 Web UI 的 Docker taiyi (太乙)，中文 Stable Diffusion WebUI 一键运行环境。

docker pytorch stable-diffusion taiyi webui

Last synced: 23 Jan 2025

https://github.com/donydchen/fmpn-fer

😁[VCIP'19 Oral] Official PyTorch Implementation of Facial Motion Prior Networks for Facial Expression Recognition

expression facial-expression-recognition pytorch vcip-2019

Last synced: 09 Nov 2024

https://github.com/thunlp/re-context-or-names

Bert-based models(BERT, MTB, CP) for relation extraction.

bert contrastive-learning pretrain pytorch relation-extraction

Last synced: 10 Nov 2024

https://github.com/zllrunning/hand-detection.pytorch

FaceBoxes for hand detection in PyTorch

cnn egohands-dataset hand-detection object-detection pytorch

Last synced: 08 Nov 2024

https://github.com/kaylode/vehicle-counting

Vehicle counting using Pytorch

deepsort machine-learning object-detection object-tracking pytorch vehicle-counting vehicle-tracking yolov5

Last synced: 09 Nov 2024

https://github.com/linxueyuanstdio/pytorch-ddpm

500 行代码实现降噪扩散模型 DDPM，干净无依赖

classifier-free ddpm pytorch

Last synced: 19 Nov 2024

https://github.com/lonepatient/electra_pytorch

ELECTRA: Pre-training Text Encoders as Discriminators Rather Than Generators

bert deeplearning electra glue language-model nlp pytorch

Last synced: 06 Nov 2024

https://github.com/zhiyiYo/Alpha-Gobang-Zero

A gobang robot based on reinforcement learning.

alphazero gobang pyqt5 pytorch reinforcement-learning

Last synced: 14 Nov 2024

https://github.com/sophiefy/StellaVoiceChanger

Deep-learning-based voice changer, supporting local inference.

deep-learning python pytorch voice-conversion

Last synced: 18 Nov 2024

https://github.com/devsinghsachan/emdr2

Code and Models for the paper "End-to-End Training of Multi-Document Reader and Retriever for Open-Domain Question Answering" (NeurIPS 2021)

information-retrieval natural-language-processing natural-questions nlp open-domain-qa open-domain-question-answering pytorch question-answering triviaqa webq

Last synced: 20 Dec 2024

https://github.com/sberbank-ai-lab/pytorch-lifestream

A library built upon PyTorch for building embeddings on discrete event sequences using self-supervision

machine-learning pytorch self-supervised-learning

Last synced: 27 Nov 2024

https://github.com/epwalsh/nlp-models

NLP research experiments, built on PyTorch within the AllenNLP framework.

allennlp nlp pytorch pytorch-nlp

Last synced: 01 Nov 2024

https://github.com/maxhalford/pytorch-resample

🎲 Iterable dataset resampling in PyTorch

imbalanced-learning oversampling pytorch resampling undersampling

Last synced: 02 Nov 2024

https://github.com/kamalkraj/stable-diffusion-tritonserver

Deploy stable diffusion model with onnx/tenorrt + tritonserver

deploy docker fp16 inference machine-learning nvidia onnx python3 pytorch stablediffusion tensorrt tensorrt-inference transformers triton-inference-server

Last synced: 07 Nov 2024

https://github.com/edvardlindelof/graph-neural-networks-for-drug-discovery

astrazeneca deep-learning drug-discovery graph-neural-networks pytorch

Last synced: 30 Oct 2024

https://github.com/zhiyiyo/alpha-gobang-zero

A gobang robot based on reinforcement learning.

alphazero gobang pyqt5 pytorch reinforcement-learning

Last synced: 27 Oct 2024

https://github.com/thudm/grb

Graph Robustness Benchmark: A scalable, unified, modular, and reproducible benchmark for evaluating the adversarial robustness of Graph Machine Learning.

adversarial-attacks deep-learning graph-neural-networks machine-learning pytorch

Last synced: 19 Dec 2024

https://github.com/tnbar/tednet

TedNet: A Pytorch Toolkit for Tensor Decomposition Networks

artificial-intelligence deep-neural-networks pytorch tensor-decomposition

Last synced: 27 Jan 2025

https://github.com/MaxHalford/pytorch-resample

🎲 Iterable dataset resampling in PyTorch

imbalanced-learning oversampling pytorch resampling undersampling

Last synced: 15 Nov 2024

https://github.com/jonasricker/diffusion-model-deepfake-detection

[VISAPP2024] Towards the Detection of Diffusion Model Deepfakes

deepfake-detection deepfakes diffusion-models gan pytorch

Last synced: 14 Feb 2025

https://github.com/jahongir7174/efficientnetv2

EfficientNetV2 implementation using PyTorch

efficientnetv2 imagenet pytorch training

Last synced: 18 Nov 2024

https://github.com/cedrickchee/data-science-notebooks

Data science Python notebooks—a collection of Jupyter notebooks on machine learning, deep learning, statistical inference, data analysis and visualization.

data-science deep-learning fastai kaggle keras machine-learning notebooks numpy pandas python pytorch tensorflow