JAX 数组模式 API 库 einopx 发布

A_K_Nain · x · 2026-09-02

einopx 是一个统一的 JAX 原生数组模式 API。它将形状检查、分片检查、reshape、rearrange、reshard 和 einsum 收缩整合到一个操作中,解决了单一逻辑操作分散在多个不兼容 API 中的问题。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →