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-12-30 00:17:09 UTC
- JSON Representation
https://github.com/harry24k/adversarial-attacks-pytorch
PyTorch implementation of adversarial attacks [torchattacks]
adversarial-attacks deep-learning pytorch
Last synced: 26 Dec 2024
https://github.com/rguthrie3/DeepLearningForNLPInPytorch
An IPython Notebook tutorial on deep learning for natural language processing, including structure prediction.
deep-learning lstm neural-network nlp pytorch tutorial
Last synced: 02 Nov 2024
https://github.com/IgorSusmelj/pytorch-styleguide
An unofficial styleguide and best practices summary for PyTorch
best-practices pytorch styleguide
Last synced: 12 Nov 2024
https://github.com/pytorch/torchrec
Pytorch domain library for recommendation systems
cuda deep-learning gpu pytorch recommendation-system recommender-system sharding
Last synced: 24 Dec 2024
https://github.com/open-mmlab/mmgeneration
MMGeneration is a powerful toolkit for generative models, based on PyTorch and MMCV.
diffusion-models gan generative generative-adversarial-network mmcv openmmlab pytorch
Last synced: 28 Dec 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: 24 Dec 2024
https://github.com/microsoft/deepspeed-mii
MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.
deep-learning inference pytorch
Last synced: 24 Dec 2024
https://github.com/project-monai/tutorials
MONAI Tutorials
jupyter-notebook monai monai-tutorials monai-workflows pytorch
Last synced: 25 Dec 2024
https://github.com/po-hsun-su/pytorch-ssim
pytorch structural similarity (SSIM) loss
image-analysis image-processing pytorch
Last synced: 26 Dec 2024
https://github.com/open-mmlab/mmrotate
OpenMMLab Rotated Object Detection Toolbox and Benchmark
detection openmmlab pytorch rotated-object
Last synced: 24 Dec 2024
https://github.com/rlworkgroup/garage
A toolkit for reproducible reinforcement learning research.
pytorch reproducibility rl-algorithms tensorflow
Last synced: 26 Dec 2024
https://github.com/nobody132/masr
中文语音识别; Mandarin Automatic Speech Recognition;
chinese-speech-recognition mandarin-chinese pytorch speech-recognition
Last synced: 26 Dec 2024
https://github.com/isl-org/open3d-ml
An extension of Open3D to address 3D Machine Learning tasks
3d-object-detection 3d-perception datasets lidar object-detection pretrained-models pytorch rgbd semantic-segmentation tensorflow visualization
Last synced: 26 Dec 2024
https://github.com/jiesutd/ncrfpp
NCRF++, a Neural Sequence Labeling Toolkit. Easy use to any sequence labeling tasks (e.g. NER, POS, Segmentation). It includes character LSTM/CNN, word LSTM/CNN and softmax/CRF components.
artificial-intelligence char-cnn char-rnn chunking cnn crf lstm lstm-crf named-entity-recognition natural-language-processing nbest ner neural-networks part-of-speech-tagger pytorch sequence-labeling
Last synced: 28 Dec 2024
https://github.com/curt-park/rainbow-is-all-you-need
Rainbow is all you need! A step-by-step tutorial from DQN to Rainbow
colab-notebook dqn gym-environment nbviewer pytorch rainbow reinforcement-learning
Last synced: 26 Dec 2024
https://github.com/jiesutd/NCRFpp
NCRF++, a Neural Sequence Labeling Toolkit. Easy use to any sequence labeling tasks (e.g. NER, POS, Segmentation). It includes character LSTM/CNN, word LSTM/CNN and softmax/CRF components.
artificial-intelligence char-cnn char-rnn chunking cnn crf lstm lstm-crf named-entity-recognition natural-language-processing nbest ner neural-networks part-of-speech-tagger pytorch sequence-labeling
Last synced: 06 Nov 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: 24 Dec 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: 09 Nov 2024
https://github.com/Harry24k/adversarial-attacks-pytorch
PyTorch implementation of adversarial attacks [torchattacks]
adversarial-attacks deep-learning pytorch
Last synced: 30 Oct 2024
https://github.com/adobe-research/custom-diffusion
Custom Diffusion: Multi-Concept Customization of Text-to-Image Diffusion (CVPR 2023)
computer-vision customization diffusion-models few-shot fine-tuning pytorch text-to-image-generation
Last synced: 28 Dec 2024
https://github.com/microsoft/DeepSpeed-MII
MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.
deep-learning inference pytorch
Last synced: 30 Oct 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: 26 Dec 2024
https://github.com/uber/orbit
A Python package for Bayesian forecasting with object-oriented design and probabilistic models under the hood.
arima bayesian bayesian-methods bayesian-statistics changepoint exponential-smoothing forecast forecasting machine-learning orbit probabilistic probabilistic-programming pyro pystan python pytorch regression regression-models stan time-series
Last synced: 24 Dec 2024
https://github.com/javierantoran/bayesian-neural-networks
Pytorch implementations of Bayes By Backprop, MC Dropout, SGLD, the Local Reparametrization Trick, KF-Laplace, SG-HMC and more
approximate-inference bayes-by-backprop bayesian-inference bayesian-neural-networks classification deep-learning hmc kronecker-factored-approximation langevin-dynamics local-reparametrization-trick mc-dropout mcmc out-of-distribution-detection pytorch regression reproducible-research sgld uncertainty uncertainty-neural-networks variational-inference
Last synced: 28 Dec 2024
https://github.com/open-mmlab/mmaction
An open-source toolbox for action understanding based on PyTorch
action-detection action-recognition pytorch spatial-temporal-action-detection temporal-action-detection temporal-action-localization video-understanding
Last synced: 28 Dec 2024
https://github.com/nitrain/nitrain
Train AI models efficiently on medical images using any framework
deep-learning keras medical-imaging neuroimaging pytorch
Last synced: 26 Dec 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: 26 Dec 2024
https://github.com/haitongli/knowledge-distillation-pytorch
A PyTorch implementation for exploring deep and shallow knowledge distillation (KD) experiments with flexibility
cifar10 computer-vision dark-knowledge deep-neural-networks knowledge-distillation model-compression pytorch
Last synced: 28 Dec 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: 24 Dec 2024
https://github.com/Project-MONAI/tutorials
MONAI Tutorials
jupyter-notebook monai monai-tutorials monai-workflows pytorch
Last synced: 12 Nov 2024
https://github.com/isl-org/Open3D-ML
An extension of Open3D to address 3D Machine Learning tasks
3d-object-detection 3d-perception datasets lidar object-detection pretrained-models pytorch rgbd semantic-segmentation tensorflow visualization
Last synced: 28 Oct 2024
https://github.com/JavierAntoran/Bayesian-Neural-Networks
Pytorch implementations of Bayes By Backprop, MC Dropout, SGLD, the Local Reparametrization Trick, KF-Laplace, SG-HMC and more
approximate-inference bayes-by-backprop bayesian-inference bayesian-neural-networks classification deep-learning hmc kronecker-factored-approximation langevin-dynamics local-reparametrization-trick mc-dropout mcmc out-of-distribution-detection pytorch regression reproducible-research sgld uncertainty uncertainty-neural-networks variational-inference
Last synced: 12 Nov 2024
https://github.com/sierkinhane/crnn_chinese_characters_rec
(CRNN) Chinese Characters Recognition.
deep-learning ocr pytorch recognition
Last synced: 25 Dec 2024
https://github.com/rentruewang/koila
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.
deep-learning gradient-accumulation lazy-evaluation machine-learning memory-management neural-network out-of-memory python pytorch
Last synced: 12 Nov 2024
https://github.com/danielhanchen/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: 23 Nov 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: 11 Nov 2024
https://github.com/apple/ml-cvnets
CVNets: A library for training computer vision networks
ade20k classification computer-vision deep-learning detection imagenet machine-learning mscoco pascal-voc pytorch segmentation
Last synced: 28 Dec 2024
https://github.com/uber/petastorm
Petastorm library enables single machine or distributed training and evaluation of deep learning models from datasets in Apache Parquet format. It supports ML frameworks such as Tensorflow, Pytorch, and PySpark and can be used from pure Python code.
deep-learning machine-learning parquet parquet-files pyarrow pyspark pytorch sysml tensorflow
Last synced: 25 Dec 2024
https://github.com/waleedka/hiddenlayer
Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.
deeplearning keras pytorch tensorboard tensorflow visualization
Last synced: 26 Dec 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: 24 Dec 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: 24 Dec 2024
https://github.com/davda54/sam
SAM: Sharpness-Aware Minimization (PyTorch)
optimizer pytorch sam sharpness-aware
Last synced: 28 Dec 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: 24 Dec 2024
https://github.com/Curt-Park/rainbow-is-all-you-need
Rainbow is all you need! A step-by-step tutorial from DQN to Rainbow
colab-notebook dqn gym-environment nbviewer pytorch rainbow reinforcement-learning
Last synced: 31 Oct 2024
https://github.com/yang-song/score_sde_pytorch
PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
controllable-generation diffusion-models generative-models iclr-2021 inverse-problems pytorch score-based-generative-modeling score-matching stochastic-differential-equations
Last synced: 29 Dec 2024
https://github.com/harskish/ganspace
Discovering Interpretable GAN Controls [NeurIPS 2020]
deep-learning gan generative-adversarial-network image-generation interactive-visualizations pytorch
Last synced: 28 Dec 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: 26 Dec 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: 26 Dec 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: 26 Oct 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: 13 Nov 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: 08 Nov 2024
https://github.com/cfzd/Ultra-Fast-Lane-Detection
Ultra Fast Structure-aware Deep Lane Detection (ECCV 2020)
autonomous-driving autonomous-vehicles cnn lane-detection lane-detector lane-finding pytorch road-detection
Last synced: 28 Oct 2024
https://github.com/CSAILVision/gandissect
Pytorch-based tools for visualizing and understanding the neurons of a GAN. https://gandissect.csail.mit.edu/
deep-learning gan generative-adversarial-network image-manipulation interactive-visualizations interpretable-ml pytorch
Last synced: 12 Nov 2024
https://github.com/csailvision/gandissect
Pytorch-based tools for visualizing and understanding the neurons of a GAN. https://gandissect.csail.mit.edu/
deep-learning gan generative-adversarial-network image-manipulation interactive-visualizations interpretable-ml pytorch
Last synced: 28 Dec 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: 17 Nov 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: 28 Dec 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: 06 Nov 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: 09 Nov 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: 26 Dec 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: 28 Oct 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: 26 Dec 2024
https://github.com/siliconflow/onediff
OneDiff: An out-of-the-box acceleration library for diffusion models.
aigc-serving comfyui comfyui-workflow cuda diffusers diffusion-models inference-engine lcm lcm-lora lora performance-optimization pytorch sd-webui sdxl sdxl-turbo stable-diffusion stable-video-diffusion
Last synced: 24 Dec 2024
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/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: 28 Dec 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: 28 Dec 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: 26 Dec 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: 16 Nov 2024
https://github.com/IAHispano/Applio
A simple, high-quality voice conversion tool focused on ease of use and performance
ai applio pytorch rvc speech speech-to-speech text-to-speech tts vc vits voice voice-clone voice-cloning voice-conversion
Last synced: 14 Nov 2024
https://github.com/iahispano/applio
A simple, high-quality voice conversion tool focused on ease of use and performance
ai applio pytorch rvc speech speech-to-speech text-to-speech tts vc vits voice voice-clone voice-cloning voice-conversion
Last synced: 26 Dec 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: 29 Dec 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: 12 Nov 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: 15 Nov 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: 28 Dec 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: 10 Nov 2024
https://github.com/PyTorchLightning/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: 08 Nov 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/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: 28 Dec 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: 04 Nov 2024
https://github.com/longcw/faster_rcnn_pytorch
Faster RCNN with PyTorch
computer-vision detection faster-rcnn pytorch
Last synced: 28 Dec 2024
https://github.com/ZijunDeng/pytorch-semantic-segmentation
PyTorch for Semantic Segmentation
deep-learning fully-convolutional-networks pytorch semantic-segmentation
Last synced: 14 Nov 2024
https://github.com/pointcept/pointcept
Pointcept: a codebase for point cloud perception research. Latest works: PTv3 (CVPR'24 Oral), PPT (CVPR'24), OA-CNNs (CVPR'24), MSC (CVPR'23)
Last synced: 26 Dec 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: 29 Oct 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: 08 Nov 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: 11 Nov 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: 27 Dec 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: 07 Nov 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 Dec 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: 26 Dec 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: 06 Nov 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 Dec 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: 28 Dec 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: 09 Nov 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: 26 Dec 2024
https://github.com/vahidk/effectivepytorch
PyTorch tutorials and best practices.
deep-learning ebook machine-learning neural-network pytorch
Last synced: 29 Dec 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: 28 Oct 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 Dec 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: 28 Dec 2024
https://github.com/ymcui/chinese-xlnet
Pre-Trained Chinese XLNet(中文XLNet预训练模型)
natural-language-processing nlp pytorch tensorflow xlnet
Last synced: 28 Dec 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: 29 Dec 2024