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

An Open-Source Package for Textual Adversarial Attack.

adversarial-attacks adversarial-example natural-language-processing nlp pytorch

Last synced: 01 Nov 2024

https://github.com/locuslab/qpth

A fast and differentiable QP solver for PyTorch.

deep-learning machine-learning optimization pytorch quadratic-programming

Last synced: 10 Nov 2024

https://github.com/zdevito/aten

ATen: A TENsor library for C++11

pytorch tensor torch7

Last synced: 01 Nov 2024

https://github.com/iceychris/LibreASR

:speech_balloon: An On-Premises, Streaming Speech Recognition System

asr deep-learning esp32-lyrat fastai python pytorch rnn-transducer speech-recognition

Last synced: 05 Nov 2024

https://github.com/lucidrains/linear-attention-transformer

Transformer based on a variant of attention that is linear complexity in respect to sequence length

artificial-intelligence attention-mechanism deep-learning pytorch transformer

Last synced: 22 Oct 2024

https://github.com/koursaros-ai/nboost

NBoost is a scalable, search-api-boosting platform for deploying transformer models to improve the relevance of search results on different platforms (i.e. Elasticsearch)

cloud deep-learning docker elasticsearch helm kubernetes machine-learning microservices nboost nlp proxy python pytorch search-api search-engine semantic-search tensorflow

Last synced: 01 Nov 2024

https://github.com/zhangxiaosong18/FreeAnchor

FreeAnchor: Learning to Match Anchors for Visual Object Detection (NeurIPS 2019)

computer-vision freeanchor neurips-2019 object-detection one-stage pytorch

Last synced: 26 Oct 2024

https://github.com/breezedeus/cnstd

CnSTD: 基于 PyTorch/MXNet 的 中文/英文 场景文字检测(Scene Text Detection)、数学公式检测(Mathematical Formula Detection, MFD)、篇章分析(Layout Analysis)的Python3 包

deep-learning math-formula-detection object-detection ocr python pytorch scene-text-detection text-detection

Last synced: 30 Oct 2024

https://github.com/nnsvs/nnsvs

Neural network-based singing voice synthesis library for research

deep-learning dnn kiritan python pytorch singing-synthesis singing-voice singing-voice-synthesis

Last synced: 05 Aug 2024

https://github.com/NeuroDiffGym/neurodiffeq

A library for solving differential equations using neural networks based on PyTorch, used by multiple research groups around the world, including at Harvard IACS.

artificial-intelligence boundary-value-problem deep-learning differential-equations initial-value-problem mathematical-modelling neural-networks ode odes pde-solver physics-informed-neural-networks pinn pypi pytorch scientific-computing time-series

Last synced: 02 Aug 2024

https://github.com/Bobo-y/flexible-yolov5

More readable and flexible yolov5 with more backbone(gcn, resnet, shufflenet, moblienet, efficientnet, hrnet, swin-transformer, etc) and (cbam,dcn and so on), and tensorrt

backbone cbam dcnv2 gcn hrnet moblienet neck object-detection ptq pytorch qat resnet shufflenet sparsity swin-transformer tensorrt triton-server yolov3 yolov5

Last synced: 09 Nov 2024

https://github.com/ThibaultGROUEIX/AtlasNet

This repository contains the source codes for the paper "AtlasNet: A Papier-Mâché Approach to Learning 3D Surface Generation ". The network is able to synthesize a mesh (point cloud + connectivity) from a low-resolution point cloud, or from an image.

3d 3d-deep-learning computer-vision cvpr2018 geometry-processing pytorch

Last synced: 28 Oct 2024

https://github.com/msurtsukov/neural-ode

Jupyter notebook with Pytorch implementation of Neural Ordinary Differential Equations

jupyter-notebook neural-ode pytorch vae

Last synced: 04 Nov 2024

https://github.com/lorenmt/mtan

The implementation of "End-to-End Multi-Task Learning with Attention" [CVPR 2019].

attention-model multi-task-learning pytorch

Last synced: 03 Aug 2024

https://github.com/TropComplique/mtcnn-pytorch

Joint Face Detection and Alignment using Multi-task Cascaded Convolutional Networks

convolutional-neural-networks face-detection pytorch

Last synced: 27 Oct 2024

https://github.com/kaituoxu/Conv-TasNet

A PyTorch implementation of Conv-TasNet described in "TasNet: Surpassing Ideal Time-Frequency Masking for Speech Separation" with Permutation Invariant Training (PIT).

audio-separation conv-tasnet permutation-invariant-training pit pytorch source-separation speech-separation tasnet

Last synced: 02 Nov 2024

https://github.com/ZhendongWang6/Uformer

[CVPR 2022] Official implementation of the paper "Uformer: A General U-Shaped Transformer for Image Restoration".

deep-learning image-deblurring image-demoireing image-denoising image-deraining image-restoration pytorch

Last synced: 15 Oct 2024

https://github.com/rasbt/llm-workshop-2024

A 4-hour coding workshop to understand how LLMs are implemented and used

large-language-models llm pytorch

Last synced: 22 Oct 2024

https://github.com/jaywalnut310/glow-tts

A Generative Flow for Text-to-Speech via Monotonic Alignment Search

deep-learning pytorch speech-synthesis text-to-speech tts

Last synced: 09 Nov 2024

https://github.com/iMoonLab/HGNN

Hypergraph Neural Networks (AAAI 2019)

complex-data deep-learning hypergraph hypergraph-neural-networks pytorch

Last synced: 28 Oct 2024

https://github.com/swz30/MIRNet

[ECCV 2020] Learning Enriched Features for Real Image Restoration and Enhancement. SOTA results for image denoising, super-resolution, and image enhancement.

attention-mechanism computer-vision eccv2020 image-denoising image-enhancement image-restoration low-level-vision multi-resolution-streams pytorch super-resolution

Last synced: 03 Nov 2024

https://github.com/iMoonLab/DeepHypergraph

A pytorch library for graph and hypergraph computation.

deep-learning graph-neural-networks hypergraph-neural-networks pytorch

Last synced: 28 Oct 2024

https://github.com/TianhongDai/reinforcement-learning-algorithms

This repository contains most of pytorch implementation based classic deep reinforcement learning algorithms, including - DQN, DDQN, Dueling Network, DDPG, SAC, A2C, PPO, TRPO. (More algorithms are still in progress)

a2c actor-critic algorithm atari2600 ddpg deep-learning deep-reinforcement-learning dqn dueling-dqn flappy-bird ppo proximal-policy-optimization pytorch sac soft-actor-critic trpo trust-region-policy-optimization

Last synced: 07 Aug 2024

https://github.com/fastai/fastai2

Temporary home for fastai v2 while it's being developed

data-science deep-learning fastai jupyter machine-learning nbdev python pytorch

Last synced: 07 Aug 2024

https://github.com/pytorch/torcharrow

High performance model preprocessing library on PyTorch

preprocessing python pytorch

Last synced: 07 Oct 2024

https://github.com/limbo018/DREAMPlace

Deep learning toolkit-enabled VLSI placement

deep-learning gpu-acceleration pytorch vlsi vlsi-physical-design vlsi-placement

Last synced: 03 Aug 2024

https://github.com/BlackHC/tfpyth

Putting TensorFlow back in PyTorch, back in TensorFlow (differentiable TensorFlow PyTorch adapters).

machine-learning pytorch tensorflow

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: 07 Nov 2024

https://github.com/blackhc/tfpyth

Putting TensorFlow back in PyTorch, back in TensorFlow (differentiable TensorFlow PyTorch adapters).

machine-learning pytorch tensorflow

Last synced: 09 Nov 2024

https://github.com/wanmeihuali/taichi_3d_gaussian_splatting

An unofficial implementation of paper 3D Gaussian Splatting for Real-Time Radiance Field Rendering by taichi lang.

3d-reconstruction 3d-rendering computer-graphics computer-vision machine-learning nerf python pytorch real-time-rendering taichi

Last synced: 07 Nov 2024

https://github.com/demiseom/specaugment

A Implementation of SpecAugment with Tensorflow & Pytorch, introduced by Google Brain

data-augmentation python pytorch specaugment speech speech-recognition tensorflow

Last synced: 11 Oct 2024

https://github.com/wvangansbeke/lanedetection_end2end

End-to-end Lane Detection for Self-Driving Cars (ICCV 2019 Workshop)

computer-vision deep-learning end-to-end lane-detection least-squares pytorch self-driving-cars

Last synced: 07 Nov 2024

https://github.com/DemisEom/SpecAugment

A Implementation of SpecAugment with Tensorflow & Pytorch, introduced by Google Brain

data-augmentation python pytorch specaugment speech speech-recognition tensorflow

Last synced: 07 Aug 2024

https://github.com/monologg/JointBERT

Pytorch implementation of JointBERT: "BERT for Joint Intent Classification and Slot Filling"

bert intent-classification joint-bert pytorch slot-filling slu transformers

Last synced: 06 Nov 2024

https://github.com/wvangansbeke/LaneDetection_End2End

End-to-end Lane Detection for Self-Driving Cars (ICCV 2019 Workshop)

computer-vision deep-learning end-to-end lane-detection least-squares pytorch self-driving-cars

Last synced: 03 Aug 2024

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

[NeurIPS 2019, Spotlight] Point-Voxel CNN for Efficient 3D Deep Learning

frustum-pointnet frustum-pointnets kitti point-cloud point-voxel-cnn pointnet pointnet2 pvcnn pytorch s3dis shapenet

Last synced: 28 Oct 2024

https://github.com/perone/euclidesdb

A multi-model machine learning feature embedding database

cpp database deep-learning machine-learning pytorch search

Last synced: 30 Oct 2024

https://github.com/DerrickXuNu/OpenCOOD

[ICRA 2022] An opensource framework for cooperative detection. Official implementation for OPV2V.

autonomous-driving collaborative-perception computer-vision deep-learning multi-agent-perception multi-agent-systems pytorch simulations

Last synced: 28 Oct 2024

https://github.com/seungwonpark/melgan

MelGAN vocoder (compatible with NVIDIA/tacotron2)

gan neural-vocoder pytorch tts

Last synced: 30 Oct 2024

https://github.com/songlab-cal/tape

Tasks Assessing Protein Embeddings (TAPE), a set of five biologically relevant semi-supervised learning tasks spread across different domains of protein biology.

benchmark dataset deep-learning language-modeling protein-sequences protein-structure pytorch semi-supervised-learning

Last synced: 03 Aug 2024

https://github.com/JackWoo0831/Yolov7-tracker

Yolo X, v7, v8 and several Multi-Object Tracker(SORT, DeepSORT, ByteTrack, BoT-SORT, etc.) in MOT17 and VisDrone2019 Dataset. It uses a unified style and integrated tracker for easy embedding in your own projects.

pytorch

Last synced: 28 Oct 2024

https://github.com/lucasjinreal/dcnv2_latest

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

centernet dcnv2 pytorch

Last synced: 06 Nov 2024

https://github.com/imoonlab/deephypergraph

A pytorch library for graph and hypergraph computation.

deep-learning graph-neural-networks hypergraph-neural-networks pytorch

Last synced: 20 Oct 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: 28 Oct 2024

https://github.com/alibaba/Megatron-LLaMA

Best practice for training LLaMA models in Megatron-LM

deepspeed distributed-training llama llm megatron-lm pretraining pytorch

Last synced: 30 Oct 2024

https://github.com/BobLiu20/YOLOv3_PyTorch

Full implementation of YOLOv3 in PyTorch

object-detection pytorch yolo yolov3

Last synced: 09 Nov 2024

https://github.com/d-x-y/nas-bench-201

NAS-Bench-201 API and Instruction

automl dataset nas nas-bench-201 pytorch

Last synced: 30 Oct 2024

https://github.com/ternaus/robot-surgery-segmentation

Wining solution and its improvement for MICCAI 2017 Robotic Instrument Segmentation Sub-Challenge

computer-vision deep-learning image-segmentation medical-imaging python pytorch robot-assisted-surgery

Last synced: 01 Nov 2024

https://github.com/YU1ut/MixMatch-pytorch

Code for "MixMatch - A Holistic Approach to Semi-Supervised Learning"

deep-learning pytorch semi-supervised-learning

Last synced: 03 Aug 2024

https://github.com/sunshineatnoon/Paper-Implementations

Use PyTorch to implement some classic frameworks

computer-vision deep-learning pytorch

Last synced: 07 Aug 2024

https://github.com/SamsungLabs/ritm_interactive_segmentation

Reviving Iterative Training with Mask Guidance for Interactive Segmentation

hrnets interactive-segmentation pretrained-models pytorch segmentation

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: 05 Nov 2024

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/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/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/d-li14/mobilenetv2.pytorch

72.8% MobileNetV2 1.0 model on ImageNet and a spectrum of pre-trained MobileNetV2 models

deep-neural-networks imagenet mobilenetv2 pretrained-models pytorch

Last synced: 07 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: 30 Oct 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/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/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/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/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