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/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/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/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: 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: 16 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/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/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: 16 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/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/ternaus/TernausNetV2

TernausNetV2: Fully Convolutional Network for Instance Segmentation

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

Last synced: 18 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/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/google-research/fast-soft-sort

Fast Differentiable Sorting and Ranking

differentiable jax pytorch ranking sorting tensorflow

Last synced: 17 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/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: 17 Nov 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/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: 18 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/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/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/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/zalandoresearch/pytorch-vq-vae

PyTorch implementation of VQ-VAE by Aäron van den Oord et al.

deep-learning pytorch vae vq-vae

Last synced: 17 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: 16 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/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/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/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/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: 17 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/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/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/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/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/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/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/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/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/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/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/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/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: 17 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/ldeecke/gmm-torch

Gaussian mixture models in PyTorch.

gaussian-mixture-models pytorch

Last synced: 05 Aug 2024