公开知识图谱记录
TorchNeuron
记录摘要
PyTorch 原生 Trainium 后端，支持 eager execution、torch.compile 和分布式 API。
来自项目知识图谱的有来源实体；具体工具属性仍需补齐。
已记录信息
- 基准测试: 开箱支持最常用的前 100 种文本生成模型架构中的 89%
- 能力: 支持 eager execution
- 能力: 支持 torch.compile
- 能力: 支持张量并行和 FSDP 分布式训练
- 能力: 为 AWS Trainium 提供 PyTorch 原生后端，支持 eager mode、torch.compile、FSDP 和 TP
- 能力: 支持 eager execution、torch.compile 和标准 PyTorch API
- 能力: 集成 NKI 以支持自定义算子和性能优化
- 能力: 支持 eager execution、torch.compile 和标准 PyTorch API，代码改动最少
资料覆盖
知识图谱当前记录了 12 个已知资料维度，另有 7 个维度尚未记录，关联 3 篇公开资料。这些数字只描述资料覆盖，不代表模型性能或适配性已经验证。
相关公开记录
- Diffusers — 支持
- TRL — 支持
- Transformers — 支持
- Transformers — 采用
- Accelerate — 支持