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/baiyuetribe/ncnn-models

awesome AI models with NCNN, and how they were converted ✨✨✨

deep-learning gfpgan image-classification nanodet ncnn ncnn-model pytorch rife yolov5 yolov7 yolox

Last synced: 16 Jan 2025

https://github.com/bytedance/flux

A fast communication-overlapping library for tensor parallelism on GPUs.

cuda cutlass gpu pytorch

Last synced: 10 Jan 2025

https://github.com/VITA-Group/SLaK

[ICLR 2023] "More ConvNets in the 2020s: Scaling up Kernels Beyond 51x51 using Sparsity"; [ICML 2023] "Are Large Kernels Better Teachers than Transformers for ConvNets?"

51x51 convnet deep-learning dynamic-sparsity knowledge-distillation large-kernels object-detection pytorch segmentation

Last synced: 27 Oct 2024

https://github.com/jinglescode/time-series-forecasting-pytorch

Acquiring data from Alpha Vantage and predicting stock prices with PyTorch's LSTM

deep-learning lstm-model machine-learning neural-network python pytorch time-series-forecasting

Last synced: 13 Jan 2025

https://github.com/roboflow/roboflow-100-benchmark

Code for replicating Roboflow 100 benchmark results and programmatically downloading benchmark datasets

computer-vision dataset deep-learning machine-learning object-detection pytorch

Last synced: 13 Jan 2025

https://github.com/cmhungsteve/TA3N

[ICCV 2019 (Oral)] Temporal Attentive Alignment for Large-Scale Video Domain Adaptation (PyTorch)

action-recognition cvpr2019 domain-adaptation domain-discrepancy iccv2019 pytorch temporal-dynamics video video-classification video-da-datasets

Last synced: 03 Nov 2024

https://github.com/yuyangw/molclr

Implementation of MolCLR: "Molecular Contrastive Learning of Representations via Graph Neural Networks" in PyG.

deep-learning graph-neural-networks molecule pytorch pytorch-geometric self-supervised-learning

Last synced: 15 Jan 2025

https://github.com/nvidia/bionemo-framework

BioNeMo Framework: For building and adapting AI models in drug discovery at scale

drug-discovery gpu machine-learning pytorch

Last synced: 11 Jan 2025

https://github.com/gmihaila/ml_things

This is where I put things I find useful that speed up my work with Machine Learning. Ever looked in your old projects to reuse those cool functions you created before? Well, this repo is designed to be a Python Library of functions I created in my previous project that can be reused. I also share some Notebooks Tutorials and Python Code Snippets.

google-colab machine-learning nlp nlp-machine-learning notebooks python-snippets pytorch snippets transformer

Last synced: 12 Jan 2025

https://github.com/johnolafenwa/torchfusion

A modern deep learning framework built to accelerate research and development of AI systems

convolutional-neural-networks deep-learning gan machine-learning neural-network python pytorch visualization

Last synced: 12 Jan 2025

https://github.com/johnolafenwa/TorchFusion

A modern deep learning framework built to accelerate research and development of AI systems

convolutional-neural-networks deep-learning gan machine-learning neural-network python pytorch visualization

Last synced: 27 Nov 2024

https://github.com/nikhilbarhate99/min-decision-transformer

Minimal implementation of Decision Transformer: Reinforcement Learning via Sequence Modeling in PyTorch for mujoco control tasks in OpenAI gym

deep-learning deep-reinforcement-learning machine-learning mujoco offline-reinforcement-learning openai-gym pytorch pytorch-transformers reinforcement-learning robotics transformer

Last synced: 16 Jan 2025

https://github.com/nyumedml/gnn_for_ehr

Code for "Graph Neural Network on Electronic Health Records for Predicting Alzheimer’s Disease"

alzheimer-disease-prediction deep-learning disease-prediction ehr electronic-health-records gnn graph-neural-networks pytorch

Last synced: 15 Jan 2025

https://github.com/dgriff777/a3c_continuous

A continuous action space version of A3C LSTM in pytorch plus A3G design

a3c a3c-gpu a3c-lstm a3g openai-gym pytorch pytorch-a3c

Last synced: 28 Sep 2024

https://github.com/lucidrains/sinkhorn-transformer

Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention

artificial-intelligence attention-mechanism deep-learning pytorch transformers

Last synced: 12 Jan 2025

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

[CVPR'20] Fast-MVSNet: Sparse-to-Dense Multi-View Stereo With Learned Propagation and Gauss-Newton Refinement

cvpr2020 multi-view-stereo pytorch

Last synced: 14 Jan 2025

https://github.com/orktes/go-torch

LibTorch (PyTorch) bindings for Golang

golang golang-library machine-learning pytorch

Last synced: 16 Jan 2025

https://github.com/ruhyadi/yolo3d

YOLO 3D Object Detection for Autonomous Driving Vehicle

3d-object-detection deep-learning pytorch self-driving-car yolo

Last synced: 15 Jan 2025

https://github.com/zc-alexfan/hold

[CVPR 2024✨Highlight] Official repository for HOLD, the first method that jointly reconstructs articulated hands and objects from monocular videos without assuming a pre-scanned object template and 3D hand-object training data.

3d-reconstruction ai artificial-intelligence augmented-reality computer-vision hand-object-interaction hand-object-reconstruction hand-tracking mano mixed-reality neural-networks pose-estimation pytorch virtual-reality

Last synced: 13 Nov 2024

https://github.com/wutianyiRosun/CGNet

CGNet: A Light-weight Context Guided Network for Semantic Segmentation [IEEE Transactions on Image Processing 2020]

camvid cityscapes pytorch semantic-segmentation

Last synced: 18 Nov 2024

https://github.com/misads/easy_detection

一个简单方便的目标检测框架(PyTorch环境可直接运行,不需要cuda编译),支持Faster_RCNN、Cascade_RCNN、Yolo系列、SSD等经典网络。

coco detection pytorch voc yolo

Last synced: 09 Nov 2024

https://github.com/francescosaveriozuppichini/mirror

Visualisation tool for CNNs in pytorch

deep-learning neural-network python pytorch visualis

Last synced: 16 Jan 2025

https://github.com/paperspace/pytorch-101-tutorial-series

PyTorch 101 series covering everything from the basic building blocks all the way to building custom architectures.

101 deep-learning neural-networks pytorch pytorch-tutorial sample-code

Last synced: 16 Jan 2025

https://github.com/g1910/HyperNetworks

PyTorch implementation of HyperNetworks (Ha et al., ICLR 2017) for ResNet (Residual Networks)

hypernetworks metalearning pytorch resnet

Last synced: 11 Nov 2024

https://github.com/ltkong218/IFRNet

IFRNet: Intermediate Feature Refine Network for Efficient Frame Interpolation (CVPR 2022)

deep-learning deep-networks frame-interpolation knowledge-distillation optical-flow pytorch real-time slow-motion video view-synthesis

Last synced: 20 Oct 2024

https://github.com/cloneofsimo/consistency_models

Unofficial Implementation of Consistency Models in pytorch

generative-model pytorch

Last synced: 16 Jan 2025

https://github.com/andreamad8/Universal-Transformer-Pytorch

Implementation of Universal Transformer in Pytorch

pytorch universal-transformer

Last synced: 07 Nov 2024

https://github.com/gitabcworld/FewShotLearning

Pytorch implementation of the paper "Optimization as a Model for Few-Shot Learning"

few-shot optimization optimization-algorithms python pytorch

Last synced: 15 Nov 2024

https://github.com/Baiyuetribe/ncnn-models

awesome AI models with NCNN, and how they were converted ✨✨✨

deep-learning gfpgan image-classification nanodet ncnn ncnn-model pytorch rife yolov5 yolov7 yolox

Last synced: 09 Nov 2024

https://github.com/FrancescoSaverioZuppichini/mirror

Visualisation tool for CNNs in pytorch

deep-learning neural-network python pytorch visualis

Last synced: 15 Nov 2024

https://github.com/Khrylx/AgentFormer

[ICCV 2021] Official PyTorch Implementation of "AgentFormer: Agent-Aware Transformers for Socio-Temporal Multi-Agent Forecasting".

autonomous-driving deep-learning iccv2021 multi-agent-forecasting pytorch trajectory-forecasting trajectory-prediction transformer

Last synced: 28 Oct 2024

https://github.com/donydchen/ganimation_replicate

An Out-of-the-Box Replication of GANimation using PyTorch, pretrained weights are available!

action-units gan ganimation ganimation-pytorch out-of-the-box pretrained-checkpoints pretrained-models pretrained-weights pytorch replicate

Last synced: 16 Jan 2025

https://github.com/picsart-ai-research/semask-segmentation

[NIVT Workshop @ ICCV 2023] SeMask: Semantically Masked Transformers for Semantic Segmentation

ade20k cityscapes coco-stuff-10k pytorch semantic-segmentation semask vision-transformers

Last synced: 17 Jan 2025

https://github.com/lucasjinreal/faceswap_pytorch

Deep fake ready to train on any 2 pair dataset with higher resolution

deepfakes deeplearning faceswap fakeface gan pytorch

Last synced: 16 Jan 2025

https://github.com/woshidandan/TANet-image-aesthetics-and-quality-assessment

[IJCAI 2022, Official Code] for paper "Rethinking Image Aesthetics Assessment: Models, Datasets and Benchmarks". Official Weights and Demos provided. 首个面向多主题场景的美学评估数据集、算法和benchmark.

iaa image-aesthetics-assessment image-quality-assessment pytorch tad66k

Last synced: 13 Nov 2024

https://github.com/tim-salzmann/l4casadi

Use PyTorch Models with CasADi and Acados in Python, C(++) or Matlab

acados casadi cplusplus deep-learning learning-control machine-learning optimization optimization-algorithms python pytorch

Last synced: 14 Jan 2025

https://github.com/savan77/The-GAN-World

Everything about Generative Adversarial Networks

gan generative-adversarial-network pytorch tensorflow

Last synced: 02 Nov 2024

https://github.com/fcakyon/craft-text-detector

Packaged, Pytorch-based, easy to use, cross-platform version of the CRAFT text detector

actions anaconda computer-vision craft deep-learning document hacktoberfest linux macos neural-network ocr pypi python pytorch text text-detection vision windows workflow

Last synced: 03 Nov 2024

https://github.com/pudae/kaggle-hpa

Code for 3rd place solution in Kaggle Human Protein Atlas Image Classification Challenge.

kaggle python pytorch

Last synced: 31 Oct 2024

https://github.com/cure-lab/PnPInversion

[ICLR2024] Official repo for paper "PnP Inversion: Boosting Diffusion-based Editing with 3 Lines of Code"

deep-learning diffusion-inversion diffusion-model diffusion-models iclr2024 image-editing image-generation paper-implementation paper-implementations pytorch

Last synced: 04 Nov 2024

https://github.com/likyoo/Siam-NestedUNet

The pytorch implementation for "SNUNet-CD: A Densely Connected Siamese Network for Change Detection of VHR Images"

change-detection pytorch

Last synced: 01 Nov 2024

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

Official Code of Paper "Reversible Column Networks" "RevColv2"

cnn computer-vision iclr2023 mae pytorch transformer vit

Last synced: 28 Oct 2024

https://github.com/wjn1996/hugnlp

HugNLP is a unified and comprehensive NLP library based on HuggingFace Transformer. Please hugging for NLP now!😊 HugNLP will released to @HugAILab

code-understanding deep-learning few-shot-learning knowledge-enhancement natural-language-processing pre-trained-language-models prompt-based-learning pytorch semi-supervised-learning supervised-learning transformers

Last synced: 16 Jan 2025

https://github.com/jolibrain/joligen

Generative AI Image Toolset with GANs and Diffusion for Real-World Applications

augmented-reality deep-learning diffusion-models gan generative-model image-generation image-to-image pytorch

Last synced: 15 Jan 2025

https://github.com/hkproj/pytorch-llama

LLaMA 2 implemented from scratch in PyTorch

language-model llama llama2 paper-implementations pytorch

Last synced: 15 Jan 2025

https://github.com/l0sg/relational-rnn-pytorch

An implementation of DeepMind's Relational Recurrent Neural Networks (NeurIPS 2018) in PyTorch.

deep-learning deepmind language-model language-modeling pytorch recurrent-neural-networks self-attention transformer word-language-model

Last synced: 21 Nov 2024

https://github.com/hasnainraz/fast-aginggan

A deep learning model to age faces in the wild, currently runs at 60+ fps on GPUs

cnn cycle-gan face-aging faceapp gan generative-adversarial-network neural-network pytorch pytorch-lightning realtime-face-aging tensorboard

Last synced: 15 Jan 2025

https://github.com/coderSkyChen/Action_Recognition_Zoo

Codes for popular action recognition models, verified on the something-something data set.

action-recognition pytorch

Last synced: 05 Nov 2024

https://github.com/L0SG/relational-rnn-pytorch

An implementation of DeepMind's Relational Recurrent Neural Networks (NeurIPS 2018) in PyTorch.

deep-learning deepmind language-model language-modeling pytorch recurrent-neural-networks self-attention transformer word-language-model

Last synced: 27 Nov 2024

https://github.com/jolibrain/joliGEN

Generative AI Image Toolset with GANs and Diffusion for Real-World Applications

augmented-reality deep-learning diffusion-models gan generative-model image-generation image-to-image pytorch

Last synced: 15 Nov 2024

https://github.com/imoonlab/dhgnn

DHGNN source code for IJCAI19 paper: "Dynamic Hypergraph Neural Networks"

graph-embedding nerual-network pytorch

Last synced: 16 Jan 2025

https://github.com/biomedsciai/histocartography

A standardized Python API with necessary preprocessing, machine learning and explainability tools to facilitate graph-analytics in computational pathology.

deep-learning graph-neural-networks healthcare machine-learning pathology pytorch

Last synced: 13 Jan 2025

https://github.com/nathanhubens/fasterai

FasterAI: Prune and Distill your models with FastAI and PyTorch

compression fastai knowledge-distillation pruning pytorch

Last synced: 13 Nov 2024

https://github.com/inferno-pytorch/inferno

A utility library around PyTorch

deep-learning neural-networks pytorch

Last synced: 14 Nov 2024

https://github.com/bingogome/samm

A 3D Slicer integration to Meta's SAM.

pytorch sam

Last synced: 30 Nov 2024

https://github.com/godisboy/SN-GAN

pyTorch implementation of Spectral Normalization for Generative Adversarial Networks

gan pytorch sngan weight-normalization

Last synced: 17 Nov 2024

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

Unofficial PyTorch Implementation of SUM-GAN from "Unsupervised Video Summarization with Adversarial LSTM Networks" (CVPR 2017)

gan pytorch summarization vae video

Last synced: 15 Jan 2025

https://github.com/vinthony/deep-blind-watermark-removal

[AAAI 2021] Split then Refine: Stacked Attention-guided ResUNets for Blind Single Image Visible Watermark Removal

aaai2021 pytorch watermark-removal

Last synced: 15 Jan 2025

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

Merlin Models is a collection of deep learning recommender system model reference implementations

dask deep-learning gpu machine-learning pytorch rapidsai recommendation-system recommender-system recsys tensorflow

Last synced: 15 Jan 2025

https://github.com/tcapelle/timeseries_fastai

fastai V2 implementation of Timeseries classification papers.

classification fastai pytorch timeseries ucr

Last synced: 13 Jan 2025

https://github.com/BiomedSciAI/histocartography

A standardized Python API with necessary preprocessing, machine learning and explainability tools to facilitate graph-analytics in computational pathology.

deep-learning graph-neural-networks healthcare machine-learning pathology pytorch

Last synced: 13 Nov 2024

https://github.com/txsun1997/Black-Box-Tuning

ICML'2022: Black-Box Tuning for Language-Model-as-a-Service & EMNLP'2022: BBTv2: Towards a Gradient-Free Future with Large Language Models

black-box-optimization deep-learning few-shot-learning language-model natural-language-processing pytorch

Last synced: 19 Dec 2024

https://github.com/louis-she/minetorch

Build deep learning applications in a new and easy way.

digits ignite mnist pytorch pytorch-visualization tensorboard tools torchpack

Last synced: 28 Nov 2024

https://github.com/Eclectic-Sheep/sheeprl

Distributed Reinforcement Learning accelerated by Lightning Fabric

distributed lightning pytorch reinforcement-learning

Last synced: 06 Nov 2024

https://github.com/spfrommer/torchexplorer

Interactively inspect module inputs, outputs, parameters, and gradients.

pytorch visualization wandb

Last synced: 28 Oct 2024

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

Unofficial PyTorch Implementation of SUM-GAN from "Unsupervised Video Summarization with Adversarial LSTM Networks" (CVPR 2017)

gan pytorch summarization vae video

Last synced: 07 Nov 2024

https://github.com/deeplearningturkiye/pratik-derin-ogrenme-uygulamalari

Çeşitli kütüphaneler kullanılarak Türkçe kod açıklamalarıyla TEMEL SEVİYEDE pratik derin öğrenme uygulamaları.

cnn keras mnist pytorch tensorflow

Last synced: 13 Jan 2025

https://github.com/cyanogenoid/pytorch-vqa

Strong baseline for visual question answering

baseline pytorch visual-question-answering vqa

Last synced: 15 Jan 2025

https://github.com/ToluClassics/candle-tutorial

Tutorial for Porting PyTorch Transformer Models to Candle (Rust)

candle pytorch rust rust-lang

Last synced: 16 Jan 2025

https://github.com/bshall/universalvocoding

A PyTorch implementation of "Robust Universal Neural Vocoding"

neural-vocoder pytorch speech-synthesis wavernn

Last synced: 15 Jan 2025

https://github.com/spico197/docee

🕹️ A toolkit for document-level event extraction, containing some SOTA model implementations.

event-extraction information-extraction natural-language-understanding pytorch

Last synced: 15 Jan 2025

https://github.com/Mayukhdeb/torch-dreams

Flexible Feature visualization on PyTorch, for research and art :mag_right: :computer: :brain: :art:

colab deep-dream deep-learning feature-visualization pytorch

Last synced: 06 Nov 2024

https://github.com/mayukhdeb/torch-dreams

Flexible Feature visualization on PyTorch, for research and art :mag_right: :computer: :brain: :art:

colab deep-dream deep-learning feature-visualization pytorch

Last synced: 11 Jan 2025

https://github.com/pyro-ppl/funsor

Functional tensors for probabilistic programming

jax machine-learning numpy probabilistic-programming pyro pytorch symbolic

Last synced: 11 Jan 2025

https://github.com/SamsungLabs/pytorch-ensembles

Pitfalls of In-Domain Uncertainty Estimation and Ensembling in Deep Learning, ICLR 2020

deep-learning ensembles iclr2020 pytorch uncertainty

Last synced: 15 Nov 2024

https://github.com/luopeixiang/textclf

TextClf :基于Pytorch/Sklearn的文本分类框架,包括逻辑回归、SVM、TextCNN、TextRNN、TextRCNN、DRNN、DPCNN、Bert等多种模型,通过简单配置即可完成数据处理、模型训练、测试等过程。

bert cnn-text-classification configurable document-classification dpcnn drnn glove logistic-regression lstm-text-classification neuralclassifier pytorch sentiment-analysis sklearn-classify svm textcnn textrnn word2vec

Last synced: 15 Jan 2025

https://github.com/feiyuhuahuo/Yolact_minimal

Minimal PyTorch implementation of YOLACT.

instance-segmentation pytorch real-time

Last synced: 27 Nov 2024

https://github.com/ydwen/opensphere

A hyperspherical face recognition library based on PyTorch

deep-learning face-recognition pytorch sphereface sphereface2

Last synced: 27 Oct 2024