PyPI page
Home page
Author:
None
License:
Apache 2.0
Summary:
**Megatron-FSDP** is an NVIDIA-developed PyTorch extension that provides a high-performance implementation of Fully Sharded Data Parallelism (FSDP)
Latest version:
0.5.0
Required dependencies:
einops
|
packaging
|
torch
Downloads last day:
2,870
Downloads last week:
13,734
Downloads last month:
68,290