Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
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.
- GitHub: https://github.com/topics/pytorch
- Wikipedia: https://en.wikipedia.org/wiki/PyTorch
- Repo: https://github.com/pytorch/pytorch
- Created by: Facebook's AI Research lab (FAIR)
- Released: September 2016
- Related Topics: python, pytorch-tutorial,
- Last updated: 2024-07-29 13:55:27 UTC
- JSON Representation
https://github.com/wkentaro/pytorch-fcn
PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)
computer-vision convolutional-networks deep-learning fcn fcn8s pytorch semantic-segmentation
Last synced: 25 Sep 2024
https://github.com/clementchadebec/benchmark_VAE
Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)
benchmarking beta-vae comparison normalizing-flows pixel-cnn pytorch reproducibility reproducible-research vae vae-gan vae-implementation vae-pytorch variational-autoencoder vq-vae wasserstein-autoencoder
Last synced: 02 Aug 2024
https://github.com/thu-coai/CDial-GPT
A Large-scale Chinese Short-Text Conversation Dataset and Chinese pre-training dialog models
dialogue gpt gpt-2 lccc pytorch text-generation
Last synced: 03 Aug 2024
https://github.com/huggingface/transfer-learning-conv-ai
🦄 State-of-the-Art Conversational AI with Transfer Learning
chatbots deep-learning dialog gpt gpt-2 neural-networks nlp pytorch transfer-learning
Last synced: 24 Sep 2024
https://github.com/facebookresearch/theseus
A library for differentiable nonlinear optimization
bilevel-optimization computer-vision deep-learning differentiable-optimization embodied-ai gauss-newton implicit-differentiation levenberg-marquardt nonlinear-least-squares pytorch robotics
Last synced: 25 Sep 2024
https://github.com/Lightning-Universe/lightning-flash
Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains
classification deep-learning fiftyone icevision machine-learning object-detection open3d pytorch pytorch-lightning pytorch-video tabular-data tasks-flash torch-geometric
Last synced: 02 Aug 2024
https://github.com/lightning-universe/lightning-flash
Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains
classification deep-learning fiftyone icevision machine-learning object-detection open3d pytorch pytorch-lightning pytorch-video tabular-data tasks-flash torch-geometric
Last synced: 26 Sep 2024
https://github.com/alibaba/EasyCV
An all-in-one toolkit for computer vision
classification computer-vision object-detection pytorch self-supervised-learning transformers vision-transformer
Last synced: 30 Jul 2024
https://github.com/gram-ai/capsule-networks
A PyTorch implementation of the NIPS 2017 paper "Dynamic Routing Between Capsules".
capsule-networks machine-learning mnist pytorch
Last synced: 24 Sep 2024
https://shunsukesaito.github.io/PIFu/
This repository contains the code for the paper "PIFu: Pixel-Aligned Implicit Function for High-Resolution Clothed Human Digitization"
3d-printing 3d-reconstruction arvr computer-graphics fashion geometry-processing human iccv2019 pifu pytorch
Last synced: 03 Aug 2024
https://github.com/shunsukesaito/PIFu
This repository contains the code for the paper "PIFu: Pixel-Aligned Implicit Function for High-Resolution Clothed Human Digitization"
3d-printing 3d-reconstruction arvr computer-graphics fashion geometry-processing human iccv2019 pifu pytorch
Last synced: 01 Aug 2024
https://github.com/shunsukesaito/pifu
This repository contains the code for the paper "PIFu: Pixel-Aligned Implicit Function for High-Resolution Clothed Human Digitization"
3d-printing 3d-reconstruction arvr computer-graphics fashion geometry-processing human iccv2019 pifu pytorch
Last synced: 24 Sep 2024
https://github.com/svip-lab/impersonator
PyTorch implementation of our ICCV 2019 paper: Liquid Warping GAN: A Unified Framework for Human Motion Imitation, Appearance Transfer and Novel View Synthesis
Last synced: 24 Sep 2024
https://github.com/deepset-ai/farm
:house_with_garden: Fast & easy transfer learning for NLP. Harvesting language models for the industry. Focus on Question Answering.
bert deep-learning germanbert language-models ner nlp nlp-framework nlp-library pretrained-models pytorch question-answering roberta transfer-learning xlnet-pytorch
Last synced: 25 Sep 2024
https://github.com/deepset-ai/FARM
:house_with_garden: Fast & easy transfer learning for NLP. Harvesting language models for the industry. Focus on Question Answering.
bert deep-learning germanbert language-models ner nlp nlp-framework nlp-library pretrained-models pytorch question-answering roberta transfer-learning xlnet-pytorch
Last synced: 01 Aug 2024
https://github.com/unslothai/hyperlearn
2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.
data-analysis data-science deep-learning econometrics gpu machine-learning neural-network optimization python pytorch regression-models research scikit-learn statistics statsmodels tensor
Last synced: 24 Sep 2024
https://github.com/swz30/restormer
[CVPR 2022--Oral] Restormer: Efficient Transformer for High-Resolution Image Restoration. SOTA for motion deblurring, image deraining, denoising (Gaussian/real data), and defocus deblurring.
cvpr2022 defocus-deblurring efficient-transformers high-resolution image-deblurring image-deraining image-restoration low-level-vision motion-deblurring pytorch transformer
Last synced: 24 Sep 2024
https://github.com/ContinualAI/avalanche
Avalanche: an End-to-End Library for Continual Learning based on PyTorch.
benchmarks continual-learning continualai deep-learning evaluation framework library lifelong-learning metrics pytorch strategies training
Last synced: 02 Aug 2024
https://github.com/Project-MONAI/tutorials
MONAI Tutorials
jupyter-notebook monai monai-tutorials monai-workflows pytorch
Last synced: 02 Aug 2024
https://github.com/Djdefrag/QualityScaler
QualityScaler - image/video AI upscaler app
amd anime compression-artifact-reduction deep-learning directx-12 gui-application intel manga noise-reduction nvidia opencv python python3 pytorch pytorch-directml super-resolution video video-processing windows
Last synced: 31 Jul 2024
https://github.com/djdefrag/qualityscaler
QualityScaler - image/video AI upscaler app
amd anime compression-artifact-reduction deep-learning directx-12 gui-application intel manga noise-reduction nvidia opencv python python3 pytorch pytorch-directml super-resolution video video-processing windows
Last synced: 25 Sep 2024
https://github.com/thudm/cogdl
CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)
gnn-model graph-classification graph-embedding graph-neural-networks leaderboard link-prediction node-classification pytorch
Last synced: 24 Sep 2024
https://github.com/THUDM/cogdl
CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)
gnn-model graph-classification graph-embedding graph-neural-networks leaderboard link-prediction node-classification pytorch
Last synced: 01 Aug 2024
https://github.com/THUDM/CogDL
CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)
gnn-model graph-classification graph-embedding graph-neural-networks leaderboard link-prediction node-classification pytorch
Last synced: 02 Aug 2024
https://github.com/ZijunDeng/pytorch-semantic-segmentation
PyTorch for Semantic Segmentation
deep-learning fully-convolutional-networks pytorch semantic-segmentation
Last synced: 03 Aug 2024
https://github.com/wxinlong/solo
SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.
instance-segmentation object-detection pytorch solo solov2
Last synced: 24 Sep 2024
https://github.com/thudm/cogview
Text-to-Image generation. The repo for NeurIPS 2021 paper "CogView: Mastering Text-to-Image Generation via Transformers".
pretrained-models pytorch text-to-image transformers
Last synced: 24 Sep 2024
https://github.com/tugstugi/dl-colab-notebooks
Try out deep learning models online on Google Colab
cnn colab-notebook deep-learning deep-neural-networks google-colab google-colaboratory jupyter-notebook jupyter-notebooks pytorch tensorflow
Last synced: 24 Sep 2024
https://github.com/pytorch/opacus
Training PyTorch models with differential privacy
deep-learning differential-privacy machine-learning neural-network privacy-preserving-machine-learning pytorch
Last synced: 24 Sep 2024
https://github.com/argusswift/yolov4-pytorch
This is a pytorch repository of YOLOv4, attentive YOLOv4 and mobilenet YOLOv4 with PASCAL VOC and COCO
attention cbam mobilenetv2 mobilenetv3 object-detection pytorch senet yolov4
Last synced: 24 Sep 2024
https://github.com/argusswift/YOLOv4-pytorch
This is a pytorch repository of YOLOv4, attentive YOLOv4 and mobilenet YOLOv4 with PASCAL VOC and COCO
attention cbam mobilenetv2 mobilenetv3 object-detection pytorch senet yolov4
Last synced: 02 Aug 2024
https://github.com/bearpaw/pytorch-classification
Classification with PyTorch.
cifar10 cifar100 classification densenet imagenet preresnet pytorch resnet resnext wide-residual-networks wrn
Last synced: 24 Sep 2024
https://github.com/yxlllc/DDSP-SVC
Real-time end-to-end singing voice conversion system based on DDSP (Differentiable Digital Signal Processing)
Last synced: 01 Aug 2024
https://github.com/median-research-group/libmtl
A PyTorch Library for Multi-Task Learning
deep-learning mmoe mtl multi-domain-learning multi-objective-optimization multi-task-learning multiobjective-optimization multitask-learning ple python pytorch
Last synced: 24 Sep 2024
https://github.com/median-research-group/LibMTL
A PyTorch Library for Multi-Task Learning
deep-learning mmoe mtl multi-domain-learning multi-objective-optimization multi-task-learning multiobjective-optimization multitask-learning ple python pytorch
Last synced: 31 Jul 2024
https://github.com/graph4ai/graph4nlp
Graph4nlp is the library for the easy use of Graph Neural Networks for NLP. Welcome to visit our DLG4NLP website (https://dlg4nlp.github.io/index.html) for various learning resources!
deep-learning graph-neural-networks machine-learning natural-language-processing nlp pytorch
Last synced: 25 Sep 2024
https://github.com/pytorch/tnt
A lightweight library for PyTorch training tools and utilities
deep-learning machine-learning neural-network python pytorch reinforcement-learning
Last synced: 24 Sep 2024
https://github.com/lightning-universe/lightning-bolts
Toolbox of models, callbacks, and datasets for AI/ML researchers.
ai gan image-processing machine-learning natural-language-processing pytorch supervised-learning
Last synced: 25 Sep 2024
https://github.com/Lightning-Universe/lightning-bolts
Toolbox of models, callbacks, and datasets for AI/ML researchers.
ai gan image-processing machine-learning natural-language-processing pytorch supervised-learning
Last synced: 01 Aug 2024
https://github.com/ai-forever/ru-dalle
Generate images from texts. In Russian
dalle image-generation openai python pytorch russian russian-language text-to-image transformer
Last synced: 25 Sep 2024
https://github.com/huggingface/huggingface_hub
The official Python client for the Huggingface Hub.
deep-learning hacktoberfest machine-learning model-hub models natural-language-processing pretrained-models pytorch
Last synced: 24 Sep 2024
https://github.com/ymcui/Chinese-XLNet
Pre-Trained Chinese XLNet(中文XLNet预训练模型)
natural-language-processing nlp pytorch tensorflow xlnet
Last synced: 31 Jul 2024
https://github.com/nikhilbarhate99/ppo-pytorch
Minimal implementation of clipped objective Proximal Policy Optimization (PPO) in PyTorch
deep-learning deep-reinforcement-learning policy-gradient ppo ppo-pytorch proximal-policy-optimization pytorch pytorch-implmention pytorch-tutorial reinforcement-learning reinforcement-learning-algorithms
Last synced: 24 Sep 2024
https://github.com/vita-group/transgan
[NeurIPS‘2021] "TransGAN: Two Pure Transformers Can Make One Strong GAN, and That Can Scale Up", Yifan Jiang, Shiyu Chang, Zhangyang Wang
gan pytorch transformer transformer-encoder transformer-models
Last synced: 24 Sep 2024
https://github.com/hkchengrex/XMem
[ECCV 2022] XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Model
computer-vision deep-learning eccv-2022 eccv2022 pytorch segmentation video-object-segmentation video-segmentation
Last synced: 01 Aug 2024
https://github.com/VITA-Group/TransGAN
[NeurIPS‘2021] "TransGAN: Two Pure Transformers Can Make One Strong GAN, and That Can Scale Up", Yifan Jiang, Shiyu Chang, Zhangyang Wang
gan pytorch transformer transformer-encoder transformer-models
Last synced: 03 Aug 2024
https://github.com/nvlabs/pwc-net
PWC-Net: CNNs for Optical Flow Using Pyramid, Warping, and Cost Volume, CVPR 2018 (Oral)
caffe computer-vision cvpr2018 deeplearning optical-flow pwc-net pytorch
Last synced: 25 Sep 2024
https://github.com/vahidk/EffectivePyTorch
PyTorch tutorials and best practices.
deep-learning ebook machine-learning neural-network pytorch
Last synced: 31 Jul 2024
https://github.com/THUDM/CogView
Text-to-Image generation. The repo for NeurIPS 2021 paper "CogView: Mastering Text-to-Image Generation via Transformers".
pretrained-models pytorch text-to-image transformers
Last synced: 01 Aug 2024
https://github.com/swz30/Restormer
[CVPR 2022--Oral] Restormer: Efficient Transformer for High-Resolution Image Restoration. SOTA for motion deblurring, image deraining, denoising (Gaussian/real data), and defocus deblurring.
cvpr2022 defocus-deblurring efficient-transformers high-resolution image-deblurring image-deraining image-restoration low-level-vision motion-deblurring pytorch transformer
Last synced: 31 Jul 2024
https://github.com/black0017/MedicalZooPytorch
A pytorch-based deep learning framework for multi-modal 2D/3D medical image segmentation
3d-convolutional-network brats2018 brats2019 deep-learning densenet iseg iseg-challenge medical-image-processing medical-image-segmentation medical-imaging mrbrains18 pytorch resnet segmentation segmentation-models unet unet-image-segmentation
Last synced: 03 Aug 2024
https://github.com/dusty-nv/jetson-containers
Machine Learning Containers for NVIDIA Jetson and JetPack-L4T
containers docker dockerfiles jetson machine-learning numpy nvidia pandas pytorch ros-containers ros2-foxy scikit-learn tensorflow
Last synced: 24 Sep 2024
https://github.com/LoSealL/VideoSuperResolution
A collection of state-of-the-art video or single-image super-resolution architectures, reimplemented in tensorflow.
carn dbpn dncnn edsr frvsr ntire2019 pytorch rcan rdn srcnn srgan srmd super-resolution tensorflow vdsr vespcn vsr
Last synced: 31 Jul 2024
https://github.com/yuliangxiu/icon
[CVPR'22] ICON: Implicit Clothed humans Obtained from Normals
3d-reconstruction animation avatar-generator cloth-simulation computer-graphics computer-vision human-pose-estimation implicit-functions mesh-deformation metaverse normal-maps pifu pifuhd pose-estimation pytorch smpl smpl-body smpl-model smplx virtual-humans
Last synced: 24 Sep 2024
https://github.com/kakaobrain/fast-autoaugment
Official Implementation of 'Fast AutoAugment' in PyTorch.
augmentation automated-machine-learning automl cnn computer-vision convolutional-neural-networks deep-learning distributed image-classification pytorch
Last synced: 24 Sep 2024
https://github.com/ialhashim/densedepth
High Quality Monocular Depth Estimation via Transfer Learning
deep-learning deep-neural-networks depth-estimation encoder-decoder-model keras-tensorflow python pytorch transfer-learning
Last synced: 24 Sep 2024
https://github.com/microsoft/edgeml
This repository provides code for machine learning algorithms for edge devices developed at Microsoft Research India.
bonsai classifier cpp deep-learning edge-computing edge-devices edge-machine-learning emi-rnn fastgrnn fastrnn iot-device machine-learning machine-learning-algorithms microsoft microsoft-research protonn pytorch resource-constrained-ml sensor tensorflow
Last synced: 24 Sep 2024
https://github.com/airaria/TextBrewer
A PyTorch-based knowledge distillation toolkit for natural language processing
bert distillation knowledge nlp pytorch
Last synced: 01 Aug 2024
https://github.com/YuliangXiu/ICON
[CVPR'22] ICON: Implicit Clothed humans Obtained from Normals
3d-reconstruction animation avatar-generator cloth-simulation computer-graphics computer-vision human-pose-estimation implicit-functions mesh-deformation metaverse normal-maps pifu pifuhd pose-estimation pytorch smpl smpl-body smpl-model smplx virtual-humans
Last synced: 04 Aug 2024
https://github.com/ranahanocka/MeshCNN
Convolutional Neural Network for 3D meshes in PyTorch
3d computer-graphics convolutional-neural-networks geometric-deep-learning machine-learning mesh pytorch segmentation
Last synced: 31 Jul 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: 24 Sep 2024
https://github.com/Microsoft/EdgeML
This repository provides code for machine learning algorithms for edge devices developed at Microsoft Research India.
bonsai classifier cpp deep-learning edge-computing edge-devices edge-machine-learning emi-rnn fastgrnn fastrnn iot-device machine-learning machine-learning-algorithms microsoft microsoft-research protonn pytorch resource-constrained-ml sensor tensorflow
Last synced: 02 Aug 2024
https://github.com/microsoft/EdgeML
This repository provides code for machine learning algorithms for edge devices developed at Microsoft Research India.
bonsai classifier cpp deep-learning edge-computing edge-devices edge-machine-learning emi-rnn fastgrnn fastrnn iot-device machine-learning machine-learning-algorithms microsoft microsoft-research protonn pytorch resource-constrained-ml sensor tensorflow
Last synced: 01 Aug 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: 25 Sep 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: 02 Aug 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: 27 Sep 2024
https://github.com/tsurumeso/vocal-remover
Vocal Remover using Deep Neural Networks
audio deep-learning pytorch segmentation spectrogram vocal-separation
Last synced: 25 Sep 2024
https://github.com/nikhilbarhate99/PPO-PyTorch
Minimal implementation of clipped objective Proximal Policy Optimization (PPO) in PyTorch
deep-learning deep-reinforcement-learning policy-gradient ppo ppo-pytorch proximal-policy-optimization pytorch pytorch-implmention pytorch-tutorial reinforcement-learning reinforcement-learning-algorithms
Last synced: 02 Aug 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: 25 Sep 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: 27 Sep 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: 01 Aug 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: 24 Sep 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: 30 Jul 2024
https://github.com/gpleiss/efficient_densenet_pytorch
A memory-efficient implementation of DenseNets
deep-learning densenet pytorch
Last synced: 24 Sep 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: 24 Sep 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: 24 Sep 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: 24 Sep 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: 24 Sep 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: 26 Sep 2024
https://github.com/caogang/wgan-gp
A pytorch implementation of Paper "Improved Training of Wasserstein GANs"
Last synced: 24 Sep 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: 24 Sep 2024
https://github.com/tensorly/tensorly
TensorLy: Tensor Learning in Python.
cupy decomposition jax machine-learning mxnet numpy python pytorch regression tensor tensor-algebra tensor-decomposition tensor-factorization tensor-learning tensor-methods tensor-regression tensorflow tensorly
Last synced: 25 Sep 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: 02 Aug 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: 31 Jul 2024
https://github.com/google-research/torchsde
Differentiable SDE solvers with GPU support and efficient sensitivity analysis.
deep-learning deep-neural-networks differential-equations dynamical-systems neural-differential-equations pytorch stochastic-differential-equations stochastic-processes stochastic-volatility-models
Last synced: 24 Sep 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.
Last synced: 24 Sep 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: 25 Sep 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: 02 Aug 2024
https://github.com/IBM/pytorch-seq2seq
An open source framework for seq2seq models in PyTorch.
Last synced: 01 Aug 2024
https://github.com/BloodAxe/pytorch-toolbelt
PyTorch extensions for fast R&D prototyping and Kaggle farming
augmentation deep-learning focal-loss image-classification image-processing image-segmentation jaccard-loss kaggle machine-learning object-detection pipeline python pytorch segmentation test-time-augmentation tta
Last synced: 01 Aug 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: 25 Sep 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.
Last synced: 01 Aug 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: 31 Jul 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: 02 Aug 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/rusty1s/pytorch_scatter
PyTorch Extension Library of Optimized Scatter Operations
gather pytorch scatter segment
Last synced: 24 Sep 2024