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-07-29 13:55:27 UTC
JSON Representation

https://github.com/AlignmentResearch/tuned-lens

Tools for understanding how transformer predictions are built layer-by-layer

machine-learning pytorch transformers

Last synced: 31 Jul 2024

https://github.com/lium-lst/nmtpytorch

Sequence-to-Sequence Framework in PyTorch

asr cnn deep-learning multimodality neural-machine-translation nmt pytorch seq2seq speech-recognition

Last synced: 03 Aug 2024

https://github.com/AdamCobb/hamiltorch

PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks

bnn hmc pytorch

Last synced: 31 Jul 2024

https://github.com/PersiaML/PERSIA

High performance distributed framework for training deep learning recommendation models based on PyTorch.

bagua deep-learning distributed-computing machine-learning persia pytorch recommender-system rust-lang

Last synced: 02 Aug 2024

https://github.com/persiaml/persia

High performance distributed framework for training deep learning recommendation models based on PyTorch.

bagua deep-learning distributed-computing machine-learning persia pytorch recommender-system rust-lang

Last synced: 08 Aug 2024

https://github.com/rohan-paul/machinelearning-deeplearning-code-for-my-youtube-channel

The full collection of all codes for my Youtube Channel segregated as per topic.

computer-vision data-science data-science-portfolio datascience deep-learning deep-neural-networks machine-learning machine-learning-algorithms math neural-network python pytorch pytorch-implementation pytorch-tutorial statistics tensorflow tensorflow-examples tensorflow-tutorials tensorflow2 youtube

Last synced: 26 Sep 2024

https://github.com/wuzhihao7788/yolodet-pytorch

reproduce the YOLO series of papers in pytorch, including YOLOv4, PP-YOLO, YOLOv5，YOLOv3, etc.

object-detection ppyolo pytorch yolo yolov3 yolov4 yolov5 yolov5-yolov3

Last synced: 02 Aug 2024

https://github.com/GunhoChoi/Kind-PyTorch-Tutorial

Kind PyTorch Tutorial for beginners

deep-learning python pytorch pytorch-tutorial

Last synced: 03 Aug 2024

https://github.com/maxrmorrison/torchcrepe

Pytorch implementation of the CREPE pitch tracker

crepe pitch pitch-tracker pytorch

Last synced: 05 Aug 2024

https://github.com/OpenMOSS/CoLLiE

Collaborative Training of Large Language Models in an Efficient Way

deep-learning deepspeed nlp pytorch

Last synced: 03 Aug 2024

https://github.com/keenon/nimblephysics

Nimble: Physics Engine for Biomechanics and Deep Learning

biomechanics biomechatronics human-robot-interaction physics-engine pytorch robotics robotics-control robotics-simulation

Last synced: 08 Aug 2024

https://github.com/MarwanDebbiche/post-tuto-deployment

Build and deploy a machine learning app from scratch 🚀

api aws character-level-cnn deployment docker machine-learning pytorch scraping scrapy selenium

Last synced: 07 Aug 2024

https://github.com/microsoft/AzureML-BERT

End-to-End recipes for pre-training and fine-tuning BERT using Azure Machine Learning Service

azure-machine-learning azureml-bert bert bert-model finetuning language-model nlp pretrained-models pretraining pytorch tuning

Last synced: 07 Aug 2024

https://github.com/Microsoft/AzureML-BERT

End-to-End recipes for pre-training and fine-tuning BERT using Azure Machine Learning Service

azure-machine-learning azureml-bert bert bert-model finetuning language-model nlp pretrained-models pretraining pytorch tuning

Last synced: 01 Aug 2024

https://github.com/kyegomez/MultiModalMamba

A novel implementation of fusing ViT with Mamba into a fast, agile, and high performance Multi-Modal Model. Powered by Zeta, the simplest AI framework ever.

ai artificial-intelligence attention-mechanism machine-learning mamba ml pytorch ssm torch transformer-architecture transformers zeta

Last synced: 31 Jul 2024

https://github.com/daniilrobnikov/vits2

VITS2: Improving Quality and Efficiency of Single-Stage Text-to-Speech with Adversarial Learning and Architecture Design

deep-learning pytorch speech speech-synthesis text-to-speech tts vits2 voice-conversion

Last synced: 31 Jul 2024

https://github.com/sshuair/torchsat

🔥TorchSat 🌏 is an open-source deep learning framework for satellite imagery analysis based on PyTorch.

classification data-augmentation deep-learning pytorch remote-sensing satellite satellite-imagery semantic-segmentation torchvision

Last synced: 01 Aug 2024

https://github.com/greydanus/hamiltonian-nn

Code for our paper "Hamiltonian Neural Networks"

deep-learning dynamics hamiltonian neural-network physics pytorch research

Last synced: 31 Jul 2024

https://github.com/aleju/self-driving-truck

Self-Driving Truck in Euro Truck Simulator 2, trained via Reinforcement Learning

autonomous deep-learning pytorch reinforcement-learning self-driving-car

Last synced: 07 Aug 2024

https://github.com/modelscope/AdaSeq

AdaSeq: An All-in-One Library for Developing State-of-the-Art Sequence Understanding Models

bert chinese-nlp crf entity-typing information-extraction multi-modal-ner named-entity-recognition natural-language-processing natural-language-understanding ner nlp pytorch relation-extraction sequence-labeling token-classification word-segmentation

Last synced: 31 Jul 2024

https://github.com/nanoporetech/bonito

A PyTorch Basecaller for Oxford Nanopore Reads

basecalling nanopore pytorch

Last synced: 02 Aug 2024

https://github.com/SymbioticLab/FedScale

FedScale is a scalable and extensible open-source federated learning (FL) platform.

benchmark dataset deep-learning deployment distributed federated-learning icml machine-learning mlsys osdi pytorch tensorflow

Last synced: 02 Aug 2024

https://github.com/vlomme/Multi-Tacotron-Voice-Cloning

Phoneme multilingual(Russian-English) voice cloning based on

deep-learning g2p pytorch russian tacotron tensorflow tts voice-cloning wavernn

Last synced: 07 Aug 2024

https://github.com/vadimkantorov/caffemodel2pytorch

Convert Caffe models to PyTorch

caffe converter pytorch

Last synced: 03 Aug 2024

https://github.com/ternaus/retinaface

The remake of the https://github.com/biubug6/Pytorch_Retinaface

deeplearning face-detection object-detection pytorch retinaface

Last synced: 31 Jul 2024

https://github.com/Tramac/Fast-SCNN-pytorch

A PyTorch Implementation of Fast-SCNN: Fast Semantic Segmentation Network

computer-vision deep-learning fast-scnn pytorch semantic-segmentation

Last synced: 03 Aug 2024

https://github.com/JohnGiorgi/DeCLUTR

The corresponding code from our paper "DeCLUTR: Deep Contrastive Learning for Unsupervised Textual Representations". Do not hesitate to open an issue if you run into any trouble!

allennlp contrastive-learning metric-learning natural-language-processing pytorch representation-learning self-supervised-learning semantic-search semantic-text-similarity sentence-embeddings sentence-similarity transformers

Last synced: 01 Aug 2024

https://github.com/Dana-Farber-AIOS/pathml

Tools for computational pathology

biomedical-image-processing computational-pathology deep-learning digital-pathology fluorescence-microscopy-imaging histopathology image-analysis machine-learning microscopy pathml pathology python pytorch research spatial-transcriptomics

Last synced: 02 Aug 2024

https://github.com/bobazooba/xllm

🦖 X—LLM: Cutting Edge & Easy LLM Finetuning

alpaca bitsandbytes cerebras chatgpt deep-learning deep-neural-networks gpt gpt-4 gptq large-language-models llama llama2 llm mistral openai pytorch torch vicuna zephyr

Last synced: 27 Sep 2024

https://github.com/datawhalechina/torch-rechub

A Lighting Pytorch Framework for Recommendation Models, Easy-to-use and Easy-to-extend.

ctr-prediction pytorch recommendation-system recsys

Last synced: 08 Aug 2024

https://github.com/msamogh/nonechucks

Deal with bad samples in your dataset dynamically, use Transforms as Filters, and more!

data-cleaning data-pipeline data-preprocessing data-processing machine-learning preprocessing pytorch torch

Last synced: 03 Aug 2024

https://github.com/davidtvs/PyTorch-ENet

PyTorch implementation of ENet

camvid cityscape enet pytorch

Last synced: 01 Aug 2024

https://github.com/zuoxingdong/lagom

lagom: A PyTorch infrastructure for rapid prototyping of reinforcement learning algorithms.

artificial-intelligence cem cmaes ddpg deep-deterministic-policy-gradient deep-learning deep-reinforcement-learning evolution-strategies machine-learning mujoco policy-gradient ppo proximal-policy-optimization python pytorch reinforcement-learning research sac soft-actor-critic td3

Last synced: 03 Aug 2024

https://github.com/nupurkmr9/vision-aided-gan

Ensembling Off-the-shelf Models for GAN Training (CVPR 2022 Oral)

computer-graphics computer-vision gans generative-adversarial-network image-generation pretrained-models pytorch

Last synced: 03 Aug 2024

https://github.com/texttron/tevatron

Tevatron - A flexible toolkit for neural retrieval research and development.

dense-retrieval dpr flax information-retrieval jax pytorch question-answering transformer

Last synced: 02 Aug 2024

https://github.com/gordicaleksa/pytorch-GANs

My implementation of various GAN (generative adversarial networks) architectures like vanilla GAN (Goodfellow et al.), cGAN (Mirza et al.), DCGAN (Radford et al.), etc.

conditional-gan dc-gan deep-learning deep-learning-tutorial gan-tutorial gans generative-adversarial-nets generative-adversarial-networks machine-learning python pytorch vanilla-gan

Last synced: 02 Aug 2024

https://github.com/swz30/MIRNetv2

[TPAMI 2022] Learning Enriched Features for Fast Image Restoration and Enhancement. Results on Defocus Deblurring, Denoising, Super-resolution, and image enhancement

contrast-enhancement defocus-deblurring denoising high-resolution image-denoising image-enhancement image-processing image-restoration low-level-vision low-light-image-enhancement pytorch tpami

Last synced: 01 Aug 2024

https://github.com/microsoft/ptgnn

A PyTorch Graph Neural Network Library

deep-learning geometric-deep-learning gnn graph-neural-networks pytorch

Last synced: 24 Sep 2024

https://github.com/operatorai/modelstore

🏬 modelstore is a Python library that allows you to version, export, and save a machine learning model to your filesystem or a cloud storage provider.

data-science keras machine-learning mlops modelstore python-library pytorch s3-storage scikit-learn tensorflow transformer

Last synced: 25 Sep 2024

https://github.com/SmallMunich/nutonomy_pointpillars

Convert pointpillars Pytorch Model To ONNX for TensorRT Inference

nutonomy onnx pointpillars pytorch tensorrt

Last synced: 31 Jul 2024

https://github.com/mir-aidj/all-in-one

All-In-One Music Structure Analyzer

allin1 beat-tracking music-structure-analysis pytorch

Last synced: 05 Aug 2024

https://github.com/mariusbrataas/flowpoints_ml

An intuitive approach to creating deep learning models

deep-learning flowchart js machine-learning python pytorch react tensorflow

Last synced: 03 Aug 2024

https://github.com/cgnorthcutt/benchmarking-keras-pytorch

🔥 Reproducibly benchmarking Keras and PyTorch models

benchmarking cnn cnn-classification cnn-keras deep-learning imagenet keras keras-tutorials pytorch pytorch-tutorial

Last synced: 07 Aug 2024

https://github.com/UM-ARM-Lab/pytorch_mppi

Model Predictive Path Integral (MPPI) with approximate dynamics implemented in pytorch

approximate-dynamics controls model-predictive-control mppi pytorch

Last synced: 31 Jul 2024

https://github.com/arthurdouillard/incremental_learning.pytorch

A collection of incremental learning paper implementations including PODNet (ECCV20) and Ghost (CVPR-W21).

continual-learning deep-learning incremental-learning lifelong-learning pytorch research

Last synced: 02 Aug 2024

https://github.com/mratsim/Amazon-Forest-Computer-Vision

Amazon Forest Computer Vision: Satellite Image tagging code using PyTorch / Keras with lots of PyTorch tricks

computer-vision data-augmentation deep-learning kaggle kaggle-competition keras neural-network-example neural-networks pytorch transfer-learning

Last synced: 03 Aug 2024

https://github.com/cezannec/capsule_net_pytorch

Readable implementation of a Capsule Network as described in "Dynamic Routing Between Capsules" [Hinton et. al.]

capsule-network pytorch pytorch-tutorial

Last synced: 02 Aug 2024

https://github.com/itailang/SampleNet

Differentiable Point Cloud Sampling (CVPR 2020 Oral)

cvpr2020 deep-learning geometry-processing neural-network point-cloud pytorch sampling tensorlow

Last synced: 31 Jul 2024

https://github.com/yifanzhang713/IA-SSD

Not All Points Are Equal: Learning Highly Efficient Point-based Detectors for 3D LiDAR Point Clouds (CVPR 2022, Oral)

3d-object-detection autonomous-driving ia-ssd object-detection point-cloud pytorch

Last synced: 31 Jul 2024

https://github.com/msaroufim/ml-design-patterns

Software Architecture for ML engineers

deep-learning design-patterns python pytorch systems

Last synced: 03 Aug 2024

https://github.com/felixgwu/img_classification_pk_pytorch

Quickly comparing your image classification models with the state-of-the-art models (such as DenseNet, ResNet, ...)

cifar computer-vision deep-learning pytorch

Last synced: 03 Aug 2024

https://github.com/uvipen/Yolo-v2-pytorch

YOLO for object detection tasks

cnn computer-vision deep-learning deep-neural-networks deeplearning object-detection pytorch yolo yolov2 you-only-look-once

Last synced: 01 Aug 2024

https://github.com/Owen-Liuyuxuan/visualDet3D

Official Repo for Ground-aware Monocular 3D Object Detection for Autonomous Driving / YOLOStereo3D: A Step Back to 2D for Efficient Stereo 3D Detection

computer-vision icra monocular-3d-detection object-detection pytorch stereo-3d-detection

Last synced: 31 Jul 2024

https://github.com/Atcold/pytorch-CortexNet

PyTorch implementation of the CortexNet predictive model

deep-learning predictive-modeling pytorch self-supervised unsupervised-learning video

Last synced: 07 Aug 2024

https://github.com/DerryHub/BEVFormer_tensorrt

BEVFormer inference on TensorRT, including INT8 Quantization and Custom TensorRT Plugins (float/half/half2/int8).

bevformer cuda int8-inference pytorch quantization tensorrt-plugins

Last synced: 31 Jul 2024

https://github.com/wuyifan18/DeepLog

Pytorch Implementation of DeepLog.

anomaly-detection deeplearning pytorch sequence-prediction

Last synced: 01 Aug 2024

https://github.com/ReshotAI/gaussian-painters

Gaussian Painters using 3D Gaussian Splatting

3d ai art gaussian-splatting machine-learning nerf painting python pytorch

Last synced: 01 Aug 2024

https://github.com/kach/gradient-descent-the-ultimate-optimizer

Code for our NeurIPS 2022 paper

deep-learning differentiable-programming machine-learning neurips neurips-2022 pytorch

Last synced: 31 Jul 2024

https://github.com/NeuromorphicProcessorProject/snn_toolbox

Toolbox for converting analog to spiking neural networks (ANN to SNN), and running them in a spiking neuron simulator.

brian brian2 caffe deep-learning deep-neural-networks keras lasagne loihi nest pynn pytorch spiking-neural-networks spinnaker tensorflow

Last synced: 02 Aug 2024

https://github.com/benedekrozemberczki/APPNP

A PyTorch implementation of "Predict then Propagate: Graph Neural Networks meet Personalized PageRank" (ICLR 2019).

appnp attention deep-learning deep-neural-networks deepwalk gcn graph-attention graph-classification graph-convolutional-neural-networks graph-embedding graph-neural-network iclr machine-learning network-embedding node-embedding node2vec pagerank ppnp pytorch research

Last synced: 01 Aug 2024

https://github.com/wenqifan03/GraphRec-WWW19

Graph Neural Networks for Social Recommendation, WWW'19

deep-social-recommendations graphrec pytorch recommender-systems socal-recommendation social-network

Last synced: 08 Aug 2024

https://github.com/kakaoenterprise/JORLDY

Repository for Open Source Reinforcement Learning Framework JORLDY

framework gym mlagents pytorch reinforcement-learning

Last synced: 08 Aug 2024

https://github.com/BlackHC/toma

Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory

data-science gpu machine-learning python pytorch

Last synced: 03 Aug 2024

https://github.com/ymfa/seq2seq-summarizer

Pointer-generator reinforced seq2seq summarization in PyTorch

abstractive-summarization attention attention-mechanism coverage coverage-mechanism pointer-network pytorch reinforcement-learning seq2seq summarization summarizer

Last synced: 02 Aug 2024

https://github.com/kefirski/pytorch_RVAE

Recurrent Variational Autoencoder that generates sequential data implemented with pytorch

deep-learning nlp python pytorch vae

Last synced: 01 Aug 2024

https://github.com/dptech-corp/Uni-Fold

An open-source platform for developing protein models beyond AlphaFold.

alphafold alphafold-multimer deep-learning protein-structure pytorch

Last synced: 02 Aug 2024

https://github.com/stepankonev/waymo-motion-prediction-challenge-2022-multipath-plus-plus

Solution for Waymo Motion Prediction Challenge 2022. Our implementation of MultiPath++

autonomous-driving autonomous-vehicles cvpr2022 motion-prediction multipath pytorch sdc self-driving-car trajectory-prediction waymo-open-dataset workshop-autonomous-driving

Last synced: 31 Jul 2024

https://github.com/kevinmusgrave/pytorch-adapt

Domain adaptation made easy. Fully featured, modular, and customizable.

computer-vision deep-learning domain-adaptation machine-learning pytorch transfer-learning

Last synced: 27 Sep 2024

https://github.com/HikariTJU/LD

Localization Distillation for Object Detection (CVPR 2022, TPAMI 2023)

deep-learning detection knowledge-distillation ld object-detection pytorch

Last synced: 01 Aug 2024

https://github.com/uber-research/DeepPruner

DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch (ICCV 2019)

iccv2019 patchmatch pytorch real-time stereo-matching stereo-vision

Last synced: 31 Jul 2024

https://github.com/megvii-research/MOTRv2

[CVPR2023] MOTRv2: Bootstrapping End-to-End Multi-Object Tracking by Pretrained Object Detectors

end-to-end multi-object-tracking pytorch transformer

Last synced: 31 Jul 2024

https://github.com/Megvii-CSG/MegReader

A research project for text detection and recognition using PyTorch 1.2.

ctc deep-learning ocr pytorch text-detection text-detection-recognition text-recognition

Last synced: 01 Aug 2024

https://github.com/davidmascharka/tbd-nets

PyTorch implementation of "Transparency by Design: Closing the Gap Between Performance and Interpretability in Visual Reasoning"

deep-learning machine-learning neural-networks pytorch visual-question-answering visualization vqa

Last synced: 07 Aug 2024

https://github.com/Jungjee/RawNet

Official repository for RawNet, RawNet2, and RawNet3

extracted-speaker-embeddings pytorch rawnet speaker-embeddings speaker-verification spk-embd voxceleb2

Last synced: 01 Aug 2024

https://github.com/talebolano/yolov3-network-slimming

yolov3 network slimming剪枝的一种实现

network pruning pytorch slimming yolo

Last synced: 02 Aug 2024

https://github.com/Haiyang-W/DSVT

[CVPR2023] Official Implementation of "DSVT: Dynamic Sparse Voxel Transformer with Rotated Sets"

3d-object-detection backbone cvpr2023 point-cloud pytorch transformer

Last synced: 31 Jul 2024

https://github.com/alibaba/lightweight-neural-architecture-search

This is a collection of our zero-cost NAS and efficient vision applications.

automl classfication cvpr2023 deepmad iclr2023 icml-2022 maedet nas neurips-2022 object-detection pytorch training-free-nas zennas zero-cost-nas

Last synced: 03 Aug 2024

https://github.com/prigoyal/pytorch_memonger

Experimental ground for optimizing memory of pytorch models

memory-management pytorch

Last synced: 01 Aug 2024

https://github.com/ymcui/PERT

PERT: Pre-training BERT with Permuted Language Model

bert nlp plm pre-trained-model pytorch tensorflow transformers

Last synced: 03 Aug 2024

https://github.com/square/pysurvival

Open source package for Survival Analysis modeling

deep-learning machine-learning numpy python pytorch survival-analysis

Last synced: 02 Aug 2024

https://github.com/emidan19/deep-tempest

Restoration for TEMPEST images using deep-learning

deep-learning gnu-radio image-restoration pytorch sdr tempest

Last synced: 08 Aug 2024

https://github.com/KevinMusgrave/pytorch-adapt

Domain adaptation made easy. Fully featured, modular, and customizable.

computer-vision deep-learning domain-adaptation machine-learning pytorch transfer-learning

Last synced: 01 Aug 2024

https://github.com/atgambardella/pytorch-es

Evolution Strategies in PyTorch

machinelearning python pytorch

Last synced: 01 Aug 2024

https://github.com/vballoli/nfnets-pytorch

NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/

adaptive-gradient-clipping deepmind image-classification nfnets paper pytorch sgd sota

Last synced: 31 Jul 2024

https://github.com/AI4Finance-Foundation/FinRL_Podracer

Cloud-native Financial Reinforcement Learning

algorithmic-trading deep-reinforcement-learning machine-learning-algorithms openai-gym pytorch quant quantitative-finance

Last synced: 08 Aug 2024

https://github.com/nicola-decao/s-vae-pytorch

Pytorch implementation of Hyperspherical Variational Auto-Encoders

deep-learning hyperspherical-vae machine-learning manifold-learning pytorch vae variational-autoencoder von-mises-fisher

Last synced: 05 Aug 2024

https://github.com/apple/ml-gmpi

[ECCV 2022, Oral Presentation] Official PyTorch implementation of GMPI

3d eccv2022 generative-adversarial-network generative-model multiplane-images neural-rendering pytorch

Last synced: 25 Sep 2024

https://github.com/ibm/zshot

Zero and Few shot named entity & relationships recognition

ai deep-learning few-shot few-shot-learning machine-learning named-entity-recognition natural-language-processing natural-language-understanding ned ner nlp nlp-library pytorch relation-extraction relationship-extraction spacy transformer zero-shot zero-shot-learning

Last synced: 24 Sep 2024

https://github.com/justusschock/shapenet

PyTorch implementation of "Super-Realtime Facial Landmark Detection and Shape Fitting by Deep Regression of Shape Model Parameters" predicting facial landmarks with up to 400 FPS

deep-learning delira facial-keypoints facial-keypoints-cnn facial-landmarks landmark-detection pytorch shape shape-models shape-prediction shape-predictor

Last synced: 01 Aug 2024

https://github.com/ShirAmir/dino-vit-features

Official implementation for the paper "Deep ViT Features as Dense Visual Descriptors".

co-segmentation computer-vision deep-learning dino part-segmentation pytorch semantic-correspondence vision-transformers

Last synced: 01 Aug 2024

https://github.com/Algolzw/image-restoration-sde

Image Restoration with Mean-Reverting Stochastic Differential Equations, ICML 2023. Winning solution of the NTIRE 2023 Image Shadow Removal Challenge.

deblurring dehazing denoising deraining diffusion-models image-restoration inpainting pytorch score-based stochastic-differential-equations super-resolution