fsdp_optimizers

supporting pytorch FSDP for optimizers

run cifar10.py for tests, tested with torch 2.5, arguments at top of file.

Name		Name	Last commit message	Last commit date
Latest commit History 21 Commits
fsdp_optimizers		fsdp_optimizers
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
build.sh		build.sh
cifar10.py		cifar10.py
setup.py		setup.py

Navigation Menu