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.3.1
Required dependencies:
einops
|
packaging
|
torch
Downloads last day:
1,890
Downloads last week:
9,591
Downloads last month:
63,825