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/lcswillems/rl-starter-files

RL starter files in order to immediately train, visualize and evaluate an agent without writing any line of code

a2c a3c minigrid multi-process ppo preprocessed-observations pytorch reward-shaping

Last synced: 07 Aug 2024

https://github.com/lucasjinreal/DCNv2_latest

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

centernet dcnv2 pytorch

Last synced: 27 Oct 2024

https://github.com/lucastabelini/LaneATT

Code for the paper entitled "Keep your Eyes on the Lane: Real-time Attention-guided Lane Detection" (CVPR 2021)

computer-vision deep-learning lane-detection pytorch

Last synced: 28 Oct 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/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/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/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: 15 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/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/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/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/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/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/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/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/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/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/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: 15 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/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/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/enot-autodl/onnx2torch

Convert ONNX models to PyTorch.

convert export onnx pytorch

Last synced: 13 Oct 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/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/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/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/fxia22/stn.pytorch

pytorch version of spatial transformer networks

pytorch stn

Last synced: 13 Nov 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/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/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/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/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/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/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/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/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/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/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/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/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/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/project-monai/monailabel

MONAI Label is an intelligent open source image labeling and learning tool.

3d 3d-slicer-extension active-learning deep-learning machine-learning medical-imaging monai pytorch segmentation

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

Last synced: 11 Nov 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: 09 Nov 2024

https://github.com/xinjli/allosaurus

Allosaurus is a pretrained universal phone recognizer for more than 2000 languages

phonetics pytorch speech speech-recognition

Last synced: 12 Oct 2024

https://github.com/ternaus/ternausnetv2

TernausNetV2: Fully Convolutional Network for Instance Segmentation

computer-vision deep-learning image-segmentation python pytorch satellite-imagery

Last synced: 01 Nov 2024

https://github.com/tencent/forward

A library for high performance deep learning inference on NVIDIA GPUs.

cuda deep-learning forward gpu inference inference-engine keras neural-network onnx pytorch tensorflow tensorrt

Last synced: 13 Nov 2024

https://github.com/Tencent/Forward

A library for high performance deep learning inference on NVIDIA GPUs.

cuda deep-learning forward gpu inference inference-engine keras neural-network onnx pytorch tensorflow tensorrt

Last synced: 09 Nov 2024