Hugging Face推出AWS Trainium兼容性查询工具

发生了什么

据Hugging Face博客，其推出了一款基于torch-neuronx的模型支持空间（Space），旨在帮助用户查询Transformers模型在AWS Trainium硬件上的兼容性。该工具提供模型运行结果、所需核心数及编译脚本，并支持通过公共数据集进行直接查询。目前测试范围涵盖部分文本生成及图文生成模型，且仅限在4个Trainium2设备上可运行的模型。