arXiv · Machine Learning· Adrian Hill, Guillaume Dalle·2026-10-12 12:00· 8 小时前AI 评分50

asdex：JAX 中的自动稀疏微分

asdex: Automatic Sparse Differentiation in JAX

AI 导读

asdex 为 JAX 生态提供首个独立的自动稀疏微分工具包，用于高效计算函数的稀疏 Jacobian 或 Hessian。工具通过检测稀疏模式、图着色、压缩微分和解压缩减少 AD passes；带有 b 个连续带的带状 Jacobian 无论规模多大都只需 b 个颜色。