Pinned Loading

1. foundation-model-stack foundation-model-stack Public

   🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.

   Python 194 68

2. fms-fsdp fms-fsdp Public

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

   Python 244 38

3. fms-hf-tuning fms-hf-tuning Public

   🚀 Collection of tuning recipes with HuggingFace SFTTrainer and PyTorch FSDP.

   Python 41 56

4. fms-acceleration fms-acceleration Public

   🚀 Collection of libraries used with fms-hf-tuning to accelerate fine-tuning and training of large models.

   Python 9 15

5. fms-guardrails-orchestrator fms-guardrails-orchestrator Public

   🚀 Guardrails orchestration server for application of various detections on text generation input and output.

   Rust 15 28

6. bamba bamba Public

   Train, tune, and infer Bamba model

   Python 119 14