GPU向けODE集合解法ライブラリ GRADSOLVE をJAXで公開

arXiv cs.LG ・ 2026-09-02

原題: GRADSOLVE: fast exact gradients for ODE ensembles on GPUs

AI による要約

GPU向けにODE集合解法とその逆モード微分を高速に行うオープンソースライブラリ GRADSOLVE をJAXで公開した。従来は解法と微分のいずれかを犠牲にしていたが、固定ステップ再生で正確な離散随伴微分を得る手法で両立させた。

この要約は当サイトの AI が生成したものです。正確な内容は元記事（arXiv cs.LG）をご確認ください。