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

A PyTorch-based knowledge distillation toolkit for natural language processing

bert distillation knowledge nlp pytorch

Last synced: 03 Nov 2024

https://github.com/airaria/textbrewer

A PyTorch-based knowledge distillation toolkit for natural language processing

bert distillation knowledge nlp pytorch

Last synced: 09 Oct 2024

https://github.com/chaofengc/iqa-pytorch

👁️ 🖼️ 🔥PyTorch Toolbox for Image Quality Assessment, including LPIPS, FID, NIQE, NRQM(Ma), MUSIQ, TOPIQ, NIMA, DBCNN, BRISQUE, PI and more...

blind-image-quality-assessment image-aesthetic-assessment image-quality-assessment iqa python pytorch pytorch-implementation

Last synced: 15 Oct 2024

https://github.com/rwightman/efficientdet-pytorch

A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights

efficientdet efficientnet object-detection pytorch semantic-segmentation

Last synced: 21 Oct 2024

https://github.com/D-X-Y/AutoDL-Projects

Automated deep learning algorithms implemented in PyTorch.

autodl automl nas neural-architecture-search pytorch

Last synced: 09 Nov 2024

https://github.com/rwightman/gen-efficientnet-pytorch

Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS

caffe2 efficientnet fbnet mnasnet mobilenetv3 onnx pretrained-models pytorch

Last synced: 22 Oct 2024

https://github.com/kan-bayashi/ParallelWaveGAN

Unofficial Parallel WaveGAN (+ MelGAN & Multi-band MelGAN & HiFi-GAN & StyleMelGAN) with Pytorch

hifigan melgan neural-vocoder parallel-wavenet pytorch realtime speech-synthesis style-melgan text-to-speech tts vocoder wavenet

Last synced: 06 Nov 2024

https://github.com/tsurumeso/vocal-remover

Vocal Remover using Deep Neural Networks

audio deep-learning pytorch segmentation spectrogram vocal-separation

Last synced: 15 Oct 2024

https://github.com/d-x-y/autodl-projects

Automated deep learning algorithms implemented in PyTorch.

autodl automl nas neural-architecture-search pytorch

Last synced: 15 Oct 2024

https://github.com/longcw/yolo2-pytorch

YOLOv2 in PyTorch

darknet detection pytorch yolo yolo2

Last synced: 14 Oct 2024

https://github.com/kan-bayashi/parallelwavegan

Unofficial Parallel WaveGAN (+ MelGAN & Multi-band MelGAN & HiFi-GAN & StyleMelGAN) with Pytorch

hifigan melgan neural-vocoder parallel-wavenet pytorch realtime speech-synthesis style-melgan text-to-speech tts vocoder wavenet

Last synced: 11 Oct 2024

https://github.com/ELS-RD/kernl

Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackable.

cuda cuda-kernel pytorch transformer triton

Last synced: 12 Nov 2024

https://github.com/ssbuild/chatglm_finetuning

chatglm 6b finetuning and alpaca finetuning

adalora chatglm deep-learning freeze ia3 lora p-tuning-v2 pytorch qlora sft

Last synced: 15 Oct 2024

https://github.com/els-rd/kernl

Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackable.

cuda cuda-kernel pytorch transformer triton

Last synced: 14 Oct 2024

https://github.com/OpenPPL/ppq

PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.

caffe cuda deep-learning neural-network onnx open-source pytorch quantization

Last synced: 28 Oct 2024

https://github.com/predibase/lorax

Multi-LoRA inference server that scales to 1000s of fine-tuned LLMs

fine-tuning gpt llama llm llm-inference llm-serving llmops lora model-serving pytorch transformers

Last synced: 10 Oct 2024

https://github.com/openppl-public/ppq

PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.

caffe cuda deep-learning neural-network onnx open-source pytorch quantization

Last synced: 05 Oct 2024

https://github.com/openppl/ppq

PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.

caffe cuda deep-learning neural-network onnx open-source pytorch quantization

Last synced: 15 Oct 2024

https://github.com/n00mkrad/flowframes

Flowframes Windows GUI for video interpolation using DAIN (NCNN) or RIFE (CUDA/NCNN)

dain gui ncnn pytorch rife video-interpolation

Last synced: 10 Nov 2024

https://github.com/gpleiss/efficient_densenet_pytorch

A memory-efficient implementation of DenseNets

deep-learning densenet pytorch

Last synced: 14 Oct 2024

https://github.com/lufficc/SSD

High quality, fast, modular reference implementation of SSD in PyTorch

computer-vision deep-learning object-detection pytorch ssd

Last synced: 28 Oct 2024

https://github.com/devnag/pytorch-generative-adversarial-networks

A very simple generative adversarial network (GAN) in PyTorch

deep-learning gan gans machine-learning pytorch

Last synced: 26 Oct 2024

https://github.com/lufficc/ssd

High quality, fast, modular reference implementation of SSD in PyTorch

computer-vision deep-learning object-detection pytorch ssd

Last synced: 14 Oct 2024

https://github.com/sniklaus/3d-ken-burns

an implementation of 3D Ken Burns Effect from a Single Image using PyTorch

cuda cupy deep-learning python pytorch

Last synced: 14 Oct 2024

https://github.com/sgrvinod/Deep-Tutorials-for-PyTorch

In-depth tutorials for implementing deep learning models on your own with PyTorch.

deep-learning-tutorial learning-pytorch neural-network-tutorials pytorch pytorch-tutorial pytorch-tutorials

Last synced: 12 Nov 2024

https://github.com/Eric-mingjie/rethinking-network-pruning

Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)

convolutional-neural-networks deep-learning network-pruning pytorch

Last synced: 12 Nov 2024

https://github.com/caogang/wgan-gp

A pytorch implementation of Paper "Improved Training of Wasserstein GANs"

pytorch wgan-gp

Last synced: 14 Oct 2024

https://github.com/huggingface/pytorch-openai-transformer-lm

🐥A PyTorch implementation of OpenAI's finetuned transformer language model with a script to import the weights pre-trained by OpenAI

language-model neural-networks openai pytorch transformer

Last synced: 29 Oct 2024

https://github.com/erikwijmans/pointnet2_pytorch

PyTorch implementation of Pointnet2/Pointnet++

hydra point-cloud pointnet2 pytorch pytorch-lightning

Last synced: 29 Oct 2024

https://github.com/eric-mingjie/rethinking-network-pruning

Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)

convolutional-neural-networks deep-learning network-pruning pytorch

Last synced: 29 Oct 2024

https://github.com/google-research/big_transfer

Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.

convolutional-neural-networks deep-learning imagenet jax pytorch tensorflow2 transfer-learning

Last synced: 14 Oct 2024

https://github.com/tjwei/ganotebooks

wgan, wgan2(improved, gp), infogan, and dcgan implementation in lasagne, keras, pytorch

cyclegan dcgan dragan infogan keras lasagne pix2pix pytorch tensorflow wgan wgan-gp

Last synced: 10 Oct 2024

https://github.com/tjwei/GANotebooks

wgan, wgan2(improved, gp), infogan, and dcgan implementation in lasagne, keras, pytorch

cyclegan dcgan dragan infogan keras lasagne pix2pix pytorch tensorflow wgan wgan-gp

Last synced: 07 Aug 2024

https://github.com/sgrvinod/deep-tutorials-for-pytorch

In-depth tutorials for implementing deep learning models on your own with PyTorch.

deep-learning-tutorial learning-pytorch neural-network-tutorials pytorch pytorch-tutorial pytorch-tutorials

Last synced: 14 Oct 2024

https://github.com/xinntao/EDVR

Winning Solution in NTIRE19 Challenges on Video Restoration and Enhancement (CVPR19 Workshops) - Video Restoration with Enhanced Deformable Convolutional Networks. EDVR has been merged into BasicSR and this repo is a mirror of BasicSR.

basicsr edvr pytorch

Last synced: 03 Nov 2024

https://github.com/xinntao/edvr

Winning Solution in NTIRE19 Challenges on Video Restoration and Enhancement (CVPR19 Workshops) - Video Restoration with Enhanced Deformable Convolutional Networks. EDVR has been merged into BasicSR and this repo is a mirror of BasicSR.

basicsr edvr pytorch

Last synced: 29 Oct 2024

https://github.com/IBM/pytorch-seq2seq

An open source framework for seq2seq models in PyTorch.

deeplearning pytorch seq2seq

Last synced: 04 Nov 2024

https://github.com/nvidia/transformerengine

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.

cuda deep-learning fp8 gpu jax machine-learning python pytorch

Last synced: 15 Oct 2024

https://github.com/junyanz/bicyclegan

Toward Multimodal Image-to-Image Translation

deep-learning gans generative-adversarial-network pix2pix pytorch

Last synced: 14 Oct 2024

https://github.com/tencent/turbotransformers

a fast and user-friendly runtime for transformer inference (Bert, Albert, GPT2, Decoders, etc) on CPU and GPU.

albert bert decoder gpt2 gpu huggingface-transformers inference machine-translation nlp pytorch roberta transformer

Last synced: 09 Oct 2024

https://github.com/Tencent/TurboTransformers

a fast and user-friendly runtime for transformer inference (Bert, Albert, GPT2, Decoders, etc) on CPU and GPU.

albert bert decoder gpt2 gpu huggingface-transformers inference machine-translation nlp pytorch roberta transformer

Last synced: 27 Oct 2024

https://github.com/Swall0w/torchstat

Model analyzer in PyTorch

python pytorch

Last synced: 04 Nov 2024

https://github.com/junyanz/BicycleGAN

Toward Multimodal Image-to-Image Translation

deep-learning gans generative-adversarial-network pix2pix pytorch

Last synced: 03 Aug 2024

https://github.com/swall0w/torchstat

Model analyzer in PyTorch

python pytorch

Last synced: 10 Oct 2024

https://github.com/rusty1s/pytorch_scatter

PyTorch Extension Library of Optimized Scatter Operations

gather pytorch scatter segment

Last synced: 15 Oct 2024

https://github.com/erikwijmans/Pointnet2_PyTorch

PyTorch implementation of Pointnet2/Pointnet++

hydra point-cloud pointnet2 pytorch pytorch-lightning

Last synced: 28 Oct 2024

https://github.com/marcoslucianops/DeepStream-Yolo

NVIDIA DeepStream SDK 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models

darknet deepstream nvidia nvidia-deepstream-sdk object-detection paddle paddlepaddle ppyoloe pytorch tensorrt ultralytics yolo yolor yolov3 yolov4 yolov5 yolov6 yolov7 yolov8 yolox

Last synced: 27 Oct 2024

https://github.com/marcoslucianops/deepstream-yolo

NVIDIA DeepStream SDK 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models

darknet deepstream nvidia nvidia-deepstream-sdk object-detection paddle paddlepaddle ppyoloe pytorch tensorrt ultralytics yolo yolor yolov3 yolov4 yolov5 yolov6 yolov7 yolov8 yolox

Last synced: 15 Oct 2024

https://github.com/V2AI/Det3D

World's first general purpose 3D object detection codebse.

3d-object-detection autonomous-driving deep-learning kitti nuscenes object-detection point-cloud pytorch

Last synced: 28 Oct 2024

https://github.com/v2ai/det3d

World's first general purpose 3D object detection codebse.

3d-object-detection autonomous-driving deep-learning kitti nuscenes object-detection point-cloud pytorch

Last synced: 15 Oct 2024

https://github.com/poodarchu/Det3D

World's first general purpose 3D object detection codebse.

3d-object-detection autonomous-driving deep-learning kitti nuscenes object-detection point-cloud pytorch

Last synced: 13 Aug 2024

https://github.com/nvidiagameworks/kaolin-wisp

NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).

3d-deep-learning nerf neuralfields pytorch sdf

Last synced: 15 Oct 2024

https://github.com/cszn/DnCNN

Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising (TIP, 2017)

image-denoising jpeg-deblocking keras-tensorflow matconvnet pytorch residual-learning super-resolution

Last synced: 03 Nov 2024

https://github.com/toandaominh1997/efficientdet.pytorch

Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch

coco computer-vision demo detection efficientdet-d0 efficientnet focalloss multibox nms object-detection pascal-voc pytorch

Last synced: 29 Oct 2024

https://github.com/szagoruyko/attention-transfer

Improving Convolutional Networks via Attention Transfer (ICLR 2017)

attention deep-learning knowledge-distillation pytorch

Last synced: 29 Oct 2024

https://github.com/DeepGraphLearning/torchdrug

A powerful and flexible machine learning platform for drug discovery

deep-learning drug-discovery graph-neural-networks pytorch

Last synced: 12 Nov 2024

https://github.com/cszn/dncnn

Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising (TIP, 2017)

image-denoising jpeg-deblocking keras-tensorflow matconvnet pytorch residual-learning super-resolution

Last synced: 29 Oct 2024

https://github.com/toandaominh1997/EfficientDet.Pytorch

Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch

coco computer-vision demo detection efficientdet-d0 efficientnet focalloss multibox nms object-detection pascal-voc pytorch

Last synced: 07 Aug 2024

https://github.com/NVIDIAGameWorks/kaolin-wisp

NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).

3d-deep-learning nerf neuralfields pytorch sdf

Last synced: 03 Aug 2024

https://github.com/keyu-tian/SparK

[ICLR'23 Spotlight🔥] The first successful BERT/MAE-style pretraining on any convolutional network; Pytorch impl. of "Designing BERT for Convolutional Networks: Sparse and Hierarchical Masked Modeling"

bert cnn convnet convolutional-neural-networks deep-learning iclr iclr2023 instance-segmentation mae mask-rcnn masked-autoencoder masked-image-modeling object-detection pre-trained-model pretrain pretraining pytorch self-supervised-learning sparse-convolution ssl

Last synced: 28 Oct 2024

https://github.com/deepgraphlearning/torchdrug

A powerful and flexible machine learning platform for drug discovery

deep-learning drug-discovery graph-neural-networks pytorch

Last synced: 15 Oct 2024

https://github.com/keyu-tian/spark

[ICLR'23 Spotlight🔥] The first successful BERT/MAE-style pretraining on any convolutional network; Pytorch impl. of "Designing BERT for Convolutional Networks: Sparse and Hierarchical Masked Modeling"

bert cnn convnet convolutional-neural-networks deep-learning iclr iclr2023 instance-segmentation mae mask-rcnn masked-autoencoder masked-image-modeling object-detection pre-trained-model pretrain pretraining pytorch self-supervised-learning sparse-convolution ssl

Last synced: 15 Oct 2024

https://github.com/speedinghzl/CCNet

CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).

cityscape pytorch scene-parsing segmentation self-attention semantic-segmentation

Last synced: 31 Oct 2024

https://github.com/filipradenovic/cnnimageretrieval-pytorch

CNN Image Retrieval in PyTorch: Training and evaluating CNNs for Image Retrieval in PyTorch

cnn convolutional-neural-networks image-retrieval python pytorch

Last synced: 14 Oct 2024

https://github.com/speedinghzl/ccnet

CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).

cityscape pytorch scene-parsing segmentation self-attention semantic-segmentation

Last synced: 29 Oct 2024

https://github.com/vturrisi/solo-learn

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning

barlow-twins byol contrastive-learning deepcluster dino mae masked-input-prediction moco nnclr nvidia-dali pytorch pytorch-lightning ressl self-supervised-learning simclr simsiam swav transformer-models vibcreg vicreg

Last synced: 15 Oct 2024

https://github.com/GOATmessi8/RFBNet

Receptive Field Block Net for Accurate and Fast Object Detection, ECCV 2018

detection mobilenet pytorch rfbnet

Last synced: 06 Nov 2024

https://github.com/nod-ai/shark-studio

SHARK Studio -- Web UI for SHARK+IREE High Performance Machine Learning Distribution

amd apple-silicon deep-learning machine-learning mlir nvidia pytorch

Last synced: 09 Oct 2024

https://github.com/deftruth/cuda-learn-notes

🎉 Modern CUDA Learn Notes with PyTorch: CUDA Cores, Tensor Cores, fp32/tf32, fp16/bf16, fp8/int8, flash_attn, rope, sgemm, hgemm, sgemv, warp/block reduce, elementwise, softmax, layernorm, rmsnorm.

block-reduce cuda cuda-programming elementwise flash-attention flash-attention-2 flash-attention-3 gemm gemv layernorm pytorch rmsnorm softmax triton warp-reduce

Last synced: 12 Nov 2024

https://github.com/goatmessi8/rfbnet

Receptive Field Block Net for Accurate and Fast Object Detection, ECCV 2018

detection mobilenet pytorch rfbnet

Last synced: 29 Oct 2024

https://github.com/breezedeus/pix2text

An Open-Source Python3 tool for recognizing layouts, tables, math formulas (LaTeX), and text in images, converting them into Markdown format. A free alternative to Mathpix, empowering seamless conversion of visual content into text-based representations. 80+ languages are supported.

image-to-markdown latex latex-pdf layout-analysis math-formula math-formula-recognition math-ocr mathpix ocr python pytorch table-ocr

Last synced: 11 Oct 2024

https://github.com/ymcui/Chinese-ELECTRA

Pre-trained Chinese ELECTRA(中文ELECTRA预训练模型)

bert chinese chinese-electra electra language-model nlp pre-trained-model pytorch tensorflow

Last synced: 07 Nov 2024

https://github.com/torchgan/torchgan

Research Framework for easy and efficient training of GANs based on Pytorch

computer-vision deep-learning gans generative-adversarial-networks generative-model machine-learning neural-networks python python3 pytorch

Last synced: 20 Oct 2024