用一套可读的 Einstein 记号统一 PyTorch、JAX、TensorFlow 等框架的张量操作,让 reshape、转置、rearrange 这类代码更清晰、更不容易出错。
…ats 模式,维度提升时先以 pack1 完成 gather,再转换为目标 packing,减少重复张量操作的 CPU fallback。(@losewayy) - LayerNorm、RMSNorm 新增 subgroup 归约,在一次统计 dispatch 中完成均值/方差或平方和计算,减少多轮中间归约;支持 pack1/pack4、