pipeline-parallelism topic

List pipeline-parallelism repositories

DeepSpeed

33.2k
Stars
3.9k
Forks
290
Watchers

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

ColossalAI

40.1k
Stars
4.5k
Forks
393
Watchers

Making large AI models cheaper, faster and more accessible

PaddleFleetX

425
Stars
158
Forks
Watchers

飞桨大模型开发套件,提供大语言模型、跨模态大模型、生物计算大模型等领域的全流程开发工具链。

libai

377
Stars
55
Forks
Watchers

LiBai(李白): A Toolbox for Large-Scale Distributed Parallel Training

EasyParallelLibrary

252
Stars
49
Forks
Watchers

Easy Parallel Library (EPL) is a general and efficient deep learning framework for distributed model training.

torchgpipe

782
Stars
94
Forks
Watchers

A GPipe implementation in PyTorch

DAPPLE

67
Stars
14
Forks
Watchers

An Efficient Pipelined Data Parallel Approach for Training Large Model

pytorch-gpt-x

29
Stars
3
Forks
Watchers

Implementation of autoregressive language model using improved Transformer and DeepSpeed pipeline parallelism.

FTPipe

41
Stars
7
Forks
Watchers

FTPipe and related pipeline model parallelism research.

awesome-distributed-ml

166
Stars
18
Forks
Watchers

A curated list of awesome projects and papers for distributed training or inference