Skip to content
@foundation-model-stack

Foundation Model Stack

IBM Research's home for incubating, advancing and upstreaming open source for foundation model development, training, tuning and inference.

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 182 62

  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 215 33

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

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

    Python 30 48

  4. fms-acceleration fms-acceleration Public

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

    Python 7 12

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

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

    Rust 5 22

  6. bamba bamba Public

    Train, tune, and infer Bamba model

    Python 75 11

Repositories

Showing 10 of 24 repositories
  • fms-fsdp Public

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

    foundation-model-stack/fms-fsdp’s past year of commit activity
    Python 215 Apache-2.0 33 15 7 Updated Jan 14, 2025
  • vllm-detector-adapter Public

    Tiny adapter that runs on top of vllm and provides detector APIs

    foundation-model-stack/vllm-detector-adapter’s past year of commit activity
    Python 2 Apache-2.0 3 3 0 Updated Jan 14, 2025
  • fms-acceleration Public

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

    foundation-model-stack/fms-acceleration’s past year of commit activity
    Python 7 Apache-2.0 12 20 (10 issues need help) 1 Updated Jan 14, 2025
  • hf-resource-scanner Public

    Scan resources consumed during LLM training

    foundation-model-stack/hf-resource-scanner’s past year of commit activity
    Python 8 Apache-2.0 1 1 0 Updated Jan 14, 2025
  • fms-model-optimizer Public

    FMS Model Optimizer is a framework for developing reduced precision neural network models.

    foundation-model-stack/fms-model-optimizer’s past year of commit activity
    Python 12 Apache-2.0 6 7 (2 issues need help) 3 Updated Jan 13, 2025
  • fms-guardrails-orchestrator Public

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

    foundation-model-stack/fms-guardrails-orchestrator’s past year of commit activity
    Rust 5 Apache-2.0 22 29 (3 issues need help) 3 Updated Jan 13, 2025
  • bamba Public

    Train, tune, and infer Bamba model

    foundation-model-stack/bamba’s past year of commit activity
    Python 75 Apache-2.0 11 4 0 Updated Jan 13, 2025
  • fm-training-estimator Public

    Estimate resources needed to train LLMs

    foundation-model-stack/fm-training-estimator’s past year of commit activity
    Python 12 Apache-2.0 7 2 0 Updated Jan 13, 2025
  • fms-hf-tuning Public

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

    foundation-model-stack/fms-hf-tuning’s past year of commit activity
    Python 30 Apache-2.0 48 23 (2 issues need help) 25 Updated Jan 9, 2025
  • foundation-model-stack Public

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

    foundation-model-stack/foundation-model-stack’s past year of commit activity
    Python 182 Apache-2.0 62 35 (4 issues need help) 41 Updated Jan 9, 2025
View all repositories

Most used topics

python pytorch