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.1
Required dependencies:
einops
|
packaging
|
torch
Downloads last day:
2,851
Downloads last week:
16,360
Downloads last month:
72,392