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/szagoruyko/functional-zoo

PyTorch and Tensorflow functional model definitions

pretrained-models pytorch tensorflow

Last synced: 05 Jan 2025

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/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: 05 Jan 2025

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

E(2)-Equivariant CNNs Library for Pytorch

cnns equivariance equivariant-network group-convolution pytorch

Last synced: 29 Nov 2024

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

E(2)-Equivariant CNNs Library for Pytorch

cnns equivariance equivariant-network group-convolution pytorch

Last synced: 04 Jan 2025

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: 05 Jan 2025

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

Simple XLNet implementation with Pytorch Wrapper

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

Last synced: 05 Jan 2025

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: 11 Jan 2025

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: 04 Jan 2025

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/opendilab/di-drive

Decision Intelligence Platform for Autonomous Driving simulation.

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

Last synced: 05 Jan 2025

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/loudinthecloud/pytorch-ntm

Neural Turing Machines (NTM) - PyTorch Implementation

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

Last synced: 27 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: 04 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/hpcaitech/fastfold

Optimizing AlphaFold Training and Inference on GPU Clusters

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

Last synced: 05 Jan 2025

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

Fast Differentiable Sorting and Ranking

differentiable jax pytorch ranking sorting tensorflow

Last synced: 05 Jan 2025

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/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: 11 Jan 2025

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: 05 Jan 2025

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: 05 Jan 2025

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: 18 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/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: 05 Jan 2025

https://github.com/hassony2/manopth

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

hand layer mano pytorch

Last synced: 09 Dec 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: 05 Jan 2025

https://github.com/OlaWod/FreeVC

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

pytorch speech voice-conversion

Last synced: 18 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/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/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: 05 Jan 2025

https://github.com/thohemp/6drepnet

Official Pytorch implementation of 6DRepNet: 6D Rotation representation for unconstrained head pose estimation.

6d aflw2000 analysis biwi estimation facial head head-pose head-pose-estimation orientation pose pytorch pytorch-implementation

Last synced: 05 Jan 2025

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/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/megvii-model/MOTR

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

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

Last synced: 20 Dec 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/ctallec/world-models

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

model-based-rl pytorch reinforcement-learning

Last synced: 27 Nov 2024

https://github.com/alex04072000/fusta

Hybrid Neural Fusion for Full-frame Video Stabilization

iccv2021 pytorch video-stabilization view-synthesis

Last synced: 04 Jan 2025

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: 05 Jan 2025

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

Python library assists deep learning on graphs

deep-learning graph-neural-networks pytorch

Last synced: 05 Jan 2025

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: 04 Jan 2025

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

RoIAlign & crop_and_resize for PyTorch

pytorch

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

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

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

framework pytorch ranking recommendation

Last synced: 28 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: 05 Jan 2025

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: 04 Jan 2025

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: 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: 07 Jan 2025

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/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: 07 Jan 2025

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: 28 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/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: 05 Jan 2025

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: 05 Jan 2025

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/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/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/farukalamai/advanced-machine-learning-engineer-roadmap-2024

A Full Stack ML (Machine Learning) Roadmap involves learning the necessary skills and technologies to become proficient in all aspects of machine learning, including data collection and preprocessing, model development, deployment, and maintenance.

aws computer-vision data-analysis data-science data-visualization deep-learning git-github machine-learning machine-learning-roadmap mlops natural-language-processing neural-network nlp opencv pandas python pytorch statistics tensorflow yolo

Last synced: 05 Jan 2025

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

Microsoft Automatic Mixed Precision Library

amp deep-learning fp8 gpu mixed-precision pytorch transformer

Last synced: 04 Jan 2025

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: 10 Jan 2025

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: 05 Jan 2025

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

基于pytorch的bert_bilstm_crf中文命名实体识别

bert crf named-entity-recognition ner pytorch

Last synced: 05 Jan 2025

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/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: 05 Jan 2025

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/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: 09 Jan 2025

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: 22 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: 27 Nov 2024