Skip to content

@foundation-model-stack Foundation Model Stack

Change the repository type filter

All

    Repositories list

    29 repositories

    • aiu-fms-testing-utils

      Public
      Tests/Scripts for use with foundation-model-stack models running on AIU
      Python
      •
      Apache License 2.0
      •31•4•2•27•Updated Dec 23, 2025Dec 23, 2025

    • vllm-detector-adapter

      Public
      Tiny adapter that runs on top of vllm and provides detector APIs
      Python
      •
      Apache License 2.0
      •7•4•7•3•Updated Dec 23, 2025Dec 23, 2025

    • fms-model-optimizer

      Public
      FMS Model Optimizer is a framework for developing reduced precision neural network models.
      pythonmachine-learningcloudframeworkaioptimizerquantization
      Python
      •
      Apache License 2.0
      •17•20•16•7•Updated Dec 19, 2025Dec 19, 2025

    • fms-hf-tuning

      Public
      🚀 Collection of tuning recipes with HuggingFace SFTTrainer and PyTorch FSDP.
      Python
      •
      Apache License 2.0
      •65•54•24•19•Updated Dec 18, 2025Dec 18, 2025

    • vllm-triton-backend

      Public
      A Triton-only attention backend for vLLM
      Python
      •
      Apache License 2.0
      •7•23•0•2•Updated Dec 18, 2025Dec 18, 2025

    • fms-acceleration

      Public
      🚀 Collection of libraries used with fms-hf-tuning to accelerate fine-tuning and training of large models.
      Python
      •
      Apache License 2.0
      •19•13•27•2•Updated Dec 18, 2025Dec 18, 2025

    • fastsafetensors

      Public
      High-performance safetensors model loader
      pythonpytorchsafetensors
      Python
      •
      Apache License 2.0
      •16•87•1•0•Updated Dec 17, 2025Dec 17, 2025

    • multi-nic-cni

      Public
      Go
      •
      Apache License 2.0
      •8•40•8•0•Updated Dec 16, 2025Dec 16, 2025

    • fms-autoconf-k8s-controller

      Public
      Go
      •
      MIT License
      •0•0•0•0•Updated Dec 15, 2025Dec 15, 2025

    • tuning-config-recommender

      Public
      ⚡️⚡️ Supercharge your fine-tuning users with completely automated tuning configurations!
      Python
      •
      Apache License 2.0
      •2•4•3•0•Updated Dec 14, 2025Dec 14, 2025

    • foundation-model-stack

      Public
      🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
      Python
      •
      Apache License 2.0
      •99•216•47•60•Updated Dec 11, 2025Dec 11, 2025

    • fm-training-estimator

      Public
      Estimate resources needed to train LLMs
      Python
      •
      Apache License 2.0
      •9•13•3•0•Updated Dec 11, 2025Dec 11, 2025

    • fms-guardrails-orchestrator

      Public
      🚀 Guardrails orchestration server for application of various detections on text generation input and output.
      Rust
      •
      Apache License 2.0
      •37•27•29•2•Updated Dec 9, 2025Dec 9, 2025

    • fms-fsdp

      Public
      🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash attention v2.
      pytorchdistributed-trainingllm
      Python
      •
      Apache License 2.0
      •46•275•23•10•Updated Nov 24, 2025Nov 24, 2025

    • fms-dgt

      Public archive
      Synthetic Data Generation for Foundation Models
      Python
      •
      Apache License 2.0
      •24•21•2•1•Updated Nov 10, 2025Nov 10, 2025

    • ocp-efa-operator

      Public
      Operator that enables EFA and/or GDRCOPY in an OpenShift cluster
      Go
      •
      Apache License 2.0
      •0•1•0•0•Updated Jul 23, 2025Jul 23, 2025

    • bamba

      Public
      Train, tune, and infer Bamba model
      Python
      •
      Apache License 2.0
      •14•137•4•1•Updated Jun 4, 2025Jun 4, 2025

    • transformers

      Public
      🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
      Python
      •
      Apache License 2.0
      •32k•0•0•0•Updated Feb 7, 2025Feb 7, 2025

    • hf-resource-scanner

      Public
      Scan resources consumed during LLM training
      Python
      •
      Apache License 2.0
      •4•8•1•0•Updated Jan 14, 2025Jan 14, 2025

    • blog

      Public
      Public repo for HF blog posts
      Jupyter Notebook
      •958•0•0•1•Updated Dec 18, 2024Dec 18, 2024

    • moe-distributed-training

      Public
      Demonstration of MoE distributed training using various techniques
      Python
      •0•10•1•0•Updated Oct 31, 2024Oct 31, 2024

    • base-images

      Public
      Dockerfile
      •
      Apache License 2.0
      •4•3•1•0•Updated Oct 28, 2024Oct 28, 2024

    • pod-vllm

      Public
      Source code to launch a number of pods, performing synthetic data generation
      Python
      •
      Apache License 2.0
      •1•1•0•1•Updated Oct 22, 2024Oct 22, 2024

    • fms-lm-eval-service

      Public
      Go
      •
      Apache License 2.0
      •2•1•9•1•Updated Oct 10, 2024Oct 10, 2024

    • fms-extras

      Public
      Python
      •
      Apache License 2.0
      •11•24•3•5•Updated Sep 9, 2024Sep 9, 2024

    • avengers

      Public
      Shell
      •
      Apache License 2.0
      •0•0•4•0•Updated Jul 20, 2024Jul 20, 2024

    • trl

      Public
      Train transformer language models with reinforcement learning.
      Python
      •
      Apache License 2.0
      •2.4k•0•0•2•Updated Mar 5, 2024Mar 5, 2024

    • train-conductor

      Public
      Training job management tool for foundation model service
      Python
      •
      Apache License 2.0
      •4•5•6•0•Updated Feb 28, 2024Feb 28, 2024

    • training-operator

      Public
      Training operators on Kubernetes.
      Python
      •
      Apache License 2.0
      •855•0•0•0•Updated Nov 16, 2022Nov 16, 2022