Documentation#

FlagOS

A unified, open-source system software stack designed for a variety of AI chips

FlagOS Overview

FlagOS Core Libraries#

Operator Libraries

High-performance operator libraries optimized for diverse hardware backends.

Compiler

FlagTree

An open-source, unified compiler for multiple AI chips, advancing and expanding the Triton ecosystem across diverse hardware platforms.

Training & Inference Framework

FlagScale

A comprehensive toolkit designed to support the entire lifecycle of large models, from training to inference and deployment.

Communication Library

FlagCX

A scalable and adaptive unified communication library for cross-chip environments, delivering high-performance collective communication capabilities.


FlagOS Plugins for Diverse Chips#

vllm-plugin-FL

A plugin for the vLLM inference/serving framework, built on FlagOS’s unified multi-chip backend.

Megatron-LM-FL

A fork of Megatron-LM that introduces a plugin-based architecture for supporting diverse AI chips, built on top of FlagOS.

TransformerEngine-FL

A fork of TransformerEngine that introduces a plugin-based architecture for supporting diverse AI chips, built on top of FlagOS.

verl-FL

A fork of veRL that extends the upstream library with multi-chip/multi-hardware support via the FlagOS ecosystem.

PyTorch-Plugin-FL

A custom PyTorch device plugin based on the PrivateUse1 extension mechanism, registering FlagGems high-performance Triton operators as the flagos device backend.

sglang-plugin-FL

An out-of-tree (OOT) plugin for SGLang, built on FlagOS’s unified multi-chip backend, extending SGLang’s inference capabilities across diverse hardware platforms.


FlagOS Domain-Specific Projects#

FlagOS-Robo

An integrated training and inference framework for AI models used in robots, so-called Embodied Intelligence.

FlagQuantum

A high-performance distributed quantum statevector simulator built on PyTorch, enabling quantum circuit simulation across multiple GPUs.


FlagOS Developer Tools#

KernelGen

An operator auto-generation tool.

KernelGenBench

A benchmark framework for evaluating LLM and agent-based Triton kernel generation across multiple hardware platforms.

FlagOS Skills

Compatible with Claude Code, Cursor, Codex, and any agent supporting the Agent Skills standard.

https://github.com/flagos-ai/skills
Online Laboratory

An online laboratory providing cloud-based development environments.


FlagOS Platform Services#

FlagRelease

An automated platform for the cross-chip migration and release of open-source large models.

FlagPerf

An integrated AI hardware evaluation engine.

FlagCICD

A CI/CD toolchain that streamlines large-model development across diverse AI chips.

https://docs.flagos.io/projects/FlagCICD/zh-cn/latest/

Start to Use FlagOS

Join us to co-build an open AI chip development ecosystem

FlagOS Homepage