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.6.0
Required dependencies:
einops
|
packaging
|
torch
Downloads last day:
529
Downloads last week:
9,338
Downloads last month:
51,289