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.

52
/ 100
Established

282 stars.

No Package No Dependents
Maintenance 6 / 25
Adoption 10 / 25
Maturity 16 / 25
Community 20 / 25

How are scores calculated?

Stars

282

Forks

46

Language

Python

License

Apache-2.0

Category

sparse-attention-optimization

Last pushed

Nov 24, 2025

Commits (30d)

0

GitHub
Sparse Attention Optimization · 20 models

Get this data via API

curl "https://pt-edge.onrender.com/api/v1/quality/transformers/foundation-model-stack/fms-fsdp"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

Related models

fla-org/flash-linear-attention

🚀 Efficient implementations of state-of-the-art linear attention models

89

thu-ml/SageAttention

[ICLR2025, ICML2025, NeurIPS2025 Spotlight] Quantized Attention achieves speedup of 2-5x...

57

thu-ml/SpargeAttn

[ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.

54

fla-org/flame

🔥 A minimal training framework for scaling FLA models

52

skylight-org/sparse-attention-hub

Advancing the frontier of efficient AI

46

Explore Transformer Models

All categories Trending Transformer directory Insights