arXiv · AI· Allex Kan, Ji Zhang, Hao Yin, Hao Li, Owen Fu, Xunxing Chen, Lin Huang, Zion Wang, Lucas Li, Jiajun Cheng, Arthur Jiang, Junping Zhao, Jia Zhang·· 8 小时前AI 评分29
Tessera：面向千万原子规模分布式等变注意力分子动力学的运行时
Ab Initio Accuracy Beyond the Cutoff: Distributed Message-Passing Equivariant Attention at Tens-of-Millions-of-Atom Scale
AI 导读
Tessera 是一种分布式运行时，通过固定工作负载形状实现千万原子规模的高效分子动力学模拟。 该框架将原子沿空间填充曲线排序至等量分片，使每层每步的张量形状保持一致；配合保守邻域皮层、融合 kernel、三级邻域搜索与 All-to-All 通信重叠，支持 CUDA Graph 重用。
来源：arXiv · AI · arxiv.org