fix(phyai): preserve HumanWAM-Goal inference precision (#94)
fix(phyai): preserve HumanWAM-Goal precision and support Torch 2.10
fix(phyai): restore upstream dependency configuration
Co-authored-by: yinyuanzhang 348179793+yinyuanzhang-ssi@users.noreply.github.com
版权所有:中国计算机学会技术支持:开源发展技术委员会
京ICP备13000930号-9
京公网安备 11010802047560号
PhyAI (pronounced “phi”) is a latency-first serving engine for Physical AI. It is designed first for latency critical workloads, such as policy and action models that run in interactive systems.
https://github.com/user-attachments/assets/bcfe5c0f-1984-4410-a6a9-be20c7895e6f
News
Key Features
Supported Models
All supported models can scale through data-parallel (DP) replicas using the shared scheduler. See parallel serving.
Performance Comparison
Installation
See the PhyAI installation guide for the latest source and nightly package instructions.
From source:
Nightly build:
Quantization
PhyAI supports post-training quantization (PTQ) when loading an unquantized checkpoint. One kernel policy YAML controls which layers are quantized, their weight and activation precision, and the execution kernels. Humming is included in the standard installation.
For example, run PI0.5 with INT4 weights and BF16 activations (W4A16, NVIDIA SM80+) from the repository root, with your PhyAI Python environment activated:
This preset quantizes the language model and action expert’s attention/MLP linear projections with group size 128. Vision and input/output heads keep their original precision. Point
--checkpointto your original, unquantized weights.Choose another PI0.5 preset, or copy one and edit its
quantizationand kernelrules. See the quantization guide for matching layers, changing precision, and selecting kernels.Multi-host serving
You can run model servers on separate machines and connect them through
phyai-gateway. On each model host, preparepi05.yamlwith that host’s checkpoint, tokenizer, and robot settings. You can start from the PI0.5 example, which servespi05on port50063. Check the configuration, then start the server:On the gateway host, register both servers at startup. Replace the IP addresses below with your server hosts;
pi05must matchserver.model_namein their YAML files. Repeat--backend MODEL HOST:PORTto add more servers:In another terminal on the gateway host, check that the backends report
healthy: true:Clients use the gateway host’s HTTP port
30000or gRPC port50111. The gateway distributes stateless requests across healthy replicas. See the server and gateway setup guide for client configuration and adding or removing servers while the gateway is running.Contribution Guidelines
We thank the contributors below and welcome more developers to join us in building PhyAI.
Sponsors & Adoption
PhyAI is a latency-first, open-source serving engine for Physical AI. It is being adopted by companies working across AI infrastructure and robotics, including Mingti, ModelBest and RLInf.
We are actively seeking partnerships with compute providers, chip vendors, and robotics companies. If you are interested in working with us, please contact us.
Citation
If you use PhyAI in research or production work, please cite the paper:
Contact:
We welcome PhD and master’s students who want to help build Physical AGI, especially those interested in systems infrastructure. We also want to work with chip and compute companies, as well as robotics companies that plan to deploy models with PhyAI.
License
PhyAI is released under the MIT License. It uses FlashInfer, Humming, and FLA. We have also learned a great deal from SGLang, vLLM, and TokenSpeed. We thank the maintainers and contributors of all these projects.
Demos
Cosmos3-Nano-Policy-DROID, 260718 nightly version
https://github.com/user-attachments/assets/12a833ce-3b47-4f08-875c-b30cc2567bef
https://github.com/user-attachments/assets/8e7d433c-e3b1-4388-8488-196836a1ba51
https://github.com/user-attachments/assets/29067c88-f51a-4ca6-ab8c-b45f4d778923