README.md
pyproject.toml
megatron_fsdp/__init__.py
megatron_fsdp/distributed_data_parallel_config.py
megatron_fsdp/fully_shard.py
megatron_fsdp/megatron_fsdp.py
megatron_fsdp/mixed_precision.py
megatron_fsdp/package_info.py
megatron_fsdp/param_and_grad_buffer.py
megatron_fsdp/uneven_dtensor.py
megatron_fsdp/utils.py
megatron_fsdp.egg-info/PKG-INFO
megatron_fsdp.egg-info/SOURCES.txt
megatron_fsdp.egg-info/dependency_links.txt
megatron_fsdp.egg-info/requires.txt
megatron_fsdp.egg-info/top_level.txt
megatron_fsdp/experimental/__init__.py
megatron_fsdp/experimental/dbuffer.py
megatron_fsdp/experimental/fully_shard.py
megatron_fsdp/experimental/indexed_order.py
megatron_fsdp/experimental/layout.py
megatron_fsdp/experimental/module.py
megatron_fsdp/experimental/optimizer.py
megatron_fsdp/experimental/parameter_group.py
megatron_fsdp/experimental/placement.py