PyTorch | Ecosyste.ms: Awesome

https://github.com/lucasjinreal/DCNv2_latest

DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)

Last synced: 27 Oct 2024

https://github.com/Ha0Tang/AttentionGAN

AttentionGAN for Unpaired Image-to-Image Translation & Multi-Domain Image-to-Image Translation

adversarial-networks attention-model computer-vision cyclegan deep-learning gans geometry ijcnn image-generation image-to-image-translation image-translation multi-domain pytorch unpaired

Last synced: 03 Aug 2024

https://github.com/chrischoy/FCGF

Fully Convolutional Geometric Features: Fast and accurate 3D features for registration and correspondence.

3d correspondence feature-extraction fully-convolutional minkowskiengine neural-network pytorch registration sparse-tensor

Last synced: 02 Nov 2024

https://github.com/mir-group/nequip

NequIP is a code for building E(3)-equivariant interatomic potentials

atomistic-simulations computational-chemistry deep-learning drug-discovery force-fields interatomic-potentials machine-learning materials-science molecular-dynamics pytorch

Last synced: 13 Nov 2024

https://github.com/ymcui/MacBERT

Revisiting Pre-trained Models for Chinese Natural Language Processing (MacBERT)

bert language-model macbert nlp pytorch tensorflow transformers

Last synced: 03 Aug 2024

https://github.com/zkywsg/daily-deeplearning

🔥机器学习/深度学习/Python/算法面试/自然语言处理教程/剑指offer/machine learning/deeplearning/Python/Algorithm interview/NLP Tutorial

deep-learning leetcode leetcode-python leetcode-solutions machine-learning nlp python pytorch pytorch-nlp pytorch-tutorial pytorch-tutorials tensorflow tensorflow-examples tensorflow-tutorials

Last synced: 10 Oct 2024

https://github.com/cdqa-suite/cdQA

⛔ [NOT MAINTAINED] An End-To-End Closed Domain Question Answering System.

artificial-intelligence bert deep-learning information-retrieval natural-language-processing nlp pytorch question-answering reading-comprehension transformers

Last synced: 13 Nov 2024

https://github.com/araffin/robotics-rl-srl

S-RL Toolbox: Reinforcement Learning (RL) and State Representation Learning (SRL) for Robotics

arm baselines baxter-robot gym kuka pybullet pytorch reinforcement-learning representation-learning robotics state

Last synced: 13 Nov 2024

https://github.com/Lightning-Universe/lightning-transformers

Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning

hydra pytorch pytorch-lightning transformers

Last synced: 03 Nov 2024

https://github.com/augmentedstartups/AS-One

Easy & Modular Computer Vision Detectors, Trackers & SAM - Run YOLOv9,v8,v7,v6,v5,R,X in under 10 lines of code.

computer-vision deep-learning object-detection opencv pytorch sam tracking ultralytics yolor yolov5 yolov7 yolov8 yolov9 yolox

Last synced: 09 Nov 2024

https://github.com/alankbi/detecto

Build fully-functioning computer vision models with PyTorch

computer-vision faster-rcnn machine-learning object-detection python pytorch

Last synced: 25 Oct 2024

https://github.com/yeyupiaoling/masr

Pytorch实现的流式与非流式的自动语音识别框架，同时兼容在线和离线识别，目前支持Conformer、Squeezeformer、DeepSpeech2模型，支持多种数据增强方法。

asr conformer deep-learning deepspeech pytorch speech speech-recognition speech-to-text squeezeformer

Last synced: 22 Oct 2024

https://github.com/OndrejTexler/Few-Shot-Patch-Based-Training

The official implementation of our SIGGRAPH 2020 paper Interactive Video Stylization Using Few-Shot Patch-Based Training

deep-learning gan generative-adversarial-network image-generation image-to-image-translation pytorch style-transfer

Last synced: 03 Aug 2024

https://github.com/alvinwan/neural-backed-decision-trees

Making decision trees competitive with neural networks on CIFAR10, CIFAR100, TinyImagenet200, Imagenet

cifar10 cifar100 decision-trees explainability image-classification imagenet interpretability neural-backed-decision-trees neural-networks pretrained-models pretrained-weights pytorch tiny-imagenet

Last synced: 01 Nov 2024

https://github.com/VincentStimper/normalizing-flows

PyTorch implementation of normalizing flow models

density-estimation glow invertible-neural-networks neural-spline-flow normalizing-flow pytorch real-nvp residual-flow variational-autoencoder variational-inference

Last synced: 03 Aug 2024

https://github.com/BlackSamorez/tensor_parallel

Automatically split your PyTorch models on multiple GPUs for training & inference

deep-learning machine-learning natural-language-processing nlp python pytorch pytorch-transformers

Last synced: 09 Aug 2024

https://github.com/yanx27/EverybodyDanceNow_reproduce_pytorch

Everybody dance now reproduced in pytorch

everybody-dance-now openpose pix2pixhd pose2pose pytorch

Last synced: 04 Nov 2024

https://github.com/abeardear/pytorch-YOLO-v1

an experiment for yolo-v1, including training and testing.

experiment object-detection pytorch yolov1

Last synced: 09 Nov 2024

https://github.com/nianticlabs/manydepth

[CVPR 2021] Self-supervised depth estimation from short sequences

cityscapes cost-volumes cvpr cvpr2021 depth-estimation depths estimating-depths kitti monodepth pytorch self-supervised self-supervised-learning

Last synced: 05 Aug 2024

https://github.com/sforaidl/kd_lib

A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quantization.

algorithm-implementations benchmarking data-science deep-learning-library knowledge-distillation machine-learning model-compression pruning pytorch quantization

Last synced: 29 Oct 2024

https://github.com/kwotsin/mimicry

[CVPR 2020 Workshop] A PyTorch GAN library that reproduces research results for popular GANs.

cgan cvpr cvpr20 cvpr2020 dcgan gan gans generative-adversarial-networks generative-models infomax-gan machine-learning python pytorch reproducibility sngan sngan-projection ssgan wgan-gp

Last synced: 13 Nov 2024

https://github.com/castorini/daam

Diffusion attentive attribution maps for interpreting Stable Diffusion.

diffusion explainable-ai generative-ai huggingface pytorch stable-diffusion

Last synced: 03 Aug 2024

https://github.com/kevinzakka/pytorch-goodies

PyTorch Boilerplate For Research

pytorch segmentation

Last synced: 12 Nov 2024

https://github.com/signatrix/efficientdet

(Pretrained weights provided) EfficientDet: Scalable and Efficient Object Detection implementation by Signatrix GmbH

coco-dataset deep-learning efficientdet efficientnet neural-network object-detection pytorch

Last synced: 07 Aug 2024

https://github.com/mkocabas/epipolarpose

Self-Supervised Learning of 3D Human Pose using Multi-view Geometry (CVPR2019)

3d-pose-estimation computer-vision cvpr cvpr-2019 cvpr19 cvpr2019 human-pose-estimation machine-learning pytorch self-supervised-learning

Last synced: 13 Nov 2024

https://github.com/mit-han-lab/lite-transformer

[ICLR 2020] Lite Transformer with Long-Short Range Attention

nlp pytorch transformer

Last synced: 03 Aug 2024

https://github.com/JiahuiYu/wdsr_ntire2018

Code of our winning entry to NTIRE super-resolution challenge, CVPR 2018

deep-neural-networks efficient-algorithm pytorch super-resolution wdsr

Last synced: 07 Aug 2024

https://github.com/richzhang/colorization-pytorch

PyTorch reimplementation of Interactive Deep Colorization

computer-graphics computer-vision convolutional-neural-networks deep-learning image-colorization pytorch siggraph

Last synced: 14 Nov 2024

https://github.com/jiahuiyu/wdsr_ntire2018

Code of our winning entry to NTIRE super-resolution challenge, CVPR 2018

deep-neural-networks efficient-algorithm pytorch super-resolution wdsr

Last synced: 09 Nov 2024

https://github.com/qinzheng93/GeoTransformer

[CVPR2022] Geometric Transformer for Fast and Robust Point Cloud Registration

attention-mechanism computer-vision deep-learning neural-network point-cloud point-cloud-registration pytorch registration transformer

Last synced: 28 Oct 2024

https://github.com/xwying/torchshow

Visualize PyTorch tensors with a single line of code.

image-tensor notebook pytorch tensor visualization

Last synced: 12 Nov 2024

https://github.com/hirofumi0810/neural_sp

End-to-end ASR/LM implementation with PyTorch

asr attention attention-mechanism automatic-speech-recognition ctc language-model language-modeling pytorch rnn-transducer seq2seq sequence-to-sequence speech speech-recognition streaming transformer transformer-xl

Last synced: 12 Nov 2024

https://github.com/castorini/hedwig

PyTorch deep learning models for document classification

deep-learning document-classification pytorch

Last synced: 06 Nov 2024

https://github.com/AviSoori1x/makeMoE

From scratch implementation of a sparse mixture of experts language model inspired by Andrej Karpathy's makemore :)

deep-learning large-language-models llm mixture-of-experts neural-networks pytorch pytorch-implementation

Last synced: 05 Nov 2024

https://github.com/facebookresearch/fairseq2

FAIR Sequence Modeling Toolkit 2

artificial-intelligence deep-learning machine-learning python pytorch

Last synced: 13 Nov 2024

https://github.com/timbmg/sentence-vae

PyTorch Re-Implementation of "Generating Sentences from a Continuous Space" by Bowman et al 2015 https://arxiv.org/abs/1511.06349

deep-learning generative-model neural-network nlp ptb pytorch vae

Last synced: 13 Nov 2024

https://github.com/szagoruyko/functional-zoo

PyTorch and Tensorflow functional model definitions

pretrained-models pytorch tensorflow

Last synced: 29 Oct 2024

https://github.com/bowenc0221/panoptic-deeplab

This is Pytorch re-implementation of our CVPR 2020 paper "Panoptic-DeepLab: A Simple, Strong, and Fast Baseline for Bottom-Up Panoptic Segmentation" (https://arxiv.org/abs/1911.10194)

bottom-up cityscapes cvpr2020 deeplab detectron2 instance-segmentation panoptic-segmentation pytorch semantic-segmentation sementation

Last synced: 29 Oct 2024

https://github.com/hrnbot/Basic-Mathematics-for-Machine-Learning

The motive behind Creating this repo is to feel the fear of mathematics and do what ever you want to do in Machine Learning , Deep Learning and other fields of AI

ai algebra beginner calculus linear-algebra machine-learning machinelearning-python mathematical-analysis mathematical-functions mathematics maths notebook prerequisites probability python pytorch siraj siraj-raval statistics

Last synced: 11 Nov 2024

https://github.com/Malllabiisc/CompGCN

ICLR 2020: Composition-Based Multi-Relational Graph Convolutional Networks

deep-learning graph-convolutional-networks graph-representation-learning iclr2020 link-prediction pytorch relation-embeddings

Last synced: 08 Aug 2024

https://github.com/volcengine/vescale

A PyTorch Native LLM Training Framework

llm-training pytorch

Last synced: 10 Nov 2024

https://github.com/quva-lab/e2cnn

E(2)-Equivariant CNNs Library for Pytorch

cnns equivariance equivariant-network group-convolution pytorch

Last synced: 08 Nov 2024

https://github.com/QUVA-Lab/e2cnn

E(2)-Equivariant CNNs Library for Pytorch

cnns equivariance equivariant-network group-convolution pytorch

Last synced: 09 Aug 2024

https://github.com/miguelvr/dropblock

Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.

computer-vision convolutional-neural-networks dropblock dropout machine-learning pytorch pytorch-implementation regularization

Last synced: 01 Nov 2024

https://github.com/sbi-dev/sbi

Simulation-based inference toolkit

bayesian-inference likelihood-free-inference machine-learning parameter-estimation pytorch simulation-based-inference

Last synced: 13 Nov 2024

https://github.com/enot-autodl/onnx2torch

Convert ONNX models to PyTorch.

convert export onnx pytorch

Last synced: 13 Oct 2024

https://github.com/timbmg/vae-cvae-mnist

Variational Autoencoder and Conditional Variational Autoencoder on MNIST in PyTorch

cvae deep-learning latent-variable-models mnist pytorch vae variational-autoencoder

Last synced: 01 Nov 2024

https://github.com/as-ideas/ForwardTacotron

⏩ Generating speech in a single forward pass without any attention!

axelspringerai deep-learning forwardtacotron python pytorch tacotron text-to-speech tts

Last synced: 04 Nov 2024

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

[ECCV 2020] Searching Efficient 3D Architectures with Sparse Point-Voxel Convolution

3d-deep-learning architecture-search computer-vision deep-learning efficiency lidar point-cloud pytorch semantickitti

Last synced: 28 Oct 2024

https://github.com/graykode/xlnet-Pytorch

Simple XLNet implementation with Pytorch Wrapper

bert natural-language-processing nlp pytorch xlnet xlnet-pytorch

Last synced: 04 Nov 2024

https://github.com/graykode/xlnet-pytorch

Simple XLNet implementation with Pytorch Wrapper

bert natural-language-processing nlp pytorch xlnet xlnet-pytorch

Last synced: 13 Nov 2024

https://github.com/HarryVolek/PyTorch_Speaker_Verification

PyTorch implementation of "Generalized End-to-End Loss for Speaker Verification" by Wan, Li et al.

pytorch speaker-identification speaker-verification

Last synced: 11 Nov 2024

https://github.com/stefanopini/simple-hrnet

Multi-person Human Pose Estimation with HRNet in Pytorch

coco-dataset computer-vision deep-learning hrnet human-pose-estimation keypoint-detection machine-learning pytorch yolov3

Last synced: 14 Nov 2024

https://github.com/Confusezius/Deep-Metric-Learning-Baselines

PyTorch Implementation for Deep Metric Learning Pipelines

cars196 computer-vision cub200 deep-learning deep-metric-learning distance-sampling metric-learning neural-networks pku-vehicle pytorch shop-clothes

Last synced: 13 Nov 2024

https://github.com/awarebayes/RecNN

Reinforced Recommendation toolkit built around pytorch 1.7

library news-recommendation pytorch pytorch-rl recommendation-system recommender-system reinforcement-learning toolkit

Last synced: 08 Aug 2024

https://github.com/songdejia/EAST

This is a pytorch re-implementation of EAST: An Efficient and Accurate Scene Text Detector.

deeplearning east icdar ocr pytorch textdetection

Last synced: 03 Nov 2024

https://github.com/benedekrozemberczki/GraphWaveletNeuralNetwork

A PyTorch implementation of "Graph Wavelet Neural Network" (ICLR 2019)

convolutional-neural-networks deep-learning deepwalk gcn gpt2 gpt3 graph-convolution graph-convolutional-network graph-neural-networks graph-representation graphsage gwnn iclr machine-learning node2vec pytorch research sklearn tensorflow transformer

Last synced: 08 Nov 2024

https://github.com/ikostrikov/pytorch-flows

PyTorch implementations of algorithms for density estimation

deep-learning density-estimation neural-networks probabilities pytorch

Last synced: 12 Nov 2024

https://github.com/hpcaitech/FastFold

Optimizing AlphaFold Training and Inference on GPU Clusters

alphafold2 cuda evoformer gpu habana-gaudi parallelism protein-folding protein-structure pytorch

Last synced: 12 Nov 2024

https://github.com/fxia22/stn.pytorch

pytorch version of spatial transformer networks

pytorch stn

Last synced: 13 Nov 2024

https://github.com/achuthasubhash/Complete-Life-Cycle-of-a-Data-Science-Project

Complete-Life-Cycle-of-a-Data-Science-Project

analysis data-analysis data-science dataset deep-learning eda exploratory-data-analysis feature-engineering federated-learning machine-learning nlp-models python python-library pytorch reinforcement-learning scraper supervised-learning transfer-learning unsupervised-learning web-scraping

Last synced: 13 Nov 2024

https://github.com/NoCLin/LightMirrors

LightMirrors is a lightweight mirror server with caching capabilities that currently supports DockerHub, K8S, PyPI, PyTorch, and NPM.

cache-proxy docker docker-compose dockerhub ghcr k8s mirror mirror-site mirrors npm nvcr pypi python pytorch quay reverse-proxy

Last synced: 29 Oct 2024

https://github.com/ShuangXieIrene/ssds.pytorch

Repository for Single Shot MultiBox Detector and its variants, implemented with pytorch, python3.

darknet fssd mobilenet pytorch rfb ssd vgg yolo

Last synced: 03 Aug 2024

https://github.com/hassony2/manopth

MANO layer for PyTorch, generating hand meshes as a differentiable layer

hand layer mano pytorch

Last synced: 17 Aug 2024

https://github.com/szymonmaszke/torchlayers

Shape and dimension inference (Keras-like) for PyTorch layers and neural networks

deep-learning framework inference keras layers library machine-learning pytorch shape sota

Last synced: 11 Oct 2024

https://github.com/benedekrozemberczki/graphwaveletneuralnetwork

A PyTorch implementation of "Graph Wavelet Neural Network" (ICLR 2019)

convolutional-neural-networks deep-learning deepwalk gcn gpt2 gpt3 graph-convolution graph-convolutional-network graph-neural-networks graph-representation graphsage gwnn iclr machine-learning node2vec pytorch research sklearn tensorflow transformer

Last synced: 10 Oct 2024

https://github.com/bryanyzhu/two-stream-pytorch

PyTorch implementation of two-stream networks for video action recognition

action-recognition pytorch two-stream video

Last synced: 26 Oct 2024

https://github.com/cheind/pytorch-blender

:sweat_drops: Seamless, distributed, real-time integration of Blender into PyTorch data pipelines

blender openai-gym pytorch reinforcement-learning zmq

Last synced: 13 Nov 2024

https://github.com/hpcaitech/fastfold

Optimizing AlphaFold Training and Inference on GPU Clusters

alphafold2 cuda evoformer gpu habana-gaudi parallelism protein-folding protein-structure pytorch

Last synced: 02 Nov 2024

https://github.com/Scalsol/mega.pytorch

Memory Enhanced Global-Local Aggregation for Video Object Detection, CVPR2020

pytorch video-object-detection

Last synced: 11 Nov 2024

https://github.com/yinboc/few-shot-meta-baseline

Meta-Baseline: Exploring Simple Meta-Learning for Few-Shot Learning, in ICCV 2021

few-shot-learning machine-learning pytorch

Last synced: 08 Nov 2024

https://github.com/loudinthecloud/pytorch-ntm

Neural Turing Machines (NTM) - PyTorch Implementation

lstm neural-network neural-turing-machines notebook ntm python pytorch

Last synced: 07 Aug 2024

https://github.com/Hawkeye-FineGrained/Hawkeye

Open source deep learning based fine-grained image recognition toolbox built on PyTorch🔥

deep-learning fine-grained pytorch toolbox

Last synced: 05 Nov 2024

https://github.com/foolwood/pytorch-slimming

Learning Efficient Convolutional Networks through Network Slimming, In ICCV 2017.

deep-learning fast-inference l1-regularization pytorch weight-pruning

Last synced: 09 Nov 2024

https://github.com/underneathall/pinferencia

Python + Inference - Model Deployment library in Python. Simplest model inference server ever.

ai artificial-intelligence computer-vision data-science deep-learning huggingface inference inference-server machine-learning model-deployment model-serving modelserver nlp paddlepaddle predict python pytorch serving tensorflow transformers

Last synced: 14 Nov 2024

https://github.com/GRAAL-Research/poutyne

A simplified framework and utilities for PyTorch

data-science deep-learning keras machine-learning neural-network python pytorch

Last synced: 30 Oct 2024

https://github.com/mackelab/sbi

Simulation-based inference toolkit

bayesian-inference likelihood-free-inference machine-learning parameter-estimation pytorch simulation-based-inference

Last synced: 24 Aug 2024

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

[ECCV2022] MOTR: End-to-End Multiple-Object Tracking with TRansformer

end-to-end multi-object-tracking pytorch transformer

Last synced: 28 Oct 2024

https://github.com/KDD-OpenSource/DeepADoTS

Repository of the paper "A Systematic Evaluation of Deep Anomaly Detection Methods for Time Series".

anomaly-detection deep-learning pytorch tensorflow time-series timeseries

Last synced: 26 Oct 2024

https://github.com/ignacio-rocco/detectorch

Detectorch - detectron for PyTorch

detectron instance-segmentation object-detection object-segmentation python pytorch

Last synced: 14 Nov 2024

https://github.com/opendilab/di-drive

Decision Intelligence Platform for Autonomous Driving simulation.

autodrive autonomous-driving carla imitation-learning metadrive pytorch reinforcement-learning

Last synced: 09 Nov 2024

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

Unofficial PyTorch Implementation for FaceShifter (https://arxiv.org/abs/1912.13457)

face-swapping pytorch pytorch-lightning

Last synced: 11 Nov 2024

https://github.com/webis-de/small-text

Active Learning for Text Classification in Python

active-learning deep-learning looking-for-contributors machine-learning natural-language-processing nlp python pytorch text-classification transformers

Last synced: 13 Nov 2024

https://github.com/aimagelab/mammoth

An Extendible (General) Continual Learning Framework based on Pytorch - official codebase of Dark Experience for General Continual Learning

continual-learning dark-experience-replay deep-learning der experience-replay knowledge-distillation neurips2020 pytorch

Last synced: 14 Nov 2024

https://github.com/RobotLocomotion/pytorch-dense-correspondence

Code for "Dense Object Nets: Learning Dense Visual Object Descriptors By and For Robotic Manipulation"

3d artificial-intelligence computer-vision deep-learning manipulation pytorch robotics self-supervised-learning vision

Last synced: 14 Nov 2024

https://github.com/longcw/RoIAlign.pytorch

RoIAlign & crop_and_resize for PyTorch

pytorch

Last synced: 14 Nov 2024

https://github.com/longcw/roialign.pytorch

RoIAlign & crop_and_resize for PyTorch

pytorch

Last synced: 05 Nov 2024

https://github.com/stochasticai/x-stable-diffusion

Real-time inference for Stable Diffusion - 0.88s latency. Covers AITemplate, nvFuser, TensorRT, FlashAttention. Join our Discord communty: https://discord.com/invite/TgHXuSJEk6

aitemplate automl cuda docker inference notebook nvfuser onnx onnxruntime pytorch stable-diffusion tensorrt

Last synced: 09 Nov 2024

https://github.com/huggingface/transformers-bloom-inference

Fast Inference Solutions for BLOOM

bloom huggingface-transformers nlp pytorch

Last synced: 03 Aug 2024

https://github.com/RelevanceAI/vectorhub

Vector Hub - Library for easy discovery, and consumption of State-of-the-art models to turn data into vectors. (text2vec, image2vec, video2vec, graph2vec, bert, inception, etc)

artificial-intelligence audio-processing deep-learning deeplearning embeddings encodings image2vec machine-learning neural-network python pytorch tensorflow tfhub transformers vector vector-similarity video-processing word2vec

Last synced: 11 Nov 2024

https://github.com/d2l-ai/d2l-vi

Một cuốn sách về Học Sâu đề cập đến nhiều framework phổ biến, được sử dụng trên 300 trường Đại học từ 55 đất nước bao gồm MIT, Stanford, Harvard, và Cambridge.

computer-vision d2l deep-learning kaggle keras machine-learning mlbvn mxnet python pytorch tensorflow vietnamese-language

Last synced: 08 Nov 2024

https://github.com/relevanceai/vectorhub

Vector Hub - Library for easy discovery, and consumption of State-of-the-art models to turn data into vectors. (text2vec, image2vec, video2vec, graph2vec, bert, inception, etc)

artificial-intelligence audio-processing deep-learning deeplearning embeddings encodings image2vec machine-learning neural-network python pytorch tensorflow tfhub transformers vector vector-similarity video-processing word2vec