名为 asdex 的新工具包已发布,适用于 JAX 生态系统,提供了首个独立的自动稀疏微分 (ASD) 实现。该方法旨在高效计算雅可比矩阵或海森矩阵,这对于科学计算和机器学习任务至关重要。ASD 通过利用这些矩阵的稀疏性来优化导数计算,减少所需的自动微分次数,特别是对于输出仅依赖于少数输入的函数。 AI
影响 能够更高效地计算机器学习模型的导数,可能加速训练和研究。
排序理由 该集群描述了一个用于自动稀疏微分的新软件工具包,该工具包在 arXiv 论文中有详细介绍。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →