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.1
Required dependencies:
einops
|
packaging
|
torch
Downloads last day:
1,060
Downloads last week:
6,506
Downloads last month:
34,691