跳到正文
arXiv cs.LG· Alex Guerra, Joshua Aduol, Yaqian Tang, Daniel Williams, Ryan P. Adams, Peter Orbanz·· 7 小时前AI 评分27

无需坐标系的自动微分:一种基于 JAX 的几何微分通用算子

Coordinate-Free Automatic Differentiation

AI 导读

研究者提出一种与基无关的单一导数算子,可表示多种几何微分运算,并能完全用现有自动微分原语表达。该变换可与 JAX 的批处理、编译和硬件加速组合,将几何作为代数数据引入,并在不同流形算子、时空逆问题流程及转子子群优化中验证了复用性。

来源:arXiv cs.LG · arxiv.org