Illustration drawn by RepoMasters from the repo’s ID and its language color. Credits
AlphaUncommon

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.

Owner
foundation-model-stack (organization)
Stars
Stars: 289
Forks
Forks: 47
Language
Python
License
Apache-2.0
Created
2024
Topics
distributed-trainingllmpytorch
Tier
Alpha (250 to 699 ★) What this tier means
View on GitHub

More repos by foundation-model-stack

All 4 repos by foundation-model-stack

Nearby in Python

All Python repos