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.

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: 12 Nov 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: 30 Oct 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: 10 Jan 2025

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

Restoration for TEMPEST images using deep-learning

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

Last synced: 28 Nov 2024

https://github.com/gordicaleksa/pytorch-neural-style-transfer

Reconstruction of the original paper on neural style transfer (Gatys et al.). I've additionally included reconstruction scripts which allow you to reconstruct only the content or the style of the image - for better understanding of how NST works.

deep-learning deep-learning-tutorial machine-learning neural-style-transfer neural-style-transfer-pytorch neural-style-transfer-tutorial non-photorealistic-rendering python pytorch style-transfer

Last synced: 14 Jan 2025

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: 27 Nov 2024

https://github.com/datawhalechina/d2l-ai-solutions-manual

《动手学深度学习》习题解答,在线阅读地址如下:

d2l-ai deep-learning pytorch

Last synced: 12 Jan 2025

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: 12 Jan 2025

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: 13 Jan 2025

https://github.com/harritaylor/torchvggish

Pytorch port of Google Research's VGGish model used for extracting audio features.

audio-embedding audioset pytorch vggish

Last synced: 12 Jan 2025

https://github.com/intelligent-machine-learning/glake

GLake: optimizing GPU memory management and IO transmission.

deepspeed gpu llm memory onnx pytorch

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

https://github.com/he-y/soft-filter-pruning

Soft Filter Pruning for Accelerating Deep Convolutional Neural Networks

model-compression pruning pytorch

Last synced: 14 Jan 2025

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: 16 Nov 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: 15 Jan 2025

https://github.com/microsoft/ptgnn

A PyTorch Graph Neural Network Library

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

Last synced: 11 Jan 2025

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: 14 Jan 2025

https://github.com/osai-ai/tensor-stream

A library for real-time video stream decoding to CUDA memory

c-plus-plus cuda python pytorch video video-processing

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

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

PyTorch implementation of ENet

camvid cityscape enet pytorch

Last synced: 04 Nov 2024

https://github.com/SmallMunich/nutonomy_pointpillars

Convert pointpillars Pytorch Model To ONNX for TensorRT Inference

nutonomy onnx pointpillars pytorch tensorrt

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

https://github.com/shobrook/sequitur

Library of autoencoders for sequential data

autoencoder autoencoder-model lstm pytorch sequence time-series

Last synced: 11 Jan 2025

https://github.com/pointcept/pointtransformerv2

[NeurIPS'22] An official PyTorch implementation of PTv2.

3d-vision point-cloud pytorch

Last synced: 14 Jan 2025

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

PyTorch implementation of DeepDream algorithm (Mordvintsev et al.). Additionally I've included playground.py to help you better understand basic concepts behind the algo.

deep-dream-tutorial deep-learning deep-learning-tutorial deepdream machine-learning python pytorch

Last synced: 13 Jan 2025

https://github.com/chineseocr/trocr-chinese

transformers ocr for chinese

chinese-ocr pytorch seq2seq

Last synced: 14 Jan 2025

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

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: 10 Jan 2025

https://github.com/taokong/foveabox

FoveaBox: Beyond Anchor-based Object Detector

anchor-free coco detection foveabox object-detection pytorch

Last synced: 11 Jan 2025

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: 12 Nov 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: 15 Nov 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: 15 Jan 2025

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

https://github.com/wuyifan18/DeepLog

Pytorch Implementation of DeepLog.

anomaly-detection deeplearning pytorch sequence-prediction

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

https://github.com/ibm/aihwkit

IBM Analog Hardware Acceleration Kit

ai analog-devices cuda neural-networks pytorch

Last synced: 13 Jan 2025

https://github.com/pmixer/sasrec.pytorch

PyTorch(1.6+) implementation of https://github.com/kang205/SASRec

pytorch recommender-system sasrec sequential-models

Last synced: 12 Jan 2025

https://github.com/svip-lab/planarreconstruction

[CVPR'19] Single-Image Piece-wise Planar 3D Reconstruction via Associative Embedding

computer-vision cvpr2019 deep-learning pytorch

Last synced: 12 Jan 2025

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: 28 Oct 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: 14 Jan 2025

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: 27 Nov 2024

https://github.com/vanderschaarlab/synthcity

A library for generating and evaluating synthetic tabular data for privacy, fairness and data augmentation.

data-augmentation fairness-ml generative-model machine-learning privacy pytorch synthetic-data tabular-data

Last synced: 15 Jan 2025

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: 28 Oct 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: 12 Nov 2024

https://github.com/amanchadha/iseebetter

iSeeBetter: Spatio-Temporal Video Super Resolution using Recurrent-Generative Back-Projection Networks | Python3 | PyTorch | GANs | CNNs | ResNets | RNNs | Published in Springer Journal of Computational Visual Media, September 2020, Tsinghua University Press

cnn convolutional-neural-network convolutional-neural-networks gan gans generative-adversarial-network psnr pytorch recurrent-neural-networks residual-networks resnet resnets rnn self-supervised-learning srgan ssim super-resolution unsupervised-learning video-processing video-super-resolution

Last synced: 12 Jan 2025

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: 28 Oct 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: 09 Nov 2024

https://github.com/lucidrains/axial-attention

Implementation of Axial attention - attending to multi-dimensional data efficiently

artificial-intelligence attention-mechanism deep-learning pytorch

Last synced: 11 Jan 2025

https://github.com/prigoyal/pytorch_memonger

Experimental ground for optimizing memory of pytorch models

memory-management pytorch

Last synced: 15 Jan 2025

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: 04 Nov 2024

https://github.com/lightning-ai/dl-fundamentals

Deep Learning Fundamentals -- Code material and exercises

ai deep-learning machine-learning python pytorch

Last synced: 12 Jan 2025

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: 11 Jan 2025

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

https://github.com/minar09/cp-vton-plus

Official implementation for "CP-VTON+: Clothing Shape and Texture Preserving Image-Based Virtual Try-On", CVPRW 2020

2d-virtual-try-on cp-vton-plus emcom emcom-lab image-based official-implementation pytorch pytorch-implementation seoultech virtual-try-on virtual-tryon vton

Last synced: 12 Jan 2025

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

https://github.com/hysts/pytorch_mpiigaze

An unofficial PyTorch implementation of MPIIGaze and MPIIFaceGaze

computer-vision gaze gaze-estimation pytorch

Last synced: 12 Jan 2025

https://github.com/kakaoenterprise/JORLDY

Repository for Open Source Reinforcement Learning Framework JORLDY

framework gym mlagents pytorch reinforcement-learning

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

https://github.com/kefirski/pytorch_RVAE

Recurrent Variational Autoencoder that generates sequential data implemented with pytorch

deep-learning nlp python pytorch vae

Last synced: 02 Nov 2024

https://github.com/quantylab/rltrader

파이썬과 케라스를 이용한 딥러닝/강화학습 주식투자 - 퀀트 투자, 알고리즘 트레이딩을 위한 최첨단 해법 입문 (개정판)

deep-learning keras python pytorch reinforcement-learning tensorflow

Last synced: 10 Jan 2025

https://github.com/quva-lab/escnn

Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/

deep-learning equivariance geometric-deep-learning geometry invariant isometries pytorch rotations symmetries

Last synced: 10 Jan 2025

https://github.com/QUVA-Lab/escnn

Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/

deep-learning equivariance geometric-deep-learning geometry invariant isometries pytorch rotations symmetries

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

https://github.com/ymcui/PERT

PERT: Pre-training BERT with Permuted Language Model

bert nlp plm pre-trained-model pytorch tensorflow transformers

Last synced: 16 Nov 2024

https://github.com/utkuozbulak/pytorch-cnn-adversarial-attacks

Pytorch implementation of convolutional neural network adversarial attack techniques

adversarial-example fast-gradient-sign fooling-images gradient-ascent pytorch pytorch-tutorial

Last synced: 15 Jan 2025

https://github.com/ybabakhin/kaggle_salt_bes_phalanx

Winning solution for the Kaggle TGS Salt Identification Challenge.

deep-learning image-segmentation keras pseudo-labeling pytorch semantic-segmentation u-net

Last synced: 15 Jan 2025

https://github.com/fendouai/pytorch1.0-cn

PyTorch 1.0 官方文档 中文版,欢迎关注微信公众号:磐创AI

pytorch pytorch-nlp pytorch-tutorial

Last synced: 15 Jan 2025

https://github.com/RLE-Foundation/RLeXplore

RLeXplore provides stable baselines of exploration methods in reinforcement learning, such as intrinsic curiosity module (ICM), random network distillation (RND) and rewarding impact-driven exploration (RIDE).

baselines efficient-algorithm exploration-strategy gym machine-learning pybullet pytorch reinforcement-learning robotics toolbox

Last synced: 04 Dec 2024

https://github.com/square/pysurvival

Open source package for Survival Analysis modeling

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

Last synced: 12 Nov 2024

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

Unofficial PyTorch Implementation for HifiFace (https://arxiv.org/abs/2106.09965)

face-swapping pytorch pytorch-lightning wandb

Last synced: 14 Jan 2025

https://github.com/dvl-tum/mot_neural_solver

Official PyTorch implementation of "Learning a Neural Solver for Multiple Object Tracking" (CVPR 2020 Oral).

graph-neural-networks multi-object-tracking pytorch

Last synced: 28 Oct 2024

https://github.com/official-stockfish/nnue-pytorch

Stockfish NNUE (Chess evaluation) trainer in Pytorch

chess deep-learning pytorch stockfish

Last synced: 11 Jan 2025

https://github.com/beomi/infinitransformer

Unofficial PyTorch/🤗Transformers(Gemma/Llama3) implementation of Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention

gemma huggingface infinitransformer llama llama3 pytorch transformers

Last synced: 12 Jan 2025

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

https://github.com/francescosaveriozuppichini/pytorch-deep-learning-template

A Pytorch Computer Vision template to quick start your next project! 🚀🚀

computer-vision deep-learning python pytorch

Last synced: 12 Jan 2025

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: 28 Oct 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: 14 Jan 2025

https://github.com/imoonlab/meshnet

MeshNet: Mesh Neural Network for 3D Shape Representation (AAAI 2019)

3d-shapes deep-learning mesh pytorch

Last synced: 12 Jan 2025