foundation-model-stack / fms-fsdp

🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash attention v2.
232Updated 2 weeks ago

Alternatives and similar repositories for fms-fsdp:

Users that are interested in fms-fsdp are comparing it to the libraries listed below