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/sinAshish/Multi-Scale-Attention

Code for our paper "Multi-scale Guided Attention for Medical Image Segmentation"

attention-mechanism biomedical-image-analysis deep-learning encoder-decoder pytorch

Last synced: 03 Aug 2024

https://github.com/clovaai/rexnet

Official Pytorch implementation of ReXNet (Rank eXpansion Network) with pretrained models

deep-learning image-classification imagenet object-detection pretrained-model pytorch rexnet transfer-learning

Last synced: 07 Aug 2024

https://github.com/salesforce/warp-drive

Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)

cuda deep-learning gpu high-throughput multiagent-reinforcement-learning numba pytorch reinforcement-learning

Last synced: 02 Aug 2024

https://github.com/dfdazac/wassdistance

Approximating Wasserstein distances with PyTorch

deep-learning machine-learning optimal-transport pytorch

Last synced: 02 Aug 2024

https://github.com/ploomber/sklearn-evaluation

Machine learning model evaluation made easy: plots, tables, HTML reports, experiment tracking and Jupyter notebook analysis.

data-science deep-learning jupyter-notebook machine-learning pytorch scikit-learn sklearn tensorflow

Last synced: 24 Sep 2024

https://github.com/walktree/libtorch-yolov3

A Libtorch implementation of the YOLO v3 object detection algorithm

cpp libtorch pytorch yolov3

Last synced: 02 Aug 2024

https://github.com/layumi/AICIty-reID-2020

:red_car: The 1st Place Submission to AICity Challenge 2020 re-id track (Baidu-UTS submission)

aicity cityflow cvpr2020 paddlepaddle pytorch vehicle vehicle-reid veri-776

Last synced: 01 Aug 2024

https://github.com/huggingface/large_language_model_training_playbook

An open collection of implementation tips, tricks and resources for training large language models

cuda large-language-models llm nccl nlp performance python pytorch scalability troubleshooting

Last synced: 02 Aug 2024

https://github.com/loretoparisi/CapsNet

CapsNet (Capsules Net) in Geoffrey E Hinton paper "Dynamic Routing Between Capsules" - State Of the Art

capsnet capsule-networks chainer deeplearning keras lasagne machine-learning mnist mxnet pytorch tensorflow

Last synced: 01 Aug 2024

https://github.com/sacmehta/ESPNetv2

A light-weight, power efficient, and general purpose convolutional neural network

cnn convolutional-neural-networks deep-learning efficient imagenet lightweight machine-learning pytorch semantic-segmentation

Last synced: 31 Jul 2024

https://github.com/chaitjo/geometric-gnn-dojo

Geometric GNN Dojo provides unified implementations and experiments to explore the design space of Geometric Graph Neural Networks.

e3nn equivariance equivariant-networks geometric-deep-learning geometric-graphs graph-isomorphism graph-neural-networks pytorch pytorch-geometric

Last synced: 02 Aug 2024

https://github.com/iBelieveCJM/Tricks-of-Semi-supervisedDeepLeanring-Pytorch

PseudoLabel 2013, VAT, PI model, Tempens, MeanTeacher, ICT, MixMatch, FixMatch

fixmatch ict meanteacher mixmatch pi pseudo-label pytorch semi-supervised-learning ssl tempens vat

Last synced: 03 Aug 2024

https://github.com/THUDM/GraphMAE

GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22

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

Last synced: 08 Aug 2024

https://github.com/nschaetti/EchoTorch

A Python toolkit for Reservoir Computing and Echo State Network experimentation based on pyTorch. EchoTorch is the only Python module available to easily create Deep Reservoir Computing models.

artificial-intelligence artificial-neural-networks echo-state-networks machine-learning machine-learning-algorithms machinelearning neural-networks python python-toolkit pytorch recurrent-networks recurrent-neural-networks reservoir-computing torch

Last synced: 03 Aug 2024

https://github.com/pykale/pykale

Knowledge-Aware machine LEarning (KALE): accessible machine learning from multiple sources for interdisciplinary research, part of the 🔥PyTorch ecosystem. ⭐ Star to support our work!

computer-vision data-science deep-learning domain-adaptation graph-analysis knowledge-aware-learning machine-learning medical-image-analysis meta-learning multimodal multimodal-learning python pytorch transfer-learning

Last synced: 01 Aug 2024

https://github.com/tlkh/ai-lab

All-in-one AI container for rapid prototyping

cuda data-science deep-learning docker jupyter nvidia pytorch tensorflow

Last synced: 02 Aug 2024

https://github.com/ruiminshen/yolo2-pytorch

PyTorch implementation of the YOLO (You Only Look Once) v2

caffe2 computer-vision deep-learning deep-neural-networks object-detection onnx python pytorch yolo2

Last synced: 01 Aug 2024

https://github.com/likyoo/open-cd

A Change Detection Repo Standing on the Shoulders of Giants

bit change-detection changer deep-learning pytorch transformer

Last synced: 03 Aug 2024

https://github.com/cuiziteng/Illumination-Adaptive-Transformer

[BMVC 2022] You Only Need 90K Parameters to Adapt Light: A Light Weight Transformer for Image Enhancement and Exposure Correction. SOTA for low light enhancement, 0.004 seconds try this for pre-processing.

bmvc exposure-correction image-enhancement image-reconstruction image-restoration low-level-vision low-light-enhance low-light-image-enhancement pytorch transformer-architecture transformer-models

Last synced: 01 Aug 2024

https://github.com/M-Nauta/TCDF

Temporal Causal Discovery Framework (PyTorch): discovering causal relationships between time series

causality cnn forecasting machine-learning neural-networks prediction python pytorch timeseries

Last synced: 03 Aug 2024

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

[NeurIPS 2020] MCUNet: Tiny Deep Learning on IoT Devices; [NeurIPS 2021] MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning

deep-learning microncontroller neural-architecture-search pytorch tinyml

Last synced: 31 Jul 2024

https://github.com/k2-fsa/sherpa

Speech-to-text server framework with next-gen Kaldi

asr cpp ctc end-to-end-asr python pytorch speech-recognition transducer websocket

Last synced: 04 Aug 2024

https://github.com/skhadem/3D-BoundingBox

PyTorch implementation for 3D Bounding Box Estimation Using Deep Learning and Geometry

3d computer-vision localization pytorch

Last synced: 31 Jul 2024

https://github.com/RaivoKoot/Video-Dataset-Loading-Pytorch

Generic PyTorch dataset implementation to load and augment VIDEOS for deep learning training loops.

action-recognition dataloader deep-learning machine-learning pytorch video-dataset videos

Last synced: 02 Aug 2024

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

Open-source evaluation toolkit of large vision-language models (LVLMs), support GPT-4v, Gemini, QwenVLPlus, 40+ HF models, 20+ 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: 02 Aug 2024

https://github.com/zihangJiang/TokenLabeling

Pytorch implementation of "All Tokens Matter: Token Labeling for Training Better Vision Transformers"

imagenet lv-vit pytorch segmentation transformer vision

Last synced: 02 Aug 2024

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

Open-source evaluation toolkit of large vision-language models (LVLMs), support GPT-4v, Gemini, QwenVLPlus, 40+ HF models, 20+ 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: 08 Aug 2024

https://github.com/Eromera/erfnet_pytorch

Pytorch code for semantic segmentation using ERFNet

cityscapes erfnet pytorch segmentation semantic semantic-segmentation

Last synced: 03 Aug 2024

https://github.com/abhiskk/fast-neural-style

pytorch implementation of fast-neural-style

deep-learning neural-style pytorch

Last synced: 01 Aug 2024

https://github.com/shm007g/LLaMA-Cult-and-More

Large Language Models for All, 🦙 Cult and More, Stay in touch !

alpaca chatgpt deepspeed ggml gpt gpt4 gptq llama llm loralib pytorch tensorflow transformers vicuna

Last synced: 31 Jul 2024

https://github.com/mv-lab/InstructIR

[ECCV 2024] InstructIR: High-Quality Image Restoration Following Human Instructions https://huggingface.co/spaces/marcosv/InstructIR

computer-vision deblurring deep-learning dehazing denoising image-enhancement image-restoration inverse-problems language-model low-light-image-enhancement multi-task multimodal neural-network photography prompt pytorch super-resolution

Last synced: 01 Aug 2024

https://github.com/Stanford-TML/EDGE

Official PyTorch Implementation of EDGE (CVPR 2023)

animation dance-generation diffusion-models pytorch

Last synced: 03 Aug 2024

https://github.com/lemonhu/NER-BERT-pytorch

PyTorch solution of named entity recognition task Using Google AI's pre-trained BERT model.

chinese-ner entity-extraction google-bert information-extraction msra named-entity-recognition ner pytorch transformer

Last synced: 01 Aug 2024

https://github.com/Jeff-sjtu/res-loglikelihood-regression

Code for "Human Pose Regression with Residual Log-likelihood Estimation", ICCV 2021 Oral

2d-human-pose 3d-human-pose human-pose-estimation iccv iccv2021 pytorch regression

Last synced: 03 Aug 2024

https://github.com/eliahuhorwitz/DeepSIM

Official PyTorch implementation of the paper: "DeepSIM: Image Shape Manipulation from a Single Augmented Training Sample" (ICCV 2021 Oral)

computer-graphics computer-vision deep-learning deep-neural-networks edge-to-image generative-adversarial-network iccv2021 image-editing image-to-image-translation pytorch segmantation-to-image single-image

Last synced: 07 Aug 2024

https://github.com/KinglittleQ/torch-batch-svd

A 100x faster SVD for PyTorch⚡️

cusolver pytorch svd

Last synced: 31 Jul 2024

https://github.com/NasirKhalid24/CLIP-Mesh

Official implementation of CLIP-Mesh: Generating textured meshes from text using pretrained image-text models

3d pytorch

Last synced: 01 Aug 2024

https://naver.github.io/roma/

RoMa: A lightweight library to deal with 3D rotations in PyTorch.

3d geometry pytorch rotation

Last synced: 03 Aug 2024

https://github.com/petuum/adaptdl

Resource-adaptive cluster scheduler for deep learning training.

aws cloud deep-learning distributed-systems distributed-training kubernetes machine-learning pytorch

Last synced: 03 Aug 2024

https://github.com/adefossez/julius

Fast PyTorch based DSP for audio and 1D signals

dsp pytorch

Last synced: 01 Aug 2024

https://github.com/kubeflow/mpi-operator

Kubernetes Operator for MPI-based applications (distributed training, HPC, etc.)

apache-mxnet distributed-computing horovod kubeflow kubernetes mpi pytorch tensorflow

Last synced: 31 Jul 2024

https://github.com/krasserm/perceiver-io

A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training

deep-learning machine-learning perceiver perceiver-ar perceiver-io pytorch pytorch-lightning

Last synced: 09 Aug 2024

https://github.com/Lotayou/densebody_pytorch

PyTorch implementation of CloudWalk's recent work DenseBody

3d-reconstruction human pose-estimation pytorch

Last synced: 01 Aug 2024

https://github.com/interpretml/interpret-community

Interpret Community extends Interpret repository with additional interpretability techniques and utility functions to handle real-world datasets and workflows.

deep-learning explainer explanationdashboard interpretable-models machine-learning mimic-explainer python pytorch tabular-explainer

Last synced: 03 Aug 2024

https://github.com/cccntu/minlora

minLoRA: a minimal PyTorch library that allows you to apply LoRA to any PyTorch model.

fastai huggingface pytorch pytorch-implementation pytorch-lightning

Last synced: 02 Aug 2024

https://github.com/clovaai/AdamP

AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights (ICLR 2021)

deep-learning iclr2021 machine-learning optimizer optimizer-algorithms pytorch

Last synced: 03 Aug 2024

https://github.com/apachecn/fastai-ml-dl-notes-zh

:book: [译] fast.ai 机器学习和深度学习中文笔记

deep-learning fastai lecture-notes machine-learning python pytorch

Last synced: 31 Jul 2024

https://github.com/Media-Smart/vedaseg

A semantic segmentation toolbox based on PyTorch

deeplabv3 deeplabv3plus fpn pspnet pytorch semantic-segmentation unet

Last synced: 03 Aug 2024

https://github.com/hysts/anime-face-detector

Anime Face Detector using mmdet and mmpose

anime computer-vision face-detection face-landmark-detection pytorch

Last synced: 25 Sep 2024

https://github.com/patrick-kidger/torchcde

Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.

controlled-differential-equations deep-learning deep-neural-networks differential-equations dynamical-systems machine-learning neural-differential-equations neural-networks pytorch time-series

Last synced: 02 Aug 2024

https://github.com/gengshan-y/high-res-stereo

Hierarchical Deep Stereo Matching on High Resolution Images, CVPR 2019.

cvpr2019 pytorch stereo-matching stereo-vision

Last synced: 04 Aug 2024

https://github.com/SforAiDl/genrl

A PyTorch reinforcement learning library for generalizable and reproducible algorithm implementations with an aim to improve accessibility in RL

algorithm-implementations benchmarking data-science deep-learning gym hacktoberfest machine-learning neural-network openai python pytorch reinforcement-learning reinforcement-learning-algorithms

Last synced: 02 Aug 2024

https://github.com/Continvvm/continuum

A clean and simple data loading library for Continual Learning

continual-learning dataloader dataset incremental-learning lifelong-learning online-learning pytorch

Last synced: 03 Aug 2024

https://github.com/sforaidl/genrl

A PyTorch reinforcement learning library for generalizable and reproducible algorithm implementations with an aim to improve accessibility in RL

algorithm-implementations benchmarking data-science deep-learning gym hacktoberfest machine-learning neural-network openai python pytorch reinforcement-learning reinforcement-learning-algorithms

Last synced: 25 Sep 2024

https://github.com/ethanhe42/epipolar-transformers

Epipolar Transformers (best paper award, CVPR 2020 workshop)

3d 3dposeestimation deep-learning pose-estimation pytorch

Last synced: 01 Aug 2024

https://github.com/fadel/pytorch_ema

Tiny PyTorch library for maintaining a moving average of a collection of parameters.

deep-learning neural-networks pytorch

Last synced: 24 Sep 2024

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

Train high-quality text-to-image diffusion models in a data & compute efficient manner

deep-learning diffusion-models large-scale-vision-models machine-learning pytorch

Last synced: 25 Sep 2024

https://github.com/wuba/dl_inference

通用深度学习推理工具,可在生产环境中快速上线由TensorFlow、PyTorch、Caffe框架训练出的深度学习模型。

ai caffe deep-learing inference machine-learning pytorch tensorflow

Last synced: 03 Aug 2024

https://github.com/MuQiuJun-AI/bert4pytorch

超轻量级bert的pytorch版本,大量中文注释,容易修改结构,持续更新

bert nlp pytorch transformer

Last synced: 01 Aug 2024

https://github.com/xiadingZ/video-caption.pytorch

pytorch implementation of video captioning

deep-learning pytorch video-captioning

Last synced: 03 Aug 2024

https://github.com/wpf535236337/real-time-network

real-time network architecture for mobile devices and semantic segmentation

efficient-networks lightweightnetwoks mobile-networks neural-architecture-search pytorch semantic-segmentation

Last synced: 31 Jul 2024

https://github.com/JusperLee/Dual-Path-RNN-Pytorch

Dual-path RNN: efficient long sequence modeling for time-domain single-channel speech separation implemented by Pytorch

deep-learning pytorch rnn-model speech-separation speech-separation-algorithm

Last synced: 01 Aug 2024

https://github.com/aelnouby/Text-to-Image-Synthesis

Pytorch implementation of Generative Adversarial Text-to-Image Synthesis paper

gans image-generation pytorch text-to-image zero-shot-learning

Last synced: 03 Aug 2024

https://github.com/cccntu/minLoRA

minLoRA: a minimal PyTorch library that allows you to apply LoRA to any PyTorch model.

fastai huggingface pytorch pytorch-implementation pytorch-lightning

Last synced: 30 Jul 2024

https://github.com/airlab-unibas/airlab

Image registration laboratory for 2D and 3D image data

autograd image-alignment image-registration pytorch rapid-prototyping

Last synced: 31 Jul 2024

https://github.com/achaiah/pywick

High-level batteries-included neural network training library for Pytorch

classification deep-learning neural-networks pytorch semantic-segmentation

Last synced: 03 Aug 2024

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/yanx27/2DPASS

2DPASS: 2D Priors Assisted Semantic Segmentation on LiDAR Point Clouds (ECCV 2022) :fire:

deep-learning knowledge-distillation lidar-point-cloud nuscenes pytorch semantic-segmentation semantickitti

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