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/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: 08 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: 03 Aug 2024

https://github.com/gmongaras/Diffusion_models_from_scratch

Creating a diffusion model from scratch in PyTorch to learn exactly how they work.

classifier-free-guidance ddim ddpm diffusion-models imagenet pytorch

Last synced: 30 Oct 2024

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

Stockfish NNUE (Chess evaluation) trainer in Pytorch

chess deep-learning pytorch stockfish

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

https://github.com/lucastabelini/PolyLaneNet

Code for the paper entitled "PolyLaneNet: Lane Estimation via Deep Polynomial Regression" (ICPR 2020)

deep-learning lane-detection pytorch

Last synced: 28 Oct 2024

https://github.com/tomrunia/pytorchwavelets

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

filtering pytorch signal-processing wavelets

Last synced: 13 Nov 2024

https://github.com/Bluefog-Lib/bluefog

Distributed and decentralized training framework for PyTorch over graph

asynchronous decentralized deeplearning distributed-computing machine-learning mpi nccl one-sided pytorch

Last synced: 07 Aug 2024

https://github.com/feedly/transfer-nlp

NLP library designed for reproducible experimentation management

framework language-model natural-language-understanding nlp playground pytorch transfer-learning

Last synced: 13 Oct 2024

https://github.com/leaderj1001/MobileNetV3-Pytorch

Implementing Searching for MobileNetV3 paper using Pytorch

cifar-10 cifar-100 imagenet mobilenetv3 pytorch searching-for-mobilenetv3

Last synced: 31 Oct 2024

https://github.com/jaxony/ShuffleNet

ShuffleNet in PyTorch. Based on https://arxiv.org/abs/1707.01083

artificial-intelligence convolution deep-learning neural-network pytorch

Last synced: 31 Oct 2024

https://github.com/luodian/generalizable-mixture-of-experts

GMoE could be the next backbone model for many kinds of generalization task.

deep-learning domain-generalization mixture-of-experts pytorch pytorch-implementation

Last synced: 26 Oct 2024

https://github.com/gentaiscool/end2end-asr-pytorch

End-to-End Automatic Speech Recognition on PyTorch

asr end-to-end pytorch speech speech-recognition transformer

Last synced: 08 Nov 2024

https://github.com/wonjongg/StyleCariGAN

[SIGGRAPH 2021] Official implementation of "StyleCariGAN: Caricature Generation via StyleGAN Feature Map Modulation"

deep-learning pytorch

Last synced: 07 Aug 2024

https://github.com/cadene/tensorflow-model-zoo.torch

InceptionV3, InceptionV4, Inception-Resnet pretrained models for Torch7 and PyTorch

inception pretrained-models pytorch tensorflow torch

Last synced: 07 Nov 2024

https://github.com/Luodian/Generalizable-Mixture-of-Experts

GMoE could be the next backbone model for many kinds of generalization task.

deep-learning domain-generalization mixture-of-experts pytorch pytorch-implementation

Last synced: 03 Aug 2024

https://github.com/ksw0306/ClariNet

A Pytorch Implementation of ClariNet

clarinet parallel-wavenet pytorch wavenet

Last synced: 04 Nov 2024

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

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

d2l-ai deep-learning pytorch

Last synced: 09 Nov 2024

https://github.com/suyukun666/UFO

Official PyTorch implementation of the “A Unified Transformer Framework for Co-Segmentation, Co-Saliency Detection and Video Salient Object Detection”. (TMM2023)

co-saliency-detection co-segmentation deep-learning inpainting matching pytorch video video-salient-object-detection

Last synced: 11 Nov 2024

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

volksdep is an open-source toolbox for deploying and accelerating PyTorch, ONNX and TensorFlow models with TensorRT.

acceleration deploy inference jetson-nano jetson-tx2 jetson-xavier keras onnx python pytorch tensorflow tensorrt

Last synced: 03 Aug 2024

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

Fully featured implementation of Routing Transformer

artificial-intelligence attention-mechanism deep-learning pytorch transformer

Last synced: 13 Nov 2024

https://github.com/dgkim5360/the-elements-of-statistical-learning-notebooks

Jupyter notebooks for summarizing and reproducing the textbook "The Elements of Statistical Learning" 2/E by Hastie, Tibshirani, and Friedman

jupyter-notebooks machine-learning matplotlib neural-network python3 pytorch scipy statistical-learning textbook

Last synced: 07 Nov 2024

https://github.com/loeweX/Greedy_InfoMax

Code for the paper: Putting An End to End-to-End: Gradient-Isolated Learning of Representations

deep-learning local-learning pytorch

Last synced: 12 Nov 2024

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

https://github.com/outcastofmusic/quick-nlp

Pytorch NLP library based on FastAI

fastai nlp-library pytorch seq2seq

Last synced: 03 Aug 2024

https://github.com/dena/handyrl

HandyRL is a handy and simple framework based on Python and PyTorch for distributed reinforcement learning that is applicable to your own environments.

deep-learning distributed-training games machine-learning policy-gradient pytorch reinforcement-learning

Last synced: 10 Nov 2024

https://github.com/merantix-momentum/squirrel-core

A Python library that enables ML teams to share, load, and transform data in a collaborative, flexible, and efficient way :chestnut:

ai cloud-computing collaboration computer-vision cv data-ingestion data-mesh data-science dataops datasets deep-learning distributed internal machine-learning ml natural-language-processing nlp python pytorch tensorflow

Last synced: 02 Nov 2024

https://github.com/DeNA/HandyRL

HandyRL is a handy and simple framework based on Python and PyTorch for distributed reinforcement learning that is applicable to your own environments.

deep-learning distributed-training games machine-learning policy-gradient pytorch reinforcement-learning

Last synced: 04 Nov 2024

https://github.com/sunzeyeah/RLHF

Implementation of Chinese ChatGPT

chatgpt deep-learning deepspeed glm nlp pangu pytorch

Last synced: 31 Oct 2024

https://github.com/aimagelab/show-control-and-tell

Show, Control and Tell: A Framework for Generating Controllable and Grounded Captions. CVPR 2019

caption-generation captioning-images cvpr2019 image-captioning pytorch visual-semantic

Last synced: 07 Nov 2024

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

https://github.com/extreme-bert/extreme-bert

ExtremeBERT is a toolkit that accelerates the pretraining of customized language models on customized datasets, described in the paper “ExtremeBERT: A Toolkit for Accelerating Pretraining of Customized BERT”.

bert deep-learning language-model language-models machine-learning natural-language-processing nlp python pytorch transformer

Last synced: 03 Aug 2024

https://github.com/fregu856/3DOD_thesis

3D Object Detection for Autonomous Driving in PyTorch, trained on the KITTI dataset.

3d-deep-learning 3d-detection autonomous-driving computer-vision deep-learning machine-learning object-detection pytorch

Last synced: 07 Aug 2024

https://github.com/shoufachen/cyclemlp

[ICLR'22 Oral] Implementation of "CycleMLP: A MLP-like Architecture for Dense Prediction"

backbone-models mlp pytorch

Last synced: 26 Oct 2024

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

NU-Wave: A Diffusion Probabilistic Model for Neural Audio Upsampling @ INTERSPEECH 2021

deep-generative-model deep-learning neural-audio-upsampling pytorch super-resolution upsampling

Last synced: 11 Nov 2024

https://github.com/DocF/multispectral-object-detection

Multispectral Object Detection with Yolov5 and Transformer

deep-learning object-detection pytorch transformer yolov5

Last synced: 09 Nov 2024

https://github.com/jacobgil/pytorch-tensor-decompositions

PyTorch implementation of [1412.6553] and [1511.06530] tensor decomposition methods for convolutional layers.

deep-learning pytorch tensorly

Last synced: 31 Oct 2024

https://github.com/usnistgov/jarvis

JARVIS-Tools: an open-source software package for data-driven atomistic materials design. Publications: https://scholar.google.com/citations?user=3w6ej94AAAAJ

ai atomistic-simulations dft lammps materials-informatics md mgi pytorch quantum-computing scikit-learn tensorflow vasp

Last synced: 12 Oct 2024

https://github.com/xyupeng/ContrastiveCrop

[CVPR 2022 Oral] Crafting Better Contrastive Views for Siamese Representation Learning

contrastive-learning pytorch self-supervised-learning

Last synced: 03 Aug 2024

https://github.com/chandlerbang/pro-gnn

Implementation of the KDD 2020 paper "Graph Structure Learning for Robust Graph Neural Networks"

adversarial-attacks attack-defense deep-learning defense graph-mining graph-neural-networks graph-structure-recovery machine-learning pytorch semi-supervised-learning

Last synced: 06 Nov 2024

https://github.com/kkanshul/finegan

FineGAN: Unsupervised Hierarchical Disentanglement for Fine-grained Object Generation and Discovery

computer-vision deep-learning disentangled-representations fine-grained gans generative-adversarial-network image-generation image-manipulation pytorch

Last synced: 03 Aug 2024

https://github.com/jingweiz/pytorch-dnc

Neural Turing Machine (NTM) & Differentiable Neural Computer (DNC) with pytorch & visdom

deep-learning dnc external-memory ntm pytorch visdom

Last synced: 04 Aug 2024

https://github.com/ChandlerBang/Pro-GNN

Implementation of the KDD 2020 paper "Graph Structure Learning for Robust Graph Neural Networks"

adversarial-attacks attack-defense deep-learning defense graph-mining graph-neural-networks graph-structure-recovery machine-learning pytorch semi-supervised-learning

Last synced: 08 Aug 2024

https://github.com/Lin-Yijie/Graph-Matching-Networks

PyTorch implementation of Graph Matching Networks, e.g., Graph Matching with Bi-level Noisy Correspondence (COMMON, ICCV 2023), Graph Matching Networks for Learning the Similarity of Graph Structured Objects (GMN, ICML 2019).

deep-learning gnn graph-neural-networks pytorch

Last synced: 04 Nov 2024

https://github.com/iArunava/ENet-Real-Time-Semantic-Segmentation

ENet - A Neural Net Architecture for real time Semantic Segmentation

artificial-intelligence deep-learning enet paper python3 pytorch

Last synced: 07 Aug 2024

https://github.com/prodramp/DeepWorks

A collection of Deep Learning projects and resources

artificial-intelligence deep-learning generative-models llm machine-learning pytorch tensorflow

Last synced: 29 Oct 2024

https://github.com/szagoruyko/pyinn

CuPy fused PyTorch neural networks ops

cupy pytorch

Last synced: 29 Oct 2024

https://github.com/dyhan0920/PyramidNet-PyTorch

A PyTorch implementation for PyramidNets (Deep Pyramidal Residual Networks, https://arxiv.org/abs/1610.02915)

cifar-10 cifar-100 imagenet pyramidnet pytorch residual-networks resnet

Last synced: 31 Oct 2024

https://github.com/krishnap25/mauve

Package to compute Mauve, a similarity score between neural text and human text. Install with `pip install mauve-text`.

deep-learning huggingface-transformers nlp pytorch text-generation

Last synced: 13 Oct 2024

https://github.com/showlab/all-in-one

[CVPR2023] All in One: Exploring Unified Video-Language Pre-training

codebase pre-training pytorch video-language

Last synced: 09 Nov 2024

https://github.com/WenjieDu/SAITS

The official PyTorch implementation of the paper "SAITS: Self-Attention-based Imputation for Time Series". A fast and state-of-the-art (SOTA) deep-learning neural network model for efficient time-series imputation (impute multivariate incomplete time series containing NaN missing data/values with machine learning). https://arxiv.org/abs/2202.08516

attention attention-mechanism deep-learning imputation imputation-model impute incomplete-data incomplete-time-series interpolation irregular-sampling machine-learning missing-values partially-observed partially-observed-data partially-observed-time-series pytorch self-attention time-series time-series-imputation transformer

Last synced: 02 Nov 2024

https://github.com/AnjieCheng/MnasNet-PyTorch

A PyTorch implementation of Mnasnet: MnasNet: Platform-Aware Neural Architecture Search for Mobile.

architecture-search mnasnet mobilenetv2 pytorch

Last synced: 31 Oct 2024

https://github.com/SamsungLabs/imvoxelnet

[WACV2022] ImVoxelNet: Image to Voxels Projection for Monocular and Multi-View General-Purpose 3D Object Detection

3d-object-detection imvoxelnet kitti mmdetection nuscenes object-detection pytorch scannet sun-rgbd

Last synced: 28 Oct 2024

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

https://github.com/swanhubx/swanlab

⚡️SwanLab: your ML experiment notebook. 你的AI实验笔记本,跟踪与可视化你的机器学习全流程

data-science deep-learning fastapi jax machine-learning mlops model-versioning python pytorch tensorboard tensorflow tracking transformers visualization

Last synced: 11 Oct 2024

https://github.com/nateraw/huggingpics

🤗🖼️ HuggingPics: Fine-tune Vision Transformers for anything using images found on the web.

huggingface machine-learning pytorch vision-transformers

Last synced: 03 Aug 2024

https://github.com/DerrickXuNu/v2x-vit

[ECCV2022] Official Implementation of paper "V2X-ViT: Vehicle-to-Everything Cooperative Perception with Vision Transformer"

3d-object-detection autonomous-driving collaborative-perception computer-vision deep-learning machine-learning multi-agent-system pytorch simulation v2x vehicle-to-everything vision-transformer

Last synced: 28 Oct 2024

https://github.com/Randl/MobileNetV2-pytorch

Impementation of MobileNetV2 in pytorch

cnn image-classification mobilenet2 pytorch

Last synced: 31 Oct 2024

https://github.com/IDEA-Research/HumanSD

[ICCV 2023] The official implementation of paper "HumanSD: A Native Skeleton-Guided Diffusion Model for Human Image Generation"

conditional-image-generation deep-learning iccv iccv2023 image-generation pytorch

Last synced: 31 Oct 2024

https://github.com/numenta/nupic.torch

Numenta Platform for Intelligent Computing PyTorch libraries

htm numenta-platform pytorch

Last synced: 06 Nov 2024

https://idea-research.github.io/HumanSD/

[ICCV 2023] The official implementation of paper "HumanSD: A Native Skeleton-Guided Diffusion Model for Human Image Generation"

conditional-image-generation deep-learning iccv iccv2023 image-generation pytorch

Last synced: 31 Oct 2024

https://github.com/durandtibo/wildcat.pytorch

PyTorch implementation of "WILDCAT: Weakly Supervised Learning of Deep ConvNets for Image Classification, Pointwise Localization and Segmentation", CVPR 2017

convnet cvpr-2017 cvpr17 cvpr2017 deep-learning image-classification pytorch weakly-supervised-learning

Last synced: 05 Nov 2024

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: 08 Aug 2024

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

NU-Wave 2: A General Neural Audio Upsampling Model for Various Sampling Rates @ INTERSPEECH 2022

deep-learning neural-audio-upsampling pytorch super-resolution upsampling

Last synced: 11 Nov 2024

https://github.com/ducha-aiki/affnet

Code and weights for local feature affine shape estimation paper "Repeatability Is Not Enough: Learning Discriminative Affine Regions via Discriminability"

affine-shape-estimator computer-vision convolutional-networks convolutional-neural-networks deep-learning hessian image-matching image-retrieval local-features pytorch

Last synced: 31 Oct 2024

https://github.com/layumi/person-reid-3d

TNNLS'22 :statue_of_liberty: Parameter-Efficient Person Re-identification in the 3D Space :statue_of_liberty:

3d dukemtmc-reid graph-neural-networks market-1501 msmt17 ognet person-reid person-reidentification pytorch re-id re-identification reid

Last synced: 31 Oct 2024

https://github.com/ysymyth/3d-sdn

[NeurIPS 2018] 3D-Aware Scene Manipulation via Inverse Graphics

3d-sdn 3d-vision deep-learning disentangled-representations gans generative-adversarial-networks pytorch

Last synced: 11 Nov 2024

https://github.com/kennethnym/infinifi

infinifi plays gentle lofi music in the background indefinitely

artificial-intelligence audiocraft lo-fi lo-fi-radio machine-learning musicgen pytorch

Last synced: 29 Oct 2024

https://github.com/kwea123/casmvsnet_pl

Cascade Cost Volume for High-Resolution Multi-View Stereo and Stereo Matching using pytorch-lightning

3d-reconstruction blendedmvs cascade-cost-volume depth-prediction dtu-dataset multi-view-stereo mvsnet pytorch pytorch-lightning tanks-and-temples

Last synced: 30 Oct 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: 07 Nov 2024

https://github.com/compvis/metric-learning-divide-and-conquer

Source code for the paper "Divide and Conquer the Embedding Space for Metric Learning", CVPR 2019

few-shot-learning metric-learning pytorch

Last synced: 05 Nov 2024