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/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: 31 Jul 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: 31 Jul 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 Jul 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: 02 Aug 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: 02 Aug 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: 30 Jul 2024

https://github.com/pytorch/torchrec

Pytorch domain library for recommendation systems

cuda deep-learning gpu pytorch recommendation-system recommender-system sharding

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/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: 01 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: 02 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: 31 Jul 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/apachecn/.github

ApacheCN 开源组织:公告、介绍、成员、活动、交流方式

dl ml python pytorch solidity spark

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

gan pose pytorch

Last synced: 02 Aug 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/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/gram-ai/capsule-networks

A PyTorch implementation of the NIPS 2017 paper "Dynamic Routing Between Capsules".

capsule-networks machine-learning mnist pytorch

Last synced: 31 Jul 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/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/yxlllc/DDSP-SVC

Real-time end-to-end singing voice conversion system based on DDSP (Differentiable Digital Signal Processing)

pytorch

Last synced: 01 Aug 2024

https://github.com/davda54/sam

SAM: Sharpness-Aware Minimization (PyTorch)

optimizer pytorch sam sharpness-aware

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

A lightweight library for PyTorch training tools and utilities

deep-learning machine-learning neural-network python pytorch reinforcement-learning

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

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

YOLOv2 in PyTorch

darknet detection pytorch yolo yolo2

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

Vocal Remover using Deep Neural Networks

audio deep-learning pytorch segmentation spectrogram vocal-separation

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

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

An open source framework for seq2seq models in PyTorch.

deeplearning pytorch seq2seq

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

https://github.com/erikwijmans/Pointnet2_PyTorch

PyTorch implementation of Pointnet2/Pointnet++

hydra point-cloud pointnet2 pytorch pytorch-lightning

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

https://github.com/Swall0w/torchstat

Model analyzer in PyTorch

python pytorch

Last synced: 01 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/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/szagoruyko/attention-transfer

Improving Convolutional Networks via Attention Transfer (ICLR 2017)

attention deep-learning knowledge-distillation pytorch

Last synced: 01 Aug 2024