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/frgfm/holocron

PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Other additions: AdEMAMix

ademamix computer-vision cspdarknet53 darknet deep-learning object-detection pytorch resnet rexnet tridentnet unet-image-segmentation yolo yolov4

Last synced: 08 Feb 2025

https://github.com/ecs-vlc/FMix

Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'

augmentation pytorch pytorch-lightning tensorflow torchbearer

Last synced: 13 Nov 2024

https://github.com/yuanyuanli85/Fast_Human_Pose_Estimation_Pytorch

Pytorch Code for CVPR2019 paper "Fast Human Pose Estimation" https://arxiv.org/abs/1811.05419

knowledge-distillation pose-estimation pytorch stacked-hourglass-networks

Last synced: 05 Nov 2024

https://github.com/bknyaz/graph_nn

Graph Classification with Graph Convolutional Networks in PyTorch (NeurIPS 2018 Workshop)

gcn graph graph-classification graph-convolutional-networks graph-unet multigraph pooling pytorch

Last synced: 12 Nov 2024

https://github.com/liuruoze/mini-alphastar

(JAIR'2022) A mini-scale reproduction code of the AlphaStar program. Note: the original AlphaStar is the AI proposed by DeepMind to play StarCraft II. JAIR = Journal of Artificial Intelligence Research.

deep-learning deep-neural-networks deep-reinforcement-learning gaming imitation-learning mini-alphastar multi-agent-systems pytorch reinforcement-learning sc2replay self-playing-bot starcraft-ii-bot starcraft2 starcraft2-ai supervised-learning

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

https://github.com/wohlert/generative-query-network-pytorch

Generative Query Network (GQN) in PyTorch as described in "Neural Scene Representation and Rendering"

deepmind generative-models gqn pytorch

Last synced: 27 Nov 2024

https://github.com/yandexdataschool/deep_vision_and_graphics

Course about deep learning for computer vision and graphics co-developed by YSDA and Skoltech.

computer-vision course course-materials deep-learning pytorch

Last synced: 10 Feb 2025

https://github.com/frgfm/Holocron

PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Other additions: AdEMAMix

ademamix computer-vision cspdarknet53 darknet deep-learning object-detection pytorch resnet rexnet tridentnet unet-image-segmentation yolo yolov4

Last synced: 05 Nov 2024

https://github.com/yinboc/dgp

Rethinking Knowledge Graph Propagation for Zero-Shot Learning, in CVPR 2019

knowledge-graph machine-learning pytorch zero-shot-learning

Last synced: 09 Feb 2025

https://github.com/jzi040941/PercepNet

Unofficial implementation of PercepNet: A Perceptually-Motivated Approach for Low-Complexity, Real-Time Enhancement of Fullband Speech

pytorch speech-enhancement

Last synced: 16 Nov 2024

https://github.com/damitkwr/ESRNN-GPU

PyTorch GPU implementation of the ES-RNN model for time series forecasting

deep-forecasting deep-learning es-rnn forecasting pytorch time-series-forecasting

Last synced: 02 Nov 2024

https://github.com/Sierkinhane/mtcnn-pytorch

A face detection algorithm

computer-vision facedetector pytorch

Last synced: 28 Oct 2024

https://github.com/sciml/scimlbenchmarks.jl

Scientific machine learning (SciML) benchmarks, AI for science, and (differential) equation solvers. Covers Julia, Python (PyTorch, Jax), MATLAB, R

ai ai-for-science benchmark dae differential-equations differentialequations jax julia matlab nerual-differential-equations neural-ode ode partial-differential-equations pde python pytorch scientific-machine-learning sciml sde

Last synced: 08 Feb 2025

https://github.com/sensorsini/v2e

V2E: From video frames to DVS events

event-camera pytorch simulator

Last synced: 05 Feb 2025

https://github.com/stonesjtu/pytorch-nce

The Noise Contrastive Estimation for softmax output written in Pytorch

importance-sampling language-model nce nce-criterion pytorch softmax speedup

Last synced: 05 Feb 2025

https://github.com/sierkinhane/mtcnn-pytorch

A face detection algorithm

computer-vision facedetector pytorch

Last synced: 06 Feb 2025

https://github.com/vita-group/neurallift-360

[CVPR 2023, Highlight] "NeuralLift-360: Lifting An In-the-wild 2D Photo to A 3D Object with 360° Views", Dejia Xu, Yifan Jiang, Peihao Wang, Zhiwen Fan, Yi Wang, Zhangyang Wang

image-to-3d nerf neural-radiance-fields pytorch stable-diffusion text-to-3d

Last synced: 06 Feb 2025

https://github.com/myungsub/CAIN

Source code for AAAI 2020 paper "Channel Attention Is All You Need for Video Frame Interpolation"

aaai2020 channel-attention dataset deep-convolutional-networks frame-interpolation pytorch video-frame-interpolation

Last synced: 28 Nov 2024

https://github.com/hysts/pytorch_mpiigaze_demo

Gaze estimation using MPIIGaze and MPIIFaceGaze

computer-vision gaze gaze-estimation pytorch

Last synced: 08 Feb 2025

https://github.com/JUGGHM/PENet_ICRA2021

ICRA 2021 "Towards Precise and Efficient Image Guided Depth Completion"

depth-completion pytorch

Last synced: 28 Oct 2024

https://github.com/tech-srl/how_attentive_are_gats

Code for the paper "How Attentive are Graph Attention Networks?" (ICLR'2022)

are attention attentive gat gatv2 graph graph-attention-networks graph-neural-networks how networks pytorch

Last synced: 09 Feb 2025

https://github.com/yqyao/FCOS_PLUS

Some improvements (center sample) about FCOS (FCOS: Fully Convolutional One-Stage Object Detection).

anchor-free computer-vision fcos object-detection one-stage pytorch

Last synced: 22 Nov 2024

https://github.com/marian42/shapegan

Generative Adversarial Networks and Autoencoders for 3D Shapes

3d autoencoder gan generative-adversarial-network machine-learning pytorch sdf

Last synced: 07 Feb 2025

https://github.com/kyegomez/screenai

Implementation of the ScreenAI model from the paper: "A Vision-Language Model for UI and Infographics Understanding"

ai artificial-intelligence attention attention-is-all-you-need gpt-4 machine-learning ml pytorch tensorflow

Last synced: 09 Feb 2025

https://github.com/huanglizi/lvit

[IEEE Transactions on Medical Imaging/TMI] This repo is the official implementation of "LViT: Language meets Vision Transformer in Medical Image Segmentation"

medical-image-analysis multimodal-learning pytorch segmentation vision-language

Last synced: 06 Feb 2025

https://github.com/bshall/ZeroSpeech

VQ-VAE for Acoustic Unit Discovery and Voice Conversion

acoustic-features pytorch speech-synthesis voice-conversion vq-vae zerospeech

Last synced: 11 Nov 2024

https://github.com/gasteigerjo/ppnp

PPNP & APPNP models from "Predict then Propagate: Graph Neural Networks meet Personalized PageRank" (ICLR 2019)

deep-learning gcn gnn graph-algorithms graph-classification graph-neural-networks machine-learning pagerank pytorch tensorflow

Last synced: 08 Nov 2024

https://github.com/dionhaefner/pyhpc-benchmarks

A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python :rocket:

benchmarks cupy gpu high-performance-computing jax parallel-computing python pytorch tensorflow

Last synced: 10 Feb 2025

https://github.com/krasserm/fairseq-image-captioning

Transformer-based image captioning extension for pytorch/fairseq

fairseq image-captioning pytorch transformer

Last synced: 07 Feb 2025

https://github.com/rlgraph/rlgraph

RLgraph: Modular computation graphs for deep reinforcement learning

deep-learning deep-reinforcement-learning dqn machine-learning neural-networks ppo pytorch reinforcement-learning tensorflow

Last synced: 02 Nov 2024

https://github.com/chenxi116/PNASNet.pytorch

PyTorch implementation of PNASNet-5 on ImageNet

automl deep-learning imagenet neural-architecture-search pytorch

Last synced: 31 Oct 2024

https://github.com/hzxie/grnet

The official implementation of "GRNet: Gridding Residual Network for Dense Point Cloud Completion". (ECCV 2020)

3d-object-reconstruction aten cubic-feature-sampling eccv-2020 gridding point-cloud-completion pytorch

Last synced: 05 Feb 2025

https://github.com/lromul/argus-freesound

Kaggle | 1st place solution for Freesound Audio Tagging 2019

audio classification deep-learning kaggle pytorch tagging

Last synced: 06 Feb 2025

https://github.com/WangFeng18/3d-gaussian-splatting

Implementation for 3d gaussian splatting

gaussian-splatting nerf pytorch real-time-rendering

Last synced: 19 Nov 2024

https://github.com/andresprados/spiga

SPIGA: Shape Preserving Facial Landmarks with Graph Attention Networks.

300w bmvc2022 cnn cofw face-alignment gnn head-pose-estimation headpose-estimation merlrav pytorch wflw

Last synced: 08 Feb 2025

https://github.com/rlcode/per

Prioritized Experience Replay (PER) implementation in PyTorch

dqn prioritized-experience-replay pytorch

Last synced: 07 Feb 2025

https://github.com/asahi417/lm-question-generation

Multilingual/multidomain question generation datasets, models, and python library for question generation.

bart nlp pytorch question-answering question-generation t5

Last synced: 04 Nov 2024

https://vita-group.github.io/NeuralLift-360/

[CVPR 2023, Highlight] "NeuralLift-360: Lifting An In-the-wild 2D Photo to A 3D Object with 360° Views", Dejia Xu, Yifan Jiang, Peihao Wang, Zhiwen Fan, Yi Wang, Zhangyang Wang

image-to-3d nerf neural-radiance-fields pytorch stable-diffusion text-to-3d

Last synced: 30 Oct 2024

https://github.com/ikostrikov/pytorch-meta-optimizer

A PyTorch implementation of Learning to learn by gradient descent by gradient descent

deep-learning gradient-descent pytorch

Last synced: 07 Feb 2025

https://github.com/tjddus9597/Proxy-Anchor-CVPR2020

Official PyTorch Implementation of Proxy Anchor Loss for Deep Metric Learning, CVPR 2020

cvpr2020 deep-metric-learning image-retrieval proxy-anchor-loss pytorch

Last synced: 03 Dec 2024

https://github.com/tai-wang/depth-from-motion

[ECCV 2022 oral] Monocular 3D Object Detection with Depth from Motion

3d-detection autonomous-driving monocular pytorch robotics structure-from-motion

Last synced: 04 Feb 2025

https://github.com/AppleHolic/source_separation

Deep learning based speech source separation using Pytorch

audio deep-learning pytorch source-separation speech speech-separation

Last synced: 02 Nov 2024

https://github.com/jw9730/tokengt

[NeurIPS'22] Tokenized Graph Transformer (TokenGT), in PyTorch

equivariance gnn graph hypergraph pytorch self-attention transformer

Last synced: 30 Oct 2024

https://github.com/lrjconan/LanczosNetwork

Lanczos Network, Graph Neural Networks, Deep Graph Convolutional Networks, Deep Learning on Graph Structured Data, QM8 Quantum Chemistry Benchmark, ICLR 2019

benchmark graph-convolutional-networks graph-neural-networks iclr2019 lanczos-network pytorch

Last synced: 12 Nov 2024

https://github.com/bytedance/byteir

A model compilation solution for various hardware

llm llvm mlir mlsys onnx pytorch tensorflow

Last synced: 05 Feb 2025

https://github.com/tomrunia/pytorchwavelets

PyTorch implementation of the wavelet analysis from Torrence & Compo (1998)

filtering pytorch signal-processing wavelets

Last synced: 07 Feb 2025

https://github.com/r9y9/tacotron_pytorch

PyTorch implementation of Tacotron speech synthesis model.

python pytorch speech speech-synthesis tacotron

Last synced: 07 Feb 2025

https://github.com/zsef123/EfficientNets-PyTorch

A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."

deep-learning efficientnet pytorch

Last synced: 04 Nov 2024

https://github.com/pangwong/pytorch-multi-label-classifier

A pytorch implemented classifier for Multiple-Label classification

accuracy-curves classifier loss-curves multi-label-classification pytorch visdom

Last synced: 16 Nov 2024

https://github.com/theEricMa/OTAvatar

This is the official repository for OTAvatar: One-shot Talking Face Avatar with Controllable Tri-plane Rendering [CVPR2023].

cvpr2023 deep-learning deepfake face-animation face-reenactment image-animation motion-transfer pose-transfer pytorch talking-head

Last synced: 17 Nov 2024

https://github.com/torchdrift/torchdrift

Drift Detection for your PyTorch Models

ai drift-detection neural-networks pytorch

Last synced: 24 Oct 2024

https://github.com/TorchDrift/TorchDrift

Drift Detection for your PyTorch Models

ai drift-detection neural-networks pytorch

Last synced: 13 Nov 2024

https://github.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement

A minimum unofficial implementation of the "A Convolutional Recurrent Neural Network for Real-Time Speech Enhancement" (CRN) using PyTorch

cnn cnn-rnn pytorch real-time rnn speech-enhancement speech-processing

Last synced: 22 Nov 2024

https://github.com/bfortuner/pytorch_tiramisu

FC-DenseNet in PyTorch for Semantic Segmentation

computer-vision deep-learning neural-network pytorch semantic-segmentation

Last synced: 07 Feb 2025

https://github.com/mmaaz60/mvits_for_class_agnostic_od

[ECCV'22] Official repository of paper titled "Class-agnostic Object Detection with Multi-modal Transformer".

class-agnostic-detection multimodal-learning object-detection open-world-detection psuedo-labels pytorch

Last synced: 07 Feb 2025

https://github.com/alexandre01/UltimateLabeling

A multi-purpose Video Labeling GUI in Python with integrated SOTA detector and tracker

computer-vision labeling labeling-tool object-detection object-tracking python pytorch video

Last synced: 10 Nov 2024

https://github.com/rwightman/posenet-pytorch

A PyTorch port of Google TensorFlow.js PoseNet (Real-time Human Pose Estimation)

deep-learning human-pose-estimation pytorch

Last synced: 05 Feb 2025

https://github.com/HUANGLIZI/LViT

[IEEE Transactions on Medical Imaging/TMI] This repo is the official implementation of "LViT: Language meets Vision Transformer in Medical Image Segmentation"

medical-image-analysis multimodal-learning pytorch segmentation vision-language

Last synced: 28 Nov 2024

https://github.com/Tai-Wang/Depth-from-Motion

[ECCV 2022 oral] Monocular 3D Object Detection with Depth from Motion

3d-detection autonomous-driving monocular pytorch robotics structure-from-motion

Last synced: 28 Oct 2024

https://github.com/microsoft/sarathi-serve

A low-latency & high-throughput serving engine for LLMs

llama llm-inference pytorch transformer

Last synced: 09 Feb 2025

https://github.com/ikostrikov/pytorch-ddpg-naf

Implementation of algorithms for continuous control (DDPG and NAF).

ddpg deep-deterministic-policy-gradient deep-learning pytorch reinforcement-learning

Last synced: 07 Feb 2025

https://github.com/1adrianb/pytorch-estimate-flops

Estimate/count FLOPS for a given neural network using pytorch

convolutional-neural-networks deep-learning flops pytorch pytorch-estimate-flops

Last synced: 08 Feb 2025

https://github.com/alibaba-edu/simple-effective-text-matching-pytorch

A pytorch implementation of the ACL2019 paper "Simple and Effective Text Matching with Richer Alignment Features".

deep-learning nlp pytorch quora-question-pairs snli

Last synced: 06 Nov 2024

https://github.com/bamos/block

An intelligent block matrix library for numpy, PyTorch, and beyond.

deep-learning linear-algebra numpy pytorch

Last synced: 08 Feb 2025

https://github.com/torch-js/torch-js

Node.js binding for PyTorch.

nodejs pytorch

Last synced: 11 Nov 2024

https://github.com/salesforce/multihopkg

Multi-hop knowledge graph reasoning learned via policy gradient with reward shaping and action dropout

action-dropout knowledge-graph multi-hop-reasoning policy-gradient pytorch reinforcement-learning reward-shaping

Last synced: 09 Feb 2025

https://github.com/assemblyai-community/minimagen

MinImagen: A minimal implementation of the Imagen text-to-image model

deep-learning diffusion-models imagen pytorch super-resolution text-to-image

Last synced: 06 Feb 2025

https://github.com/yinboc/prototypical-network-pytorch

A re-implementation of "Prototypical Networks for Few-shot Learning"

few-shot-learning machine-learning pytorch

Last synced: 10 Feb 2025

https://github.com/internlm/internevo

InternEvo is an open-sourced lightweight training framework aims to support model pre-training without the need for extensive dependencies.

910b deepspeed-ulysses flash-attention gemma internlm internlm2 llama3 llava llm-framework llm-training multi-modal pipeline-parallelism pytorch ring-attention sequence-parallelism tensor-parallelism transformers-models zero3

Last synced: 08 Feb 2025

https://github.com/InternLM/InternEvo

InternEvo is an open-sourced lightweight training framework aims to support model pre-training without the need for extensive dependencies.

910b deepspeed-ulysses flash-attention gemma internlm internlm2 llama3 llava llm-framework llm-training multi-modal pipeline-parallelism pytorch ring-attention sequence-parallelism tensor-parallelism transformers-models zero3

Last synced: 30 Oct 2024

https://github.com/megvii-research/FQ-ViT

[IJCAI 2022] FQ-ViT: Post-Training Quantization for Fully Quantized Vision Transformer

imagenet post-training-quantization pytorch quantization vision-transformer

Last synced: 28 Oct 2024

https://github.com/edouardoyallon/pyscatwave

Fast Scattering Transform with CuPy/PyTorch

deep-learning pytorch scattering-transform

Last synced: 06 Feb 2025

https://github.com/quiver-team/torch-quiver

PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.

distributed-computing geometric-deep-learning gpu-acceleration graph-learning graph-neural-networks pytorch

Last synced: 08 Feb 2025

PyTorch Awesome Lists
Awesome-pytorch-list 704 awesome-semantic-segmentation-pytorch 35 Awesome-pytorch-list-CNVersion 692 awesome-colab-notebooks 556 CV-pretrained-model 103 openstl 43 awesome-AutoML-and-Lightweight-Models 150 Awesome-Denoise 164 ai4artists 286 arctic 34 awesome-cuda-triton-hpc 563 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 awesome-mojo-max-mlir 181 PyTorch-Guide 616 awesome-llm 167 awesome-fastai 42 awesome-ai-tools 543 awesome-dl-development 91 free-pytorch-resources 72
PyTorch Categories
Blogs 548 Uncategorized 350 Frameworks 296 Pytorch & related libraries 252 Pytorch & related libraries｜Pytorch & 相关库 240 Research 200 文本数据和NLP 190 Tutorials 188 Blogposts :sparkle: 147 1.) Neural Architecture Search 141 数据可视化 134 机器学习框架 128 Learning Resources 121 图像数据与CV 115 工作流程和实验跟踪 113 Tutorials, books, & examples 110 Tutorials & books & examples｜教程 & 书籍 & 示例 107 模型的可解释性 106 超参数优化和AutoML 101 Learning 99