model-parallel topic

List model-parallel repositories

torchshard

295
Stars
15
Forks
Watchers

Slicing a PyTorch Tensor Into Parallel Shards

cavaface

455
Stars
88
Forks
Watchers

face recognition training project(pytorch)

PLSC

149
Stars
34
Forks
Watchers

Paddle Large Scale Classification Tools,supports ArcFace, CosFace, PartialFC, Data Parallel + Model Parallel. Model includes ResNet, ViT, Swin, DeiT, CaiT, FaceViT, MoCo, MAE, ConvMAE, CAE.

pytorch-model-parallel

76
Stars
19
Forks
Watchers

A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch

OpenEmbedding

28
Stars
8
Forks
Watchers

OpenEmbedding is an open source framework for Tensorflow distributed training acceleration.

hydra

20
Stars
3
Forks
Watchers

Execution framework for multi-task model parallelism. Enables the training of arbitrarily large models with a single GPU, with linear speedups for multi-gpu multi-task execution.

MPP-LLaVA

486
Stars
25
Forks
486
Watchers

Personal Project: MPP-Qwen14B & MPP-Qwen-Next(Multimodal Pipeline Parallel based on Qwen-LM). Support [video/image/multi-image] {sft/conversations}. Don't let the poverty limit your imagination! Train...