PyPI Stats
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About
Home

Search Packages

Find Python packages by name, description, GitHub topic, or filter by metrics
deepspeedai
deepspeed

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

1.3M 42K 5K
hpcaitech
colossalai

Making large AI models cheaper, faster and more accessible

28K 41K 5K
hpcaitech
colossalai-nightly

An integrated large-scale model training system with efficient parallelization techniques

2K 41K 5K
tanyuqian
redco

NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference

1K 69 7
kakaobrain
torchgpipe

A GPipe implementation in PyTorch

482 863 98
PaddlePaddle
fleet-x

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

458 479 165
hpcaitech
custom-colossalai

Making large AI models cheaper, faster and more accessible

256 41K 5K
KaiyuYue
torchshard

Slicing a PyTorch Tensor Into Parallel Shards

211 300 15
PaddlePaddle
fleet-lightning

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

183 479 165
microsoft
adeepspeed

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

124 42K 5K
ryantd
veloce

Veloce: An instant distributed computing library based on Ray stack

88 17 0
PaddlePaddle
paddle-fleet

Distributed Training Package Based on PaddlePaddle

67 479 165
    • Data from PyPI, GitHub, ClickHouse, and BigQuery