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/kengz/slm-lab

Modular Deep Reinforcement Learning framework in PyTorch. Companion library of the book "Foundations of Deep Reinforcement Learning".

a2c a3c benchmark deep-reinforcement-learning dqn policy-gradient ppo pytorch reinforcement-learning sac

Last synced: 29 Dec 2024

https://github.com/kengz/SLM-Lab

Modular Deep Reinforcement Learning framework in PyTorch. Companion library of the book "Foundations of Deep Reinforcement Learning".

a2c a3c benchmark deep-reinforcement-learning dqn policy-gradient ppo pytorch reinforcement-learning sac

Last synced: 02 Nov 2024

https://github.com/open-compass/vlmevalkit

Open-source evaluation toolkit of large vision-language models (LVLMs), support ~100 VLMs, 40+ benchmarks

chatgpt claude clip computer-vision evaluation gemini gpt gpt-4v gpt4 large-language-models llava llm multi-modal openai openai-api pytorch qwen vit vqa

Last synced: 28 Dec 2024

https://github.com/open-compass/VLMEvalKit

Open-source evaluation toolkit of large vision-language models (LVLMs), support ~100 VLMs, 40+ benchmarks

chatgpt claude clip computer-vision evaluation gemini gpt gpt-4v gpt4 large-language-models llava llm multi-modal openai openai-api pytorch qwen vit vqa

Last synced: 28 Nov 2024

https://github.com/mit-han-lab/torchsparse

[MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.

acceleration pytorch

Last synced: 26 Dec 2024

https://github.com/Jeff-sjtu/HybrIK

Official code of "HybrIK: A Hybrid Analytical-Neural Inverse Kinematics Solution for 3D Human Pose and Shape Estimation", CVPR 2021

3d-pose-estimation cvpr cvpr21 inverse-kinematics pose-estimation pytorch smpl

Last synced: 18 Nov 2024

https://github.com/ikostrikov/pytorch-a3c

PyTorch implementation of Asynchronous Advantage Actor Critic (A3C) from "Asynchronous Methods for Deep Reinforcement Learning".

a3c actor-critic asynch asynchronous-advantage-actor-critic asynchronous-methods deep-learning deep-reinforcement-learning python pytorch pytorch-a3c reinforcement-learning

Last synced: 24 Dec 2024

https://github.com/AshwinRJ/Federated-Learning-PyTorch

Implementation of Communication-Efficient Learning of Deep Networks from Decentralized Data

deep-learning distributed-computing federated-learning python pytorch

Last synced: 13 Nov 2024

https://github.com/interdigitalinc/compressai

A PyTorch library and evaluation platform for end-to-end compression research

compression deep-learning deep-neural-networks machine-learning neural-network python pytorch

Last synced: 26 Dec 2024

https://github.com/pprp/simplecvreproduction

Replication of simple CV Projects including attention, classification, detection, keypoint detection, etc.

attention classification computer-vision cv demo face-detection landmark object-detection paper-reproduction pytorch

Last synced: 27 Dec 2024

https://github.com/ShannonAI/service-streamer

Boosting your Web Services of Deep Learning Applications.

bert deep-learning model-deployment pytorch tensorflow web

Last synced: 04 Nov 2024

https://github.com/oscarknagg/few-shot

Repository for few-shot learning machine learning projects

few-shot-learning machine-learning maml meta-learning miniimagenet omniglot pytorch research

Last synced: 24 Dec 2024

https://github.com/bat67/pytorch-tutorials-examples-and-books

PyTorch tutorials, examples and some books I found 【不定期更新】整理的PyTorch 最新版教程、例子和书籍

deep-learning deep-learning-algorithms deep-learning-tutorial deeplearning learning-pytorch library machine-learning neural-network numpy python python3 pytorch pytorch-ppt pytorch-tutorials slide

Last synced: 23 Dec 2024

https://github.com/descriptinc/descript-audio-codec

State-of-the-art audio codec with 90x compression factor. Supports 44.1kHz, 24kHz, and 16kHz mono/stereo audio.

audio audio-compression codec compression-algorithm deep-learning gans generative-adversarial-network pytorch residual-vector-quantization

Last synced: 22 Nov 2024

https://github.com/dengbocong/nlp-paper

自然语言处理领域下的相关论文（附阅读笔记），复现模型以及数据处理等（代码含TensorFlow和PyTorch两版本）

bert dialogue nlp nlp-machine-learning paper pytorch speech tensorflow2

Last synced: 28 Dec 2024

https://github.com/swz30/mprnet

[CVPR 2021] Multi-Stage Progressive Image Restoration. SOTA results for Image deblurring, deraining, and denoising.

computer-vision cvpr-2021 cvpr2021 cvpr21 image-deblurring image-denoising image-deraining image-restoration low-level-vision multistage-network progressive-restoration pytorch

Last synced: 28 Dec 2024

https://github.com/chengzeyi/stable-fast

Best inference performance optimization framework for HuggingFace Diffusers on NVIDIA GPUs.

cuda deeplearnng diffusers inference-engines openai-triton performance-optimizations pytorch stable-diffusion stable-video-diffusion torch

Last synced: 26 Dec 2024

https://github.com/jhj0517/Whisper-WebUI

A Web UI for easy subtitle using whisper model.

ai gradio open-source python pytorch web-ui whisper

Last synced: 20 Oct 2024

https://github.com/opendilab/lightzero

[NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios (awesome MCTS)

alpha-beta-pruning alphazero atari board-game board-games continuous-control efficientzero gomoku gumbel-muzero gym mcts mcts-algorithm monte-carlo-tree-search muzero pytorch reinforcement-learning sampled-muzero self-play stochastic-muzero tictactoe

Last synced: 25 Dec 2024

https://github.com/huggingface/hmtl

🌊HMTL: Hierarchical Multi-Task Learning - A State-of-the-Art neural network model for several NLP tasks based on PyTorch and AllenNLP

multi-task-learning natural-language-processing nlp pytorch

Last synced: 22 Dec 2024

https://github.com/DengBoCong/nlp-paper

自然语言处理领域下的相关论文（附阅读笔记），复现模型以及数据处理等（代码含TensorFlow和PyTorch两版本）

bert dialogue nlp nlp-machine-learning paper pytorch speech tensorflow2

Last synced: 14 Nov 2024

https://github.com/InterDigitalInc/CompressAI

A PyTorch library and evaluation platform for end-to-end compression research

compression deep-learning deep-neural-networks machine-learning neural-network python pytorch

Last synced: 13 Nov 2024

https://github.com/rahulschand/gpu_poor

Calculate token/s & GPU memory requirement for any LLM. Supports llama.cpp/ggml/bnb/QLoRA quantization

ggml gpu huggingface language-model llama llama2 llamacpp llm pytorch quantization

Last synced: 27 Dec 2024

https://github.com/swz30/MPRNet

[CVPR 2021] Multi-Stage Progressive Image Restoration. SOTA results for Image deblurring, deraining, and denoising.

computer-vision cvpr-2021 cvpr2021 cvpr21 image-deblurring image-denoising image-deraining image-restoration low-level-vision multistage-network progressive-restoration pytorch

Last synced: 06 Nov 2024

https://github.com/lartpang/pytorchtricks

Some tricks of pytorch... :star:

pytorch pytorch-trick pytorch-tutorial tricks

Last synced: 27 Dec 2024

https://github.com/Xilinx/brevitas

Brevitas: neural network quantization in PyTorch

brevitas deep-learning fpga hardware-acceleration neural-networks ptq pytorch qat quantization xilinx

Last synced: 10 Nov 2024

https://github.com/pyretri/pyretri

Open source deep learning based unsupervised image retrieval toolbox built on PyTorch🔥

deep-learning image-retrieval pytorch toolbox

Last synced: 23 Dec 2024

https://github.com/PyRetri/PyRetri

Open source deep learning based unsupervised image retrieval toolbox built on PyTorch🔥

deep-learning image-retrieval pytorch toolbox

Last synced: 28 Oct 2024

https://github.com/open-mmlab/mmengine

OpenMMLab Foundational Library for Training Deep Learning Models

ai computer-vision deep-learning machine-learning python pytorch

Last synced: 24 Dec 2024

https://github.com/abhishekkrthakur/tez

Tez is a super-simple and lightweight Trainer for PyTorch. It also comes with many utils that you can use to tackle over 90% of deep learning projects in PyTorch.

deep-learning deep-neural-networks neural-networks pytorch tez

Last synced: 28 Dec 2024

https://github.com/lartpang/PyTorchTricks

Some tricks of pytorch... :star:

pytorch pytorch-trick pytorch-tutorial tricks

Last synced: 28 Oct 2024

https://github.com/dsgiitr/graph_nets

PyTorch Implementation and Explanation of Graph Representation Learning papers: DeepWalk, GCN, GraphSAGE, ChebNet & GAT.

chebyshev-polynomials deepwalk graph-attention-networks graph-convolutional-networks graph-embedding graph-representation-learning graph-sage node-embedding pytorch

Last synced: 29 Dec 2024

https://github.com/Pointcept/Pointcept

Pointcept: a codebase for point cloud perception research. Latest works: PTv3 (CVPR'24 Oral), PPT (CVPR'24), OA-CNNs (CVPR'24), MSC (CVPR'23)

3d-vision point-cloud pytorch

Last synced: 28 Oct 2024

https://github.com/lightaime/deep_gcns_torch

Pytorch Repo for DeepGCNs (ICCV'2019 Oral, TPAMI'2021), DeeperGCN (arXiv'2020) and GNN1000(ICML'2021): https://www.deepgcns.org

3d-point-clouds bioinformatics cheminformatics computer-vision data-mining deep-gcns deep-learning geometric-deep-learning graph-convolutional-networks graph-neural-networks pytorch science-research social-network

Last synced: 29 Dec 2024

https://github.com/RahulSChand/gpu_poor

Calculate token/s & GPU memory requirement for any LLM. Supports llama.cpp/ggml/bnb/QLoRA quantization

ggml gpu huggingface language-model llama llama2 llamacpp llm pytorch quantization

Last synced: 08 Nov 2024

https://github.com/pkhungurn/talking-head-anime-2-demo

Demo programs for the Talking Head Anime from a Single Image 2: More Expressive project.

ai anime computer-graphics computer-vision deep-learning machine-learning python pytorch vtuber

Last synced: 22 Dec 2024

https://github.com/nvidia-merlin/transformers4rec

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.

bert gtp huggingface language-model nlp pytorch recommender-system recsys seq2seq session-based-recommendation tabular-data transformer xlnet

Last synced: 25 Dec 2024

https://github.com/khrylx/pytorch-rl

PyTorch implementation of Deep Reinforcement Learning: Policy Gradient methods (TRPO, PPO, A2C) and Generative Adversarial Imitation Learning (GAIL). Fast Fisher vector product TRPO.

a2c deep-reinforcement-learning fisher-vectors generative-adversarial-network policy-gradient ppo proximal-policy-optimization pytorch pytorch-rl reinforcement-learning trpo

Last synced: 25 Dec 2024

https://github.com/roatienza/deep-learning-experiments

Videos, notes and experiments to understand deep learning

artificial-intelligence deep-learning deep-learning-tutorial nlp pytorch speech vision

Last synced: 26 Dec 2024

https://github.com/reshamas/fastai_deeplearn_part1

Notes for Fastai Deep Learning Course

deep-learning image-recognition python python3 pytorch

Last synced: 29 Dec 2024

https://github.com/jakesnell/prototypical-networks

Code for the NeurIPS 2017 Paper "Prototypical Networks for Few-shot Learning"

deep-learning few-shot metric-learning nips-2017 omniglot pytorch

Last synced: 23 Dec 2024

https://github.com/xl0/lovely-tensors

Tensors, for human consumption

deep-learning library pytorch statistics visualization

Last synced: 26 Dec 2024

https://github.com/shusentang/deep-learning-with-pytorch-chinese

本仓库将PyTorch官方书籍《Deep learning with PyTorch》（基本摘录版）翻译成中文版并给出可运行的相关代码。

deep-learning-tutorial pytorch pytorch-tutorial

Last synced: 25 Dec 2024

https://github.com/tjiiv-cprg/epro-pnp

[CVPR 2022 Oral, Best Student Paper] EPro-PnP: Generalized End-to-End Probabilistic Perspective-n-Points for Monocular Object Pose Estimation

3d-object-detection 6dof cvpr gauss-newton levenberg-marquardt monocular perspective-n-point pose-estimation pytorch

Last synced: 23 Dec 2024

https://github.com/roatienza/Deep-Learning-Experiments

Videos, notes and experiments to understand deep learning

artificial-intelligence deep-learning deep-learning-tutorial nlp pytorch speech vision

Last synced: 30 Oct 2024

https://github.com/vainf/pytorch-msssim

Fast and differentiable MS-SSIM and SSIM for pytorch.

image-quality ms-ssim pytorch ssim structure-similarity

Last synced: 26 Dec 2024

https://github.com/NVIDIA-Merlin/Transformers4Rec

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.

bert gtp huggingface language-model nlp pytorch recommender-system recsys seq2seq session-based-recommendation tabular-data transformer xlnet

Last synced: 05 Nov 2024

https://github.com/mit-han-lab/gan-compression

[CVPR 2020] GAN Compression: Efficient Architectures for Interactive Conditional GANs

compression conditional-gans cyclegan gans gaugan image-to-image-translation pix2pix pytorch

Last synced: 27 Dec 2024

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

This repository provides the code and model checkpoints for AIMv1 and AIMv2 research projects.

jax large-scale-vision-models mlx pytorch

Last synced: 25 Dec 2024

https://github.com/tjiiv-cprg/EPro-PnP

[CVPR 2022 Oral, Best Student Paper] EPro-PnP: Generalized End-to-End Probabilistic Perspective-n-Points for Monocular Object Pose Estimation

3d-object-detection 6dof cvpr gauss-newton levenberg-marquardt monocular perspective-n-point pose-estimation pytorch

Last synced: 04 Nov 2024

https://github.com/Khrylx/PyTorch-RL

PyTorch implementation of Deep Reinforcement Learning: Policy Gradient methods (TRPO, PPO, A2C) and Generative Adversarial Imitation Learning (GAIL). Fast Fisher vector product TRPO.

a2c deep-reinforcement-learning fisher-vectors generative-adversarial-network policy-gradient ppo proximal-policy-optimization pytorch pytorch-rl reinforcement-learning trpo

Last synced: 11 Nov 2024

https://github.com/sicara/easy-few-shot-learning

Ready-to-use code and tutorial notebooks to boost your way into few-shot learning for image classification.

deep-learning few-shot-classifcation few-shot-learning few-shot-recognition image-classification machine-learning meta-learning pytorch

Last synced: 25 Dec 2024

https://github.com/torchensemble-community/ensemble-pytorch

A unified ensemble framework for PyTorch to improve the performance and robustness of your deep learning model.

bagging deep-learning deeplearning ensemble ensemble-learning gradient-boosting neural-networks pytorch pytorch-tutorial voting-classifier

Last synced: 25 Dec 2024

https://github.com/eduardoleao052/js-pytorch

A JavaScript library like PyTorch, with GPU acceleration.

automatic-differentiation deep-learning javascript-library neural-networks pytorch

Last synced: 27 Dec 2024

https://github.com/piddnad/ddcolor

[ICCV 2023] DDColor: Towards Photo-Realistic Image Colorization via Dual Decoders

computer-vision image-colorization pytorch

Last synced: 27 Dec 2024

https://github.com/TorchEnsemble-Community/Ensemble-Pytorch

A unified ensemble framework for PyTorch to improve the performance and robustness of your deep learning model.

bagging deep-learning deeplearning ensemble ensemble-learning gradient-boosting neural-networks pytorch pytorch-tutorial voting-classifier

Last synced: 15 Nov 2024

https://github.com/piddnad/DDColor

[ICCV 2023] DDColor: Towards Photo-Realistic Image Colorization via Dual Decoders

computer-vision image-colorization pytorch

Last synced: 22 Dec 2024

https://github.com/youyuge34/anime-inpainting

An application tool of edge-connect, which can do anime inpainting and drawing. 动漫人物图片自动修复，去马赛克，填补，去瑕疵

anime computer-vision cv deep-learning gan generative-adversarial-network inpainting mosaic neural-network opencv python pytorch

Last synced: 25 Dec 2024

https://github.com/nvidia/modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

deep-learning machine-learning nvidia-gpu physics pytorch

Last synced: 26 Dec 2024

https://github.com/julvo/reloading

Change Python code while it's running without losing state

deep-learning hot-reload hot-reloading interactive keras machine-learning python pytorch utility

Last synced: 30 Oct 2024

https://github.com/dataflowr/notebooks

code for deep learning courses

deep-learning pytorch tutorials

Last synced: 27 Dec 2024

https://github.com/youyuge34/Anime-InPainting

An application tool of edge-connect, which can do anime inpainting and drawing. 动漫人物图片自动修复，去马赛克，填补，去瑕疵

anime computer-vision cv deep-learning gan generative-adversarial-network inpainting mosaic neural-network opencv python pytorch

Last synced: 29 Oct 2024

https://github.com/carpedm20/discogan-pytorch

PyTorch implementation of "Learning to Discover Cross-Domain Relations with Generative Adversarial Networks"

gan generative-model pytorch unsupervised-learning

Last synced: 25 Dec 2024

https://github.com/carpedm20/DiscoGAN-pytorch

PyTorch implementation of "Learning to Discover Cross-Domain Relations with Generative Adversarial Networks"

gan generative-model pytorch unsupervised-learning

Last synced: 27 Nov 2024

https://github.com/rucaibox/textbox

TextBox 2.0 is a text generation library with pre-trained language models

deep-learning natural-language-generation natural-language-processing pretrained-models python pytorch seq2seq text-generation

Last synced: 28 Dec 2024

https://github.com/res2net/res2net-pretrainedmodels

(ImageNet pretrained models) The official pytorch implemention of the TPAMI paper "Res2Net: A New Multi-scale Backbone Architecture"

backbone jittor multi-scale pytorch res2net

Last synced: 29 Dec 2024

https://github.com/vikparuchuri/zero_to_gpt

Go from no deep learning knowledge to implementing GPT.

deep-learning python pytorch tutorial

Last synced: 29 Dec 2024

https://github.com/RUCAIBox/TextBox

TextBox 2.0 is a text generation library with pre-trained language models

deep-learning natural-language-generation natural-language-processing pretrained-models python pytorch seq2seq text-generation

Last synced: 16 Nov 2024

PyTorch Awesome Lists
Awesome-pytorch-list 704 awesome-semantic-segmentation-pytorch 35 Awesome-pytorch-list-CNVersion 692 awesome-colab-notebooks 539 CV-pretrained-model 103 awesome-AutoML-and-Lightweight-Models 150 openstl 43 Awesome-Denoise 164 ai4artists 286 arctic 34 awesome-cuda-triton-hpc 487 awesome-yolov8-models 43 awesome-python-machine-learning-resources 1,884 awesome-adaptive-computation 173 awesome-NILM-with-code 49 awesome-dotnet-machine-learning 134 awesome-Multi-Document-Summarization 56 PyTorch-Guide 616 awesome-mojo-max-mlir 183 awesome-fastai 42 awesome-llm 167 awesome-ai-tools 535 free-pytorch-resources 72 awesome-dl-development 91
PyTorch Categories
Blogs 500 Uncategorized 339 Frameworks 279 文本数据和NLP 263 Pytorch & related libraries 225 Pytorch & related libraries｜Pytorch & 相关库 213 Research 198 Tutorials 192 数据可视化 180 机器学习框架 171 图像数据与CV 147 超参数优化和AutoML 135 Blogposts :sparkle: 135 1.) Neural Architecture Search 131 工作流程和实验跟踪 114 Learning Resources 107 模型的可解释性 106 Tutorials, books, & examples 104 Tutorials & books & examples｜教程 & 书籍 & 示例 102 Learning 99