自动微分共1篇
排序
用 JAX 梯度检查点控制激活保存与重计算-未完纪

用 JAX 梯度检查点控制激活保存与重计算

从残差观察到命名策略、主机卸载、自定义重计算与 scan 层边界，理解内存与计算的交换。
技术译文
彩雨伞的头像-未完纪钻石会员彩雨伞6小时前
0530