arXiv cs.LG· Adrian Hill, Guillaume Dalle·· 3 小时前AI 评分25
asdex：JAX 中的自动稀疏微分工具包
asdex: Automatic Sparse Differentiation in JAX
AI 导读
asdex 发布了 JAX 生态中首个独立自动稀疏微分（ASD）工具包，为 jax.jacfwd 和 jax.jacrev 提供稀疏替代方案。ASD 通过稀疏模式检测、图着色、压缩微分和解压四步，将 AD 传递次数从 n 或 m 降至与问题维度无关的颜色数，如 b 个连续带的带状 Jacobian 仅需 b 次。
来源：arXiv cs.LG · arxiv.org