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/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/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: 16 Nov 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/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/showlab/all-in-one

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

codebase pre-training pytorch video-language

Last synced: 17 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: 14 Nov 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: 17 Nov 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/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: 15 Nov 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: 15 Nov 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/szagoruyko/pyinn

CuPy fused PyTorch neural networks ops

cupy pytorch

Last synced: 29 Oct 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/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/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/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/neulab/knn-transformers

PyTorch + HuggingFace code for RetoMaton: "Neuro-Symbolic Language Modeling with Automaton-augmented Retrieval" (ICML 2022), including an implementation of kNN-LM and kNN-MT

huggingface knn knn-lm knn-mt knn-transformers knnlm knnmt language language-models machine models nearest nearest-neighbor neighbor neuro-symbolic pytorch retomaton transformers translation

Last synced: 14 Nov 2024

https://github.com/JeongJiHeon/ScoreDiffusionModel

The Pytorch Tutorial of Score-based and Diffusion Model

diffusion-models pytorch score-based score-based-generative-models

Last synced: 17 Nov 2024

https://github.com/rentainhe/pytorch-distributed-training

Simple tutorials on Pytorch DDP training

apex cuda ddp-training deep-learning pytorch

Last synced: 14 Nov 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://github.com/Sleepychord/CogLTX

The source code of NeurIPS 2020 paper "CogLTX: Applying BERT to Long Texts"

bert pytorch

Last synced: 16 Nov 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://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/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/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: 14 Nov 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/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/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: 14 Nov 2024

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

https://github.com/qinbinli/moon

Model-Contrastive Federated Learning (CVPR 2021)

contrastive-learning cvpr2021 federated-learning pytorch

Last synced: 14 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

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

PyTorch implementation of Sinusodial Representation networks (SIREN)

activation-functions deep-learning machine-learning pytorch

Last synced: 14 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/ianwhale/nsga-net

NSGA-Net, a Neural Architecture Search Algorithm

auto-ml deep-learning evolutionary-computation neural-architecture-search nsga-ii pytorch

Last synced: 16 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: 14 Nov 2024

https://github.com/maum-ai/assem-vc

Official Code for Assem-VC @ICASSP2022

deep-learning pytorch speech-synthesis voice-conversion

Last synced: 11 Nov 2024

https://github.com/asappresearch/flambe

An ML framework to accelerate research and its path to production.

deep-learning distributed machine-learning ml python pytorch research

Last synced: 11 Nov 2024

https://github.com/chingyaoc/pytorch-REINFORCE

PyTorch Implementation of REINFORCE for both discrete & continuous control

continuous-control gym mujoco pytorch reinforce reinforcement-learning

Last synced: 14 Nov 2024

https://github.com/ofnote/tsalib

Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)

deep-learning keras numpy pytorch tensor-shape tensorflow

Last synced: 16 Nov 2024

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

Differentiable computations of the signature and logsignature transforms, on both CPU and GPU. (ICLR 2021)

deep-learning deep-neural-networks logsignature logsignatures machine-learning pytorch rough-paths signature signatures

Last synced: 12 Nov 2024

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

Unofficial PyTorch Implementation of UnivNet Vocoder (https://arxiv.org/abs/2106.07889)

deep-learning gan pytorch speech-synthesis text-to-speech tts vocoder

Last synced: 11 Nov 2024

https://github.com/lonepatient/nezha_chinese_pytorch

NEZHA: Neural Contextualized Representation for Chinese Language Understanding

chinese language-model nezha pytorch

Last synced: 06 Nov 2024

https://github.com/xpixelgroup/ranksrgan

ICCV 2019 (oral) RankSRGAN: Generative Adversarial Networks with Ranker for Image Super-Resolution. PyTorch implementation

gan generative-adversarial-network iccv2019 learning-to-rank low-level-vision pytorch super-resolution

Last synced: 11 Nov 2024

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

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/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/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: 13 Nov 2024

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: 18 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/yatenglg/retinanet-pytorch

Retinanet目标检测算法(简单,明了,易用,全中文注释,单机多卡训练,视频检测)(based on pytorch,Simple, Clear, Mutil GPU)

object-detection pytorch retinanet

Last synced: 18 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/lucidrains/sinkhorn-transformer

Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention

artificial-intelligence attention-mechanism deep-learning pytorch transformers

Last synced: 13 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/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/andreamad8/Universal-Transformer-Pytorch

Implementation of Universal Transformer in Pytorch

pytorch universal-transformer

Last synced: 07 Nov 2024