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/jackroos/VL-BERT

Code for ICLR 2020 paper "VL-BERT: Pre-training of Generic Visual-Linguistic Representations".

bert iclr2020 pre-training pytorch representation-learning self-supervised-learning vision-and-language vl-bert

Last synced: 01 Aug 2024

https://github.com/williamFalcon/test-tube

Python library to easily log experiments and parallelize hyperparameter search for neural networks

caffe caffe2 chainer data-science deep-learning grid-search hyperparameter-optimization keras machine-learning neural-networks pytorch random-search tensorflow

Last synced: 31 Jul 2024

https://github.com/jacobgil/vit-explain

Explainability for Vision Transformers

deep-learning explainable-ai pytorch transformer vision-transformer

Last synced: 01 Aug 2024

https://github.com/msnh2012/Msnhnet

🔥 (yolov3 yolov4 yolov5 unet ...)A mini pytorch inference framework which inspired from darknet.

darknet inference-engine jetson-nx mobilenetv2 mobilenetyolo pytorch yolov3 yolov4 yolov5

Last synced: 02 Aug 2024

https://github.com/mosecorg/mosec

A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine

cv deep-learning gpu hacktoberfest jax llm llm-serving machine-learning machine-learning-platform mlops model-serving mxnet nerual-network python pytorch rust tensorflow tts

Last synced: 01 Aug 2024

https://github.com/tusen-ai/SST

Code for a series of work in LiDAR perception, including SST (CVPR 22), FSD (NeurIPS 22), FSD++ (TPAMI 23), FSDv2, and CTRL (ICCV 23, oral).

3d-object-detection autonomous-driving pytorch

Last synced: 31 Jul 2024

https://github.com/quic/sense

Enhance your application with the ability to see and interact with humans using any RGB camera.

activity-recognition calorie-estimation computer-vision deep-learning fitness-app gesture-recognition neural-networks pytorch video

Last synced: 05 Aug 2024

https://github.com/DrSleep/light-weight-refinenet

Light-Weight RefineNet for Real-Time Semantic Segmentation

deep-learning light-weight pytorch refinenet semantic-segmentation

Last synced: 04 Aug 2024

https://github.com/MisaOgura/flashtorch

Visualization toolkit for neural networks in PyTorch! Demo -->

cnn deep-learning explainability interpretability machine-learning neural-networks pytorch visualization

Last synced: 31 Jul 2024

https://github.com/luhua-rain/MRC_Competition_Dureader

机器阅读理解 冠军/亚军代码及中文预训练MRC模型

bert dureader mrc pytorch qa squad

Last synced: 01 Aug 2024

https://github.com/ritchieng/deep-learning-wizard

Open source guides/codes for mastering deep learning to deploying deep learning in production in PyTorch, Python, Apptainer, and more.

apptainer deep-learning llamaindex ollama python3 pytorch

Last synced: 31 Jul 2024

https://github.com/PaddlePaddle/X2Paddle

Deep learning model converter for PaddlePaddle. (『飞桨』深度学习模型转换工具)

caffe model-converter onnx paddlepaddle pytorch tensorflow x2paddle-model-zoo

Last synced: 01 Aug 2024

https://github.com/Denis2054/Transformers-for-NLP-2nd-Edition

Transformer models from BERT to GPT-4, environments from Hugging Face to OpenAI. Fine-tuning, training, and prompt engineering examples. A bonus section with ChatGPT, GPT-3.5-turbo, GPT-4, and DALL-E including jump starting GPT-4, speech-to-text, text-to-speech, text to image generation with DALL-E, Google Cloud AI,HuggingGPT, and more

bert chatgpt chatgpt-api dall-e dall-e-api deep-learning gpt-3-5-turbo gpt-4 gpt-4-api huggingface-transformers machine-learning natural-language-processing nlp openai python pytorch roberta-model transformers trax

Last synced: 29 Jul 2024

https://github.com/fboulnois/stable-diffusion-docker

Run the official Stable Diffusion releases in a Docker container with txt2img, img2img, depth2img, pix2pix, upscale4x, and inpaint.

dall-e dalle diffusion docker generative-art huggingface image-generation inpainting midjourney pytorch stable-diffusion tensorflow text-to-image

Last synced: 01 Aug 2024

https://github.com/creafz/pytorch-cnn-finetune

Fine-tune pretrained Convolutional Neural Networks with PyTorch

convolutional-neural-networks deep-learning pretrained-models pytorch

Last synced: 03 Aug 2024

https://github.com/cloneofsimo/minDiffusion

Self-contained, minimalistic implementation of diffusion models with Pytorch.

diffusion pytorch

Last synced: 31 Jul 2024

https://github.com/microsoft/CameraTraps

PyTorch Wildlife: a Collaborative Deep Learning Framework for Conservation.

camera-traps computer-vision conservation machine-learning megadetector pytorch pytorch-wildlife wildlife

Last synced: 08 Aug 2024

https://github.com/zhunzhong07/Random-Erasing

Random Erasing Data Augmentation. Experiments on CIFAR10, CIFAR100 and Fashion-MNIST

aaai2020 data-augmentation image-classification object-detection person-re-identification pytorch

Last synced: 02 Aug 2024

https://github.com/lzx1413/PytorchSSD

pytorch version of SSD and it's enhanced methods such as RFBSSD,FSSD and RefineDet

fssd pytorch refinedet rfb ssd

Last synced: 03 Aug 2024

https://github.com/run-house/runhouse

The fastest way to iterate and deploy AI workloads on your own infra. Unobtrusive, debuggable, PyTorch-like APIs.

api artificial-intelligence aws azure collaboration data-science deployment distributed fastapi gcp infrastructure machine-learning middleware observability python pytorch ray sagemaker serverless

Last synced: 31 Jul 2024

https://github.com/onepanelio/onepanel

The open source, end-to-end computer vision platform. Label, build, train, tune, deploy and automate in a unified platform that runs on any cloud and on-premises.

ai aiops annotation computer-vision deeplearning etl hyperparameter-tuning inference jupyterlab labeling machinelearning mlops pipelines pytorch tensorboard tensorflow training workflows

Last synced: 01 Aug 2024

https://github.com/zhiqwang/yolort

yolort is a runtime stack for yolov5 on specialized accelerators such as tensorrt, libtorch, onnxruntime, tvm and ncnn.

deployment detection graghsurgeon inference jit libtorch lightning ncnn nms onnx onnxruntime pytorch tensorrt torchscript trt tvm visualization yolo yolort yolov5

Last synced: 31 Jul 2024

https://github.com/SMILELab-FL/FedLab

A flexible Federated Learning Framework based on PyTorch, simplifying your Federated Learning research.

deep-learning federated-learning federated-learning-framework fedlab machine-learning pytorch pytorch-federated-learning

Last synced: 02 Aug 2024

https://github.com/lonePatient/albert_pytorch

A Lite Bert For Self-Supervised Learning Language Representations

albert bert language-model mask ngram nlp pytorch

Last synced: 01 Aug 2024

https://github.com/mosaicml/streaming

A Data Streaming Library for Efficient Neural Network Training

dataset deep-learning machine-learning neural-network pytorch streaming

Last synced: 01 Aug 2024

https://github.com/jia-zhuang/pytorch-multi-gpu-training

整理 pytorch 单机多 GPU 训练方法与原理

multi-gpu-training pytorch

Last synced: 01 Aug 2024

https://github.com/fidler-lab/polyrnn-pp-pytorch

PyTorch training/tool code for Polygon-RNN++ (CVPR 2018)

annotation cvpr2018 deep-learning instance-segmentation labelling polygon-rnn pytorch

Last synced: 07 Aug 2024

https://github.com/yl-1993/learn-to-cluster

Learning to Cluster Faces (CVPR 2019, CVPR 2020)

cvpr2019 cvpr2020 face-clustering face-recognition gcn pytorch

Last synced: 01 Aug 2024

https://github.com/stevenygd/PointFlow

PointFlow : 3D Point Cloud Generation with Continuous Normalizing Flows

3d-point-clouds computer-vision continuous-normalizing-flows machine-learning pytorch shapes

Last synced: 05 Aug 2024

https://github.com/Cadene/vqa.pytorch

Visual Question Answering in Pytorch

clevr coco deep-learning pytorch resnet skipthoughts torch vgenome vqa

Last synced: 01 Aug 2024

https://github.com/wohlert/semi-supervised-pytorch

Implementations of various VAE-based semi-supervised and generative models in PyTorch

generative-models pytorch semi-supervised-learning

Last synced: 03 Aug 2024

https://github.com/ShichenLiu/CondenseNet

CondenseNet: Light weighted CNN for mobile devices

deep-learning mobile-device pytorch

Last synced: 31 Jul 2024

https://github.com/jonasrauber/eagerpy

PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code

eager-execution jax numpy python pytorch tensorflow tensorflow2

Last synced: 02 Aug 2024

https://github.com/jayleicn/ClipBERT

[CVPR 2021 Best Student Paper Honorable Mention, Oral] Official PyTorch code for ClipBERT, an efficient framework for end-to-end learning on image-text and video-text tasks.

cvpr2021 pytorch video-question-answering video-retrieval vision-and-language vqa

Last synced: 03 Aug 2024

https://github.com/csteinmetz1/auraloss

Collection of audio-focused loss functions in PyTorch

audio loss-functions pytorch

Last synced: 05 Aug 2024

https://github.com/mhamilton723/STEGO

Unsupervised Semantic Segmentation by Distilling Feature Correspondences

computer-vision deep-learning iclr2022 pytorch semantic-segmentation unsupervised-learning

Last synced: 03 Aug 2024

https://github.com/weiaicunzai/Bag_of_Tricks_for_Image_Classification_with_Convolutional_Neural_Networks

experiments on Paper <Bag of Tricks for Image Classification with Convolutional Neural Networks> and other useful tricks to improve CNN acc

image-classification pytorch

Last synced: 01 Aug 2024

https://github.com/microsoft/tutel

Tutel MoE: An Optimized Mixture-of-Experts Implementation

mixture-of-experts moe nlp pytorch transformer

Last synced: 01 Aug 2024

https://github.com/antao97/dgcnn.pytorch

A PyTorch implementation of Dynamic Graph CNN for Learning on Point Clouds (DGCNN)

classification point-cloud pytorch segmentation

Last synced: 31 Jul 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: 01 Aug 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 Aug 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/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: 30 Jul 2024

https://github.com/open-mmlab/mmfewshot

OpenMMLab FewShot Learning Toolbox and Benchmark

few-shot-learning openmmlab pytorch

Last synced: 03 Aug 2024

https://github.com/wkentaro/pytorch-for-numpy-users

PyTorch for Numpy users. https://pytorch-for-numpy-users.wkentaro.com

numpy pytorch

Last synced: 02 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/thunlp/OpenAttack

An Open-Source Package for Textual Adversarial Attack.

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

Last synced: 01 Aug 2024

https://github.com/HumanCompatibleAI/overcooked_ai

A benchmark environment for fully cooperative human-AI performance.

artificial-intelligence deep-learning machine-learning pytorch reinforcement-learning

Last synced: 02 Aug 2024

https://github.com/shadow2496/VITON-HD

Official PyTorch implementation of "VITON-HD: High-Resolution Virtual Try-On via Misalignment-Aware Normalization" (CVPR 2021)

deep-learning pytorch

Last synced: 01 Aug 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/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: 02 Aug 2024

https://github.com/locuslab/qpth

A fast and differentiable QP solver for PyTorch.

deep-learning machine-learning optimization pytorch quadratic-programming

Last synced: 01 Aug 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: 31 Jul 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: 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: 31 Jul 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/msurtsukov/neural-ode

Jupyter notebook with Pytorch implementation of Neural Ordinary Differential Equations

jupyter-notebook neural-ode pytorch vae

Last synced: 01 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/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: 01 Aug 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: 01 Aug 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/iMoonLab/HGNN

Hypergraph Neural Networks (AAAI 2019)

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

Last synced: 31 Jul 2024

https://github.com/BlackHC/tfpyth

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

machine-learning pytorch tensorflow

Last synced: 31 Jul 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: 01 Aug 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/pytorch/torcharrow

High performance model preprocessing library on PyTorch

preprocessing python pytorch

Last synced: 02 Aug 2024

https://github.com/perone/euclidesdb

A multi-model machine learning feature embedding database

cpp database deep-learning machine-learning pytorch search

Last synced: 02 Aug 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/seungwonpark/melgan

MelGAN vocoder (compatible with NVIDIA/tacotron2)

gan neural-vocoder pytorch tts

Last synced: 31 Jul 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/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: 31 Jul 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: 01 Aug 2024

https://github.com/BobLiu20/YOLOv3_PyTorch

Full implementation of YOLOv3 in PyTorch

object-detection pytorch yolo yolov3

Last synced: 02 Aug 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: 31 Jul 2024