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
bigscience-workshop
petals

🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading

7K 10K 606
hpcaitech
colossalai-nightly

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

2K 41K 5K
torchpipe
hami-core

Serving Inside Pytorch

642 171 13
kakaobrain
torchgpipe

A GPipe implementation in PyTorch

482 863 98
PaddlePaddle
fleet-x

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

458 479 165
topal-team
elf-pipeline

Automated Memory-Aware Pipeline Parallel Training of Neural Networks

370 2 0
hpcaitech
custom-colossalai

Making large AI models cheaper, faster and more accessible

256 41K 5K
PaddlePaddle
fleet-lightning

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

183 479 165
MarcoGarofalo94
torchslicer

Split learning framework for PyTorch — pipeline-parallel training across heterogeneous devices.

144 4 0
microsoft
adeepspeed

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

124 42K 5K
PaddlePaddle
paddle-fleet

Distributed Training Package Based on PaddlePaddle

67 479 165
bigscience-workshop
test-petals

Easy way to efficiently run 100B+ language models without high-end GPUs

55 10K 606
    • Data from PyPI, GitHub, ClickHouse, and BigQuery