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 187 63

  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 221 35

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

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

    Python 34 49

  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 13

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

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

    Rust 6 23

  6. bamba bamba Public

    Train, tune, and infer Bamba model

    Python 83 13

Repositories

Showing 10 of 25 repositories
  • 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 6 Apache-2.0 23 31 (2 issues need help) 5 Updated Feb 12, 2025
  • 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 221 Apache-2.0 35 18 8 Updated Feb 12, 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 13 20 (11 issues need help) 2 Updated Feb 12, 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 34 Apache-2.0 49 27 (2 issues need help) 28 Updated Feb 12, 2025
  • multi-nic-cni Public
    foundation-model-stack/multi-nic-cni’s past year of commit activity
    Go 37 Apache-2.0 7 22 (1 issue needs help) 5 Updated Feb 12, 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 187 Apache-2.0 63 36 (4 issues need help) 40 Updated Feb 11, 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 8 3 1 Updated Feb 11, 2025
  • transformers Public Forked from mayank31398/transformers

    🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

    foundation-model-stack/transformers’s past year of commit activity
    Python 0 Apache-2.0 28,353 0 0 Updated Feb 7, 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 3 Apache-2.0 4 4 1 Updated Feb 7, 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 13 Apache-2.0 9 7 (2 issues need help) 3 Updated Jan 31, 2025