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

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

TernausNetV2: Fully Convolutional Network for Instance Segmentation

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

Last synced: 16 Nov 2024

https://github.com/snap-stanford/deepsnap

Python library assists deep learning on graphs

deep-learning graph-neural-networks pytorch

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

https://github.com/google-research/fast-soft-sort

Fast Differentiable Sorting and Ranking

differentiable jax pytorch ranking sorting tensorflow

Last synced: 10 Nov 2024

https://github.com/AmazingDD/daisyRec

Official code for "DaisyRec 2.0: Benchmarking Recommendation for Rigorous Evaluation" (TPAMI2022) and "Are We Evaluating Rigorously? Benchmarking Recommendation for Reproducible Evaluation and Fair Comparison" (RecSys2020)

afm biasmf cdae collaborative-filtering deepfm ease factorization-machines item2vec k-nearest-neighbors matrix-factorization neumf neural-collaborative-filtering nfm pytorch recommender-system slim svdpp vae

Last synced: 08 Aug 2024

https://github.com/vector-ai/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: 15 Aug 2024

https://github.com/ctallec/world-models

Reimplementation of World-Models (Ha and Schmidhuber 2018) in pytorch

model-based-rl pytorch reinforcement-learning

Last synced: 07 Aug 2024

https://github.com/pemami4911/neural-combinatorial-rl-pytorch

PyTorch implementation of Neural Combinatorial Optimization with Reinforcement Learning https://arxiv.org/abs/1611.09940

neural-combinatorial-optimization pytorch reinforcement-learning seq2seq

Last synced: 02 Nov 2024

https://github.com/voidful/textrl

Implementation of ChatGPT RLHF (Reinforcement Learning with Human Feedback) on any generation model in huggingface's transformer (blommz-176B/bloom/gpt/bart/T5/MetaICL)

chatgpt controlled-nlg gpt-2 gpt-3 language-model nlg nlp pytorch reinforcement-learning rlhf

Last synced: 09 Nov 2024

https://github.com/voidful/TextRL

Implementation of ChatGPT RLHF (Reinforcement Learning with Human Feedback) on any generation model in huggingface's transformer (blommz-176B/bloom/gpt/bart/T5/MetaICL)

chatgpt controlled-nlg gpt-2 gpt-3 language-model nlg nlp pytorch reinforcement-learning rlhf

Last synced: 31 Oct 2024

https://github.com/clcarwin/convert_torch_to_pytorch

Convert torch t7 model to pytorch model and source.

pytorch

Last synced: 14 Nov 2024

https://github.com/pyg-team/pytorch-frame

Tabular Deep Learning Library for PyTorch

data-frame deep-learning pytorch tabular-learning

Last synced: 12 Nov 2024

https://github.com/tensorlayer/tensorlayerx

TensorLayerX: A Unified Deep Learning and Reinforcement Learning Framework for All Hardwares, Backends and OS.

deep-learning jittor machine-learning mindspore neural-network oneflow paddlepaddle python pytorch tensorflow tensorlayer tensorlayerx

Last synced: 13 Oct 2024

https://github.com/akanimax/pro_gan_pytorch

Unofficial PyTorch implementation of the paper titled "Progressive growing of GANs for improved Quality, Stability, and Variation"

adversarial-machine-learning convolutional-neural-network gan progressive-growing-of-gans pytorch

Last synced: 14 Nov 2024

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

A scene text recognition toolbox based on PyTorch

ocr ocr-recognition pytorch scene-text-recognition text-recognition transformer

Last synced: 03 Nov 2024

https://github.com/Toni-SM/skrl

Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Omniverse Isaac Gym and Isaac Lab

deep-learning deepmind gym gymnasium isaac-gym isaac-lab isaac-orbit isaac-sim isaaclab jax machine-learning nvidia-omniverse openai-gym python pytorch reinforcement-learning rl robosuite robotics skrl

Last synced: 03 Nov 2024

https://github.com/WuHuRestaurant/xf_event_extraction2020Top1

科大讯飞2020事件抽取挑战赛第一名解决方案&完整事件抽取系统

competition events-extraction pytorch

Last synced: 06 Nov 2024

https://github.com/ViTAE-Transformer/ViTDet

Unofficial implementation for [ECCV'22] "Exploring Plain Vision Transformer Backbones for Object Detection"

deep-learning object-detection pytorch vision-transformer

Last synced: 15 Nov 2024

https://github.com/tensorlayer/TensorLayerX

TensorLayerX: A Unified Deep Learning and Reinforcement Learning Framework for All Hardwares, Backends and OS.

deep-learning jittor machine-learning mindspore neural-network oneflow paddlepaddle python pytorch tensorflow tensorlayer tensorlayerx

Last synced: 04 Aug 2024

https://github.com/kmeng01/rome

Locating and editing factual associations in GPT (NeurIPS 2022)

gpt interpretability pytorch transformers

Last synced: 30 Oct 2024

https://github.com/junyanz/von

[NeurIPS 2018] Visual Object Networks: Image Generation with Disentangled 3D Representation.

3d cyclegan disentangled-representations gans generative-adversarial-network pix2pix pytorch shape-generation

Last synced: 13 Nov 2024

https://github.com/610265158/Peppa_Pig_Face_Landmark

A simple face detect and alignment method, which is easy and stable.

face-alignment face-keypoint face-mask facedetect keypoint pytorch

Last synced: 13 Nov 2024

https://github.com/VITA-Group/FasterSeg

[ICLR 2020] "FasterSeg: Searching for Faster Real-time Semantic Segmentation" by Wuyang Chen, Xinyu Gong, Xianming Liu, Qian Zhang, Yuan Li, Zhangyang Wang

cityscapes latency neural-architecture-search pytorch semantic-segmentation tensorrt

Last synced: 07 Aug 2024

https://github.com/vitae-transformer/vitdet

Unofficial implementation for [ECCV'22] "Exploring Plain Vision Transformer Backbones for Object Detection"

deep-learning object-detection pytorch vision-transformer

Last synced: 14 Nov 2024

https://github.com/rowanz/neural-motifs

Code for Neural Motifs: Scene Graph Parsing with Global Context (CVPR 2018)

pytorch scene-graph vision visual-genome

Last synced: 02 Nov 2024

https://github.com/dair-ai/pytorch_notebooks

🔥 A collection of PyTorch notebooks for learning and practicing deep learning

deep-learning machine-learning nlp pytorch

Last synced: 03 Sep 2024

https://github.com/bradyz/cross_view_transformers

Cross-view Transformers for real-time Map-view Semantic Segmentation (CVPR 2022 Oral)

cvpr2022 deep-learning pytorch transformer

Last synced: 28 Oct 2024

https://github.com/allenai/allennlp-models

Officially supported AllenNLP models

allennlp nlp pytorch

Last synced: 26 Sep 2024

https://github.com/OlaWod/FreeVC

FreeVC: Towards High-Quality Text-Free One-Shot Voice Conversion

pytorch speech voice-conversion

Last synced: 03 Aug 2024

https://github.com/hkchengrex/stcn

[NeurIPS 2021] Rethinking Space-Time Networks with Improved Memory Coverage for Efficient Video Object Segmentation

computer-vision deep-learning neurips-2021 pytorch segmentation video-object-segmentation video-segmentation

Last synced: 10 Nov 2024

https://github.com/HoloClean/holoclean

A Machine Learning System for Data Enrichment.

data-enrichment data-science inference-engine machine-learning pytorch

Last synced: 12 Nov 2024

https://github.com/azure/ms-amp

Microsoft Automatic Mixed Precision Library

amp deep-learning fp8 gpu mixed-precision pytorch transformer

Last synced: 13 Nov 2024

https://github.com/locuslab/optnet

OptNet: Differentiable Optimization as a Layer in Neural Networks

deep-learning machine-learning optimization paper pytorch

Last synced: 10 Nov 2024

https://github.com/Oldpan/Faceswap-Deepfake-Pytorch

Faceswap with Pytorch or DeepFake with Pytorch

faceswap pytorch

Last synced: 08 Nov 2024

https://github.com/xiaoyufenfei/LEDNet

LEDNet: A Lightweight Encoder-Decoder Network for Real-time Semantic Segmentation

cityscape-dataset computer-vision lednet pytorch real-time semantic-segmentation

Last synced: 14 Nov 2024

https://github.com/DagnyT/hardnet

Hardnet descriptor model - "Working hard to know your neighbor's margins: Local descriptor learning loss"

computer-vision convolutional-neural-networks deep-learning image-matching image-retrieval metric-learning nips-2017 pytorch

Last synced: 04 Nov 2024

https://github.com/fancompute/wavetorch

🌊 Numerically solving and backpropagating through the wave equation

differential-equations pytorch recurrent-neural-network rnn time-series vowel-recognition wave-equation

Last synced: 14 Nov 2024

https://github.com/WangQvQ/YOLOMagic

YOLO Magic🪄 is an extension based on Ultralytics' YOLOv5, designed to provide more powerful functionality and simpler operations for visual tasks.

deep-learning gradio machine-learning onnx pytorch tflite yolo yolov5

Last synced: 09 Nov 2024

https://github.com/Audio-WestlakeU/FullSubNet

PyTorch implementation of "FullSubNet: A Full-Band and Sub-Band Fusion Model for Real-Time Single-Channel Speech Enhancement."

audio band denoising full-band narrow-band noise-reduction paper pretrained-model pytorch reproducible-research single-channel speech speech-enhancement speech-processing speech-separation sub-band

Last synced: 05 Aug 2024

https://github.com/THUwangcy/ReChorus

“Chorus” of recommendation models: a light and flexible PyTorch framework for Top-K recommendation.

framework pytorch ranking recommendation

Last synced: 08 Aug 2024

https://github.com/cardwing/codes-for-pvkd

Point-to-Voxel Knowledge Distillation for LiDAR Semantic Segmentation (CVPR 2022)

cvpr2022 lidar-segmentation nuscenes point-cloud-segmentation pytorch range-image semantickitti waymo-dataset

Last synced: 12 Nov 2024

https://github.com/allenai/tango

Organize your experiments into discrete steps that can be cached and reused throughout the lifetime of your research project.

ai machine-learning nlp python python3 pytorch

Last synced: 01 Oct 2024

https://github.com/prlz77/ResNeXt.pytorch

Reproduces ResNet-V3 with pytorch

cifar pytorch resnet resnext

Last synced: 31 Oct 2024

https://github.com/prlz77/resnext.pytorch

Reproduces ResNet-V3 with pytorch

cifar pytorch resnet resnext

Last synced: 13 Nov 2024

https://github.com/understandable-machine-intelligence-lab/Quantus

Quantus is an eXplainable AI toolkit for responsible evaluation of neural network explanations

deep-learning explainable-ai interpretability machine-learning pytorch quantification-evaluation-methods reproducibility tensorflow xai

Last synced: 15 Nov 2024

https://github.com/medipixel/rl_algorithms

Structural implementation of RL key algorithms

deep-learning dqn gym policy-gradient python3 pytorch reinforcement-learning

Last synced: 11 Nov 2024

https://github.com/jingyibysutsoftware/yolov5-deepsort-driverdistracted-driving-behavior-detection

基于深度学习的驾驶员分心驾驶行为(疲劳+危险行为)预警系统使用YOLOv5+Deepsort实现驾驶员的危险驾驶行为的预警监测

deeplearning deepsort detection dlib dlib-face-detection dlib-tracker driver-behavior object-detection opencv playsound pyqt5 python pytorch sort tired tracker yolov5 yolov5-deepsort

Last synced: 14 Nov 2024

https://github.com/facebookresearch/neuralcompression

A collection of tools for neural compression enthusiasts.

compression deep-learning jax machine-learning neural-compression python pytorch

Last synced: 13 Nov 2024

https://github.com/hfslyc/AdvSemiSeg

Adversarial Learning for Semi-supervised Semantic Segmentation, BMVC 2018

adversarial-learning computer-vision deep-learning pytorch semantic-segmentation semi-supervised-learning

Last synced: 28 Oct 2024

https://github.com/yeyupiaoling/whisper-finetune

Fine-tune the Whisper speech recognition model to support training without timestamp data, training with timestamp data, and training without speech data. Accelerate inference and support Web deployment, Windows desktop deployment, and Android deployment

android asr chinese ctranslate2 huggingface lora pytorch speech-recognition transformers web whisper

Last synced: 09 Oct 2024

https://github.com/yeyupiaoling/Whisper-Finetune

Fine-tune the Whisper speech recognition model to support training without timestamp data, training with timestamp data, and training without speech data. Accelerate inference and support Web deployment, Windows desktop deployment, and Android deployment

android asr chinese ctranslate2 huggingface lora pytorch speech-recognition transformers web whisper

Last synced: 09 Oct 2024

https://github.com/v-iashin/video_features

Extract video features from raw videos using multiple GPUs. We support RAFT flow frames as well as S3D, I3D, R(2+1)D, VGGish, CLIP, and TIMM models.

audio-features clip feature-extraction i3d ig65m laion multi-gpu optical-flow parallel pytorch r2plus1d raft resnet s3d swin timm vggish video-features visual-features vit

Last synced: 03 Nov 2024

https://github.com/flaport/fdtd

A 3D electromagnetic FDTD simulator written in Python with optional GPU support

3d-fdtd electric-fields fdtd fdtd-simulator magnetic-fields numpy optics photonics physics physics-simulation python pytorch simulation simulation-framework

Last synced: 13 Nov 2024

https://github.com/shayneobrien/generative-models

Annotated, understandable, and visually interpretable PyTorch implementations of: VAE, BIRVAE, NSGAN, MMGAN, WGAN, WGANGP, LSGAN, DRAGAN, BEGAN, RaGAN, InfoGAN, fGAN, FisherGAN

autoencoder began discriminator dragan fgan fishergan gan generative-adversarial-network generative-models infogan lsgan machine-learning mmgan nsgan python pytorch ragan vae wasserstein wgan

Last synced: 02 Nov 2024

https://github.com/cardwing/Codes-for-PVKD

Point-to-Voxel Knowledge Distillation for LiDAR Semantic Segmentation (CVPR 2022)

cvpr2022 lidar-segmentation nuscenes point-cloud-segmentation pytorch range-image semantickitti waymo-dataset

Last synced: 28 Oct 2024

https://github.com/swz30/CycleISP

[CVPR 2020--Oral] CycleISP: Real Image Restoration via Improved Data Synthesis

camera-imaging-pipeline computer-vision cvpr2020 cycleisp data-synthesis image-denoising image-restoration low-level-vision pytorch raw2rgb rgb2raw

Last synced: 03 Nov 2024

https://github.com/yihongXU/deepMOT

Official Implementation of How To Train Your Deep Multi-Object Tracker (CVPR2020)

deep-learning multi-object-tracking python pytorch

Last synced: 13 Nov 2024

https://github.com/ntmc-community/matchzoo-py

Facilitating the design, comparison and sharing of deep text matching models.

deep-learning matching natural-language-processing neural-network pytorch text text-matching

Last synced: 10 Nov 2024

https://github.com/NTMC-Community/MatchZoo-py

Facilitating the design, comparison and sharing of deep text matching models.

deep-learning matching natural-language-processing neural-network pytorch text text-matching

Last synced: 15 Nov 2024

https://github.com/chrischoy/pytorch-custom-cuda-tutorial

Tutorial for building a custom CUDA function for Pytorch

python pytorch pytorch-backend tutorial wrapper

Last synced: 14 Nov 2024

https://github.com/ldeecke/gmm-torch

Gaussian mixture models in PyTorch.

gaussian-mixture-models pytorch

Last synced: 05 Aug 2024

https://github.com/PennyLaneAI/qml

Introductions to key concepts in quantum programming, as well as tutorials and implementations from cutting-edge quantum computing research.

autograd automatic-differentiation demo key-concepts neural-networks pytorch qml quantum-chemistry quantum-computing quantum-machine-learning tensorflow tutorials

Last synced: 03 Nov 2024

https://github.com/ksw0306/FloWaveNet

A Pytorch implementation of "FloWaveNet: A Generative Flow for Raw Audio"

clarinet generative-flow glow pytorch wavenet

Last synced: 03 Aug 2024

https://github.com/adambielski/capsnet-pytorch

PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules

capsnet capsules deep-learning dynamic-routing-between-capsules machine-learning mnist pytorch

Last synced: 13 Nov 2024

https://github.com/harry24k/bayesian-neural-network-pytorch

PyTorch implementation of bayesian neural network [torchbnn]

bayesian deep-learning neural-network pytorch

Last synced: 14 Nov 2024

https://github.com/facebookresearch/NeuralCompression

A collection of tools for neural compression enthusiasts.

compression deep-learning jax machine-learning neural-compression python pytorch

Last synced: 05 Aug 2024

https://github.com/fartashf/vsepp

PyTorch Code for the paper "VSE++: Improving Visual-Semantic Embeddings with Hard Negatives"

bmvc negatives paper pytorch vse

Last synced: 15 Nov 2024

https://github.com/dongminlee94/deep_rl

PyTorch implementation of deep reinforcement learning algorithms

a2c ddpg ddqn deep-reinforcement-learning dqn model-free-rl npg ppo pytorch sac sac-aea td3 trpo vpg

Last synced: 11 Nov 2024

https://github.com/salesforce/matchbox

Write PyTorch code at the level of individual examples, then run it efficiently on minibatches.

deep-learning minibatch nlp pytorch

Last synced: 14 Nov 2024