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/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: 05 Nov 2024

https://github.com/vita-group/self-pu

[ICML2020] "Self-PU: Self Boosted and Calibrated Positive-Unlabeled Training" by Xuxi Chen, Wuyang Chen, Tianlong Chen, Ye Yuan, Chen Gong, Kewei Chen, Zhangyang Wang

pu-learning pytorch

Last synced: 16 Nov 2024

https://github.com/goktug97/estorch

Evolution Strategy Library

evolution-strategies evolution-strategy-library evolutionary-algorithms mpi neural-network parallel-computing pytorch reinforcement-learning

Last synced: 11 Nov 2024

https://github.com/dayyass/pytorch-ner

Pipeline for training NER models using PyTorch.

deep-learning hacktoberfest lstm machine-learning named-entity-recognition natural-language-processing ner nlp onnx pipeline python pytorch rnn text

Last synced: 07 Nov 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: 16 Nov 2024

https://github.com/bshall/vectorquantizedvae

A PyTorch implementation of "Continuous Relaxation Training of Discrete Latent Variable Image Models"

generative-models pytorch vae vq-vae

Last synced: 13 Nov 2024

https://github.com/tomrunia/pytorchsteerablepyramid

PyTorch implementation of the Complex Steerable Pyramid

batch computer-vision cuda image-processing mkl pyramid pytorch

Last synced: 13 Nov 2024

https://github.com/astorfi/cor-gan

:unlock: COR-GAN: Correlation-Capturing Convolutional Neural Networks for Generating Synthetic Healthcare Records

deep-learning generative-adversarial-network healthcare python pytorch synthetic-dataset-generation

Last synced: 22 Oct 2024

https://github.com/likejazz/jupyter-notebooks

This repo contains Jupyter Notebooks, miscellaneous stuff.

data-science decision-tree deep-learning jupyter-notebook keras machine-learning nlp pytorch random-forest statistics tensorflow

Last synced: 29 Oct 2024

https://github.com/pclubiitk/model-zoo

Implementations of various Deep Learning models in PyTorch and TensorFlow.

3d-vision classification cnn cnn-model deep-learning gans machine-learning model-zoo nlp object-detection pytorch super-resolution tensorflow vae-gan video

Last synced: 15 Nov 2024

https://github.com/ducha-aiki/mods-light-zmq

MODS with external deep descriptors/detectors

affnet cnn computer-vision descriptor hardnet hessian-affine image-matching local-features matching pytorch ransac sift wbs wide-baseline-stereo wxbs

Last synced: 09 Nov 2024

https://github.com/xinario/catgan_pytorch

Unsupervised and Semi-supervised Learning with Categorical Generative Adversarial Networks

catgan dcgan gan generative-adversarial-network pytorch semi-supervised-learning unsupervised-learning

Last synced: 10 Nov 2024

https://github.com/redisai/redisai-examples

RedisAI showcase

onnx pytorch redisai tensorflow

Last synced: 14 Nov 2024

https://github.com/sshkhr/Practical_RL

My solutions to Yandex Practical Reinforcement Learning course in PyTorch and Tensorflow

bandit-algorithms deep-reinforcement-learning evolutionary-algorithms markov-decision-processes monte-carlo-sampling policy-gradient pytorch reinforcement-learning td-learning tensorflow

Last synced: 13 Nov 2024

https://github.com/dl4to/dl4to

DL4TO is a Python library for 3D topology optimization that is based on PyTorch and allows easy integration with neural networks.

3d deep-learning neural-network neural-networks pytorch topology-optimization

Last synced: 08 Nov 2024

https://github.com/Scitator/catalyst-examples

Examples

computer-vision data-science deep-learning deep-neural-networks deep-reinforcement-learning machine-learning python pytorch

Last synced: 07 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/oidiotlin/torchtracer

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

lab machine-learning pytorch tools

Last synced: 31 Oct 2024

https://github.com/mryab/learning-at-home

"Towards Crowdsourced Training of Large Neural Networks using Decentralized Mixture-of-Experts" (NeurIPS 2020), original PyTorch implementation

deep-learning distributed-systems mixture-of-experts pytorch

Last synced: 22 Oct 2024

https://github.com/xushenlz/parksim

Vehicle simualtion and behavior prediction in parking lots.

deep-learning motion-prediction pytorch simulation

Last synced: 28 Oct 2024

https://github.com/tlesort/generative_continual_learning

continual-learning continuous-learning generative-adversarial-network generative-models incremental-learning lifelong-learning pytorch variational-autoencoder

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

https://github.com/Abe404/root_painter

RootPainter: Deep Learning Segmentation of Biological Images with Corrective Annotation

biological-images deep-learning gui human-in-the-loop human-in-the-loop-machine-learning interactive-machine-learning interactive-segmentation interactive-training machine-teaching painter pytorch root root-painter segmentation

Last synced: 14 Nov 2024

https://github.com/willyfh/visualtorch

VisualTorch aims to help visualize Torch-based neural network architectures.

architecture-diagram architecture-visualization artificial-intelligence deep-learning machine-learning neural-network pytorch torch visualization visualtorch

Last synced: 03 Nov 2024

https://github.com/vexilligera/strokenet

a neural network that draws digits using strokes

deep-learning nodejs pytorch reinforcement-learning

Last synced: 12 Nov 2024

https://github.com/lonepatient/ernie-text-classification-pytorch

This repo contains a PyTorch implementation of a pretrained ERNIE model for text classification.

bert chinese-text-classification ernie python3 pytorch text-classification

Last synced: 06 Nov 2024

https://github.com/princeton-nlp/datamux

[NeurIPS 2022] DataMUX: Data Multiplexing for Neural Networks

deep-learning pytorch

Last synced: 11 Nov 2024

https://github.com/zjysteven/dverge

Pytorch implementation of our NeurIPS'20 *Oral* paper "DVERGE: Diversifying Vulnerabilities for Enhanced Robust Generation of Ensembles".

adversarial-attack-and-defense adversarial-robustness diversifying-vulnerabilities ensembles pytorch

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

https://github.com/shaoxiongji/fed-att

Attentive Federated Learning for Private NLM

attention-mechanism deep-learning federated-learning language-modeling pytorch

Last synced: 07 Nov 2024

https://github.com/ari-dasci/S-TSFE-DL

Time Series Feature Extraction using Deep Learning

cnn convolutional deep-learning feature-extraction keras lstm neural-networks python3 pytorch pytorch-lightning recurrent rnn tensorflow time-series

Last synced: 30 Oct 2024

https://github.com/zzy979/pytorch-tutorial

PyTorch示例代码；复现GNN模型

graph-neural-network pytorch pytorch-tutorial

Last synced: 15 Nov 2024

https://github.com/crowsonkb/vgg_loss

A VGG-based perceptual loss function for PyTorch.

autoencoder perceptual-losses pytorch vgg

Last synced: 14 Oct 2024

https://github.com/drscotthawley/aeiou

(ML) audio engineering i/o utils

audio machine-learning pytorch utilities

Last synced: 10 Nov 2024

https://github.com/gabrielepicco/deep-learning-flower-identifier

Deep learning network to identify 102 different types of flowers

google-colaboratory image-classification pytorch pytorch-scholarship-challenge

Last synced: 08 Nov 2024

https://github.com/kumarshreshtha/image-registration-cnn

A pytorch implementation of CNN based MRI image registration based on Christodoulidis Stergios et al. "Linear and Deformable Image Registration with 3D Convolutional Neural Networks", 2018.

cnn convolutional-neural-networks deep-learning image-registration medical-imaging pytorch

Last synced: 15 Nov 2024

https://github.com/loudinthecloud/dpwa

Distributed Learning by Pair-Wise Averaging

asynchronous-learning deep-learning distributed-systems gossiping neural-networks python pytorch sgd

Last synced: 14 Nov 2024

https://github.com/kamalkraj/e5-mistral-7b-instruct

Finetune mistral-7b-instruct for sentence embeddings

finetuning huggingface lora mistral-7b peft pytorch sentence-embeddings transformers

Last synced: 07 Nov 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: 11 Nov 2024

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

A pytorch implementation of "Explaining and harnessing adversarial examples"

adversarial-attacks deep-learning pytorch

Last synced: 06 Nov 2024

https://github.com/lxuechen/bdmc

PyTorch implementation of Bidirectional Monte Carlo, Annealed Importance Sampling, and Hamiltonian Monte Carlo.

annealed-importance-sampling bayesian-inference bidirectional-monte-carlo hamiltonian-monte-carlo mcmc pytorch

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

https://github.com/Novartis/torchsurv

Deep survival analysis made easy

deep-learning pytorch survival-analysis

Last synced: 26 Sep 2024

https://github.com/finn-no/recsys_slates_dataset

FINN.no Slate Dataset for Recommender Systems. A dataset containing all interactions (viewed items + response (clicked item / no click) for users over a longer time horizon.

dataset deep-learning pytorch recommender-system

Last synced: 01 Nov 2024

https://github.com/dbraun/pytorchtop-cpumem

PyTorch / libtorch in TouchDesigner based on the CPUMemoryTOP example.

libtorch pytorch touchdesigner

Last synced: 08 Nov 2024

https://github.com/archinetai/cqt-pytorch

An invertible and differentiable implementation of the Constant-Q Transform (CQT).

audio cqt deep-learning pytorch

Last synced: 05 Nov 2024

https://github.com/armour/pytorch-nn-practice

💩 My pytorch neural network practice repo

alexnet cifar10 cifar100 deep-learning densenet imagenet inception neural-network pytorch resnet vgg

Last synced: 07 Nov 2024

https://github.com/andi611/mockingjay-speech-representation

Official Implementation of Mockingjay in Pytorch

apc feature-extraction mockingjay phone-classification phoneme-prediction pytorch pytorch-implementation representation-learning sentiment-classification speaker-classification speaker-recognition speech speech-representation

Last synced: 07 Nov 2024

https://github.com/minkaixu/confvae-icml21

An End-to-End Framework for Molecular Conformation Generation via Bilevel Programming (ICML'21)

bilevel-optimization computational-biology computational-chemistry conformation generative-models graph-neural-networks icml-2021 molecule pytorch vae

Last synced: 09 Oct 2024

https://github.com/moverseai/moai

moai is a PyTorch-based AI Model Development Kit (MDK) created to improve data-driven model workflows, design and reproducibility.

artificial-intelligence cnn computer-vision convolutional-neural-networks data-driven deep-learning depth-estimation experimentation human-pose-estimation hydra mdk neural-network object-pose-estimation pytorch reproducibility reproducible-research vae vae-pytorch

Last synced: 12 Nov 2024

https://github.com/oke-aditya/quickvision

An Easy To Use PyTorch Computer Vision Library

computer-vision deep-learning pytorch pytorch-lightning

Last synced: 27 Oct 2024

https://github.com/soloist-v/AutoStrike

ai cf cheat crossfire csgo game mouse pytorch screenshot yolo yolov5

Last synced: 09 Nov 2024

https://github.com/escorciav/roi_pooling

ROIPooling for pytorch

conda-environment cupy python37 pytorch roipooling

Last synced: 12 Nov 2024

https://github.com/mdciri/YOLOv7-Bone-Fracture-Detection

YOLOv7 to detect bone fractures on X-ray images

bonefracture bounding-boxes deep-learning deeplearning machine-learning onnx onnx-models open-datasets python pytorch trained-models trained-weights xray-images yolov7

Last synced: 09 Nov 2024

https://github.com/sksq96/pytorch-mdn

Mixture Density Networks (MDN) implemented in PyTorch

mixture-density-networks pytorch

Last synced: 16 Nov 2024

https://github.com/snap-research/mlpinit-for-gnns

[ICLR 2023] MLPInit: Embarrassingly Simple GNN Training Acceleration with MLP Initialization

deep-learning efficient-training feedforward-neural-network gnn graph-algorithm graph-neural-networks multi-layer-perceptron pytorch scalability

Last synced: 15 Nov 2024

https://github.com/bes-dev/pytorch_clip_bbox

Pytorch based library to rank predicted bounding boxes using text/image user's prompts.

computer-vision deep-learning pytorch

Last synced: 02 Oct 2024

https://github.com/icon-lab/I2I-Mamba

Official implementation of I2I-Mamba, an image-to-image translation model based on selective state spaces

artificial-intelligence deeplearning image-synthesis image-to-image-translation mamba mamba-state-space-models medical multi-modal neural-networks pytorch ssm

Last synced: 28 Oct 2024

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

NumPy-style histograms in PyTorch

histogram numpy pytorch

Last synced: 28 Oct 2024

https://github.com/IBM/MAX-Toxic-Comment-Classifier

Detect 6 types of toxicity in user comments.

comments docker-image ibm machine-learning machine-learning-models natural-language-processing natural-language-understanding nlp pytorch

Last synced: 04 Aug 2024

https://github.com/micahcarroll/unimask

Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"

bert-model deep-learning imitation-learning pytorch reinforcement-learning self-supervised-learning

Last synced: 02 Oct 2024

https://github.com/MrinmoiHossain/Udacity-Deep-Learning-Nanodegree

The course is contained knowledge that are useful to work on deep learning as an engineer. Simple neural networks & training, CNN, Autoencoders and feature extraction, Transfer learning, RNN, LSTM, NLP, Data augmentation, GANs, Hyperparameter tuning, Model deployment and serving are included in the course.

convolutional-networks convolutional-neural-networks deep-learning gans generative-adversarial-network long-short-term-memory-models lstms machine-learning nanodegree neural-network nlp pytorch recurrent-neural-networks rnn sentiment-analysis style-transfer transfer-learning udacity udacity-nanodegree

Last synced: 07 Aug 2024

https://github.com/lonepatient/chinese-word2vec-pytorch

word2vec implementation for skip-gram in pytorch

chinese pytorch skip-gram word2vec

Last synced: 06 Nov 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: 16 Nov 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: 04 Nov 2024

https://github.com/phohenecker/pytorch-transformer

A PyTorch implementation of the Transformer model from "Attention Is All You Need".

attention-is-all-you-need deep-learning python3 pytorch

Last synced: 07 Nov 2024

https://github.com/antofuller/configaformers

A python library for highly configurable transformers - easing model architecture search and experimentation.

artificial-intelligence deep-learning pytorch transformers

Last synced: 15 Nov 2024

https://github.com/markfzp/deep-whole-body-control

[CoRL 2022] Deep Whole-Body Control: Learning a Unified Policy for Manipulation and Locomotion

deep-learning locomotion pytorch reinforcement-learning robotics

Last synced: 13 Nov 2024

https://github.com/supercoderhawk/deep-keyphrase

seq2seq based keyphrase generation model sets, including copyrnn copycnn and copytransfomer

copynet keyphrase-extraction keyphrase-generation keyword-extraction pytorch seq2seq

Last synced: 08 Nov 2024

https://github.com/moskomule/ssl-suite

SSL using PyTorch

pytorch semi-supervised-learning

Last synced: 16 Nov 2024

https://github.com/ferdinandzhong/punctuator

A small seq2seq punctuator tool based on DistilBERT

bert bert-ner chinese-nlp deep-learning nlp punctuation pytorch seq2seq

Last synced: 16 Nov 2024

https://github.com/cardwing/codes-for-steering-control

Learning to Steer by Mimicking Features from Heterogeneous Auxiliary Networks (AAAI 2019, oral)

autonomous-driving python pytorch steering-angles tensorflow udacity

Last synced: 30 Oct 2024

https://github.com/crowsonkb/mdmm

The Modified Differential Multiplier Method (MDMM) for PyTorch

constrained-optimization optimization pytorch

Last synced: 14 Oct 2024

https://github.com/j-min/VPGen

Visual Programming for Text-to-Image Generation and Evaluation (NeurIPS 2023)

pytorch step-by-step text-to-image text-to-image-evaluation text-to-image-generation

Last synced: 31 Oct 2024

https://github.com/MR-HosseinzadehTaher/BenchmarkTransferLearning

Official PyTorch Implementation and Pre-trained Models for Benchmarking Transfer Learning for Medical Image Analysis

benchmarking deep-learning domain-adaptation imagenet-models inaturalist-2021 medical-imaging pytorch self-supervised-learning supervised-learning transfer-learning

Last synced: 05 Nov 2024

https://github.com/anshulpaigwar/Frustum-Pointpillars

Frustum-PointPillars: A Multi-Stage Approach for 3D Object Detection using RGB Camera and LiDAR

3d-object-detection autonomous-driving deep-learning f-pointpillars frustrum frustrum-pointpillars frustum-pointnet lidar pedestrian-detection point-cloud pointpillars pytorch real-time rgb sensor-fusion

Last synced: 28 Oct 2024

https://github.com/aimagelab/stage_action_detection

Code of the STAGE module for video action detection

action-detection action-recognition graph-convolutional-networks pytorch

Last synced: 07 Nov 2024

https://github.com/MinkaiXu/ConfVAE-ICML21

An End-to-End Framework for Molecular Conformation Generation via Bilevel Programming (ICML'21)

bilevel-optimization computational-biology computational-chemistry conformation generative-models graph-neural-networks icml-2021 molecule pytorch vae

Last synced: 08 Aug 2024

https://github.com/IBM/MAX-Fast-Neural-Style-Transfer

Generate a new image that mixes the content of a source image with the style of another image.

docker-image machine-learning machine-learning-models pytorch

Last synced: 04 Aug 2024

https://github.com/PennyLaneAI/pennylane-cirq

The PennyLane-Cirq plugin integrates Google's Cirq software library with with PennyLane's quantum machine learning capabilities.

cirq differentiable-computing jax machine-learning python pytorch qml quantum quantum-algorithms quantum-chemistry quantum-computing quantum-machine-learning tensorflow

Last synced: 06 Nov 2024

https://github.com/andi611/Mockingjay-Speech-Representation

Official Implementation of Mockingjay in Pytorch

Last synced: 07 Aug 2024

https://github.com/bage79/nlp4kor

Natural Language Processing for Korean with Deep Learning

deep-learning korean natural-language-processing pytorch tensorflow word2vec

Last synced: 25 Aug 2024

https://github.com/huangcongqing/mmdetection3d-note

mmdetection3d 代码重点注解笔记

3d-object-detection object-detection object-detection-model point-cloud pytorch

Last synced: 15 Nov 2024

https://github.com/rahulbhalley/progressive-growing-of-gans.pytorch

ProGAN with Standard, WGAN, WGAN-GP, LSGAN, BEGAN, DRAGAN, Conditional GAN, InfoGAN, and Auxiliary Classifier GAN training methods

acgan auxiliary-classifier-gan began conditional-gan discriminator dragan generative-adversarial-network generator infogan infogan-pytorch lsgan lsgan-pytorch optimal-transport progressive-gan pytorch tensorboard wasserstein-gan wgan wgan-gp wgans

Last synced: 31 Oct 2024

https://github.com/catalyst-team/alchemy

Experiments logging & visualization

deep-learning experiment-track infrastructure keras machine-learning pytorch reinforcement-learning reproducibility research tensorflow

Last synced: 01 Nov 2024

https://github.com/isl-org/adaptive-surface-reconstruction

Adaptive Surface Reconstruction for 3D Data Processing

3d 3d-reconstruction pointcloud pytorch tensorflow

Last synced: 11 Nov 2024

https://github.com/FerdinandZhong/punctuator

A small seq2seq punctuator tool based on DistilBERT

bert bert-ner chinese-nlp deep-learning nlp punctuation pytorch seq2seq

Last synced: 14 Nov 2024

https://github.com/megengine/icd

This is the official implementation of the paper "Instance-conditional Knowledge Distillation for Object Detection", based on MegEngine and Pytorch.

knowledge-distillation megengine object-detection pytorch

Last synced: 15 Nov 2024

https://github.com/kaleidophon/token2index

A lightweight but powerful library to build token indices for NLP tasks, compatible with major Deep Learning frameworks like PyTorch and Tensorflow.

deep-learning deeplearning i2t i2w indexing itos nlp numpy python pytorch rnn rnns seq2seq stoi t2i tensorflow token transformer transformers w2i

Last synced: 09 Nov 2024

https://github.com/Time0o/colorful-colorization

A from-scratch PyTorch implementation of "Colorful Image Colorization" by Zhang et al.

convolutional-neural-networks deep-learning image-colorization pytorch

Last synced: 14 Nov 2024

https://github.com/lonePatient/bert-sentence-similarity-pytorch

This repo contains a PyTorch implementation of a pretrained BERT model for sentence similarity task.

bert nlp pytorch sentence-similarity text-classification

Last synced: 02 Nov 2024

https://github.com/cadene/recipe1m.bootstrap.pytorch

Retrieve recipes from foodie pictures using Deep Learning and Pytorch

deep-learning metric-learning pytorch

Last synced: 07 Nov 2024

https://github.com/lonepatient/bert-sentence-similarity-pytorch

This repo contains a PyTorch implementation of a pretrained BERT model for sentence similarity task.

bert nlp pytorch sentence-similarity text-classification

Last synced: 06 Nov 2024

https://github.com/iamalexkorotin/Wasserstein2GenerativeNetworks

PyTorch implementation of "Wasserstein-2 Generative Networks" (ICLR 2021)

deep-learning gan generative-adversarial-network optimal-transport pytorch style-transfer

Last synced: 30 Oct 2024

https://github.com/we0091234/yolov8-tensorrt

yolov8 tensorrt 加速

pytorch tensorrt yolov8 yolov8-tensorrt

Last synced: 09 Nov 2024

https://github.com/hristo-vrigazov/dnn.cool

A framework for multi-task learning, where you may precondition tasks and compose them into bigger tasks. Conditional objectives and per-task evaluations and interpretations.

catalyst deep-learning deep-neural-network deep-neural-networks multitask-learning multitasking pytorch task-composition task-precondition