Contents

Adeepspeed 0.9.2

0

DeepSpeed library

DeepSpeed library

Stars: 41631, Watchers: 41631, Forks: 4723, Open Issues: 1270

The deepspeedai/DeepSpeed repo was created 6 years ago and the last code push was 20 hours ago.
The project is extremely popular with a mindblowing 41631 github stars!

How to Install adeepspeed

You can install adeepspeed using pip

pip install adeepspeed

or add it to a project with poetry

poetry add adeepspeed

Package Details

Author
DeepSpeed Team
License
MIT
Homepage
http://deepspeed.ai
PyPi:
https://pypi.org/project/Adeepspeed/
Documentation:
https://deepspeed.readthedocs.io
GitHub Repo:
https://github.com/microsoft/DeepSpeed

Classifiers

No  adeepspeed  pypi packages just yet.

Errors

A list of common adeepspeed errors.

Code Examples

Here are some adeepspeed code examples and snippets.

GitHub Issues

The adeepspeed package has 1270 open issues on GitHub

  • [BUG] ZeRO-3: zero.GatheredParameters([multiple params], modifier_rank=None) + in-place slice touch triggers assert not param.ds_active_sub_modules in free_param()
  • [Bugfix] Resolve Rank index out of range during BWD when sp_size < world_size in Ulysses
  • fix: Ensure full gradient reduction for Muon with reduce_scatter
  • [BUG] Cross-partition parameters incorrectly updated when using ZeRO-1/ZeRO-2 with reduce_scatter=true and Muon optimizer
  • Support custom partitioning patterns for AutoTP
  • Enable shm_comm support for arm
  • [Draft] Muon Optimizer Support for ZeRO3
  • [BUG] ZenFlow Stage 3 with full_warm_up_rounds=0 fails due to missing complete_column_offset attribute
  • Fix bf16 dtype mismatch in ZeRO-3 with zero_quantized_weights
  • Fix Muon optimizer conflict with gradient clipping in ZeRO 1/2
  • [BUG] ZeRO-3 with zero_quantized_weights=true incorrectly casts bf16 inputs to fp16, causing BERT training failure
  • [REQUEST] Python types?
  • Fix: ZenFlow Adam integration for updated PyTorch backward flow (#7759)
  • [BUG][Deepcompile] OOM during DeepCompile pre-pass eager node-by-node profiling (FX Interpreter) due to decomposed cross_entropy materializing huge intermediates
  • Introduce all_reduce_hook to support gradient aggregation across replica groups.

See more issues on GitHub

Related Packages & Articles

deepspeed 0.18.6

DeepSpeed is a Python package developed by Microsoft that provides a deep learning optimization library designed to scale across multiple GPUs and servers. It is capable of training models with billions or even trillions of parameters, achieving excellent system throughput and efficiently scaling to thousands of GPUs.

DeepSpeed is particularly useful for training and inference of large language models, and it falls under the category of Machine Learning Frameworks and Libraries. It is designed to work with PyTorch and offers system innovations such as Zero Redundancy Optimizer (ZeRO), 3D parallelism, and model-parallelism to enable efficient training of large models.

fastai 2.8.7

fastai simplifies training fast and accurate neural nets using modern best practices

auto-gptq 0.7.1

An easy-to-use LLMs quantization package with user-friendly apis, based on GPTQ algorithm.

detectors 0.1.11

This is a comprehensive library for generalized Out-of-Distribution (OOD) detection research. It provides over 20 detection methods, evaluation pipelines, OOD datasets, and model architectures integrated with timm. Additionally, it offers fast OOD evaluation metrics, multi-layer detection methods, and pipelines for open set recognition and covariate drift detection.

skypilot 0.11.1

SkyPilot: Run AI on Any Infra — Unified, Faster, Cheaper.

flwr 1.26.1

Flower: A Friendly Federated AI Framework

deeplake 4.5.2

Deep Lake is a Database for AI powered by a unique storage format optimized for deep-learning and Large Language Model (LLM) based applications. It simplifies the deployment of enterprise-grade LLM-based products by offering storage for all data types (embeddings, audio, text, videos, images, pdfs, annotations, etc.), querying and vector search, data streaming while training models at scale, data versioning and lineage for all workloads, and integrations with popular tools such as LangChain, LlamaIndex, Weights & Biases, and many more.