JAXにおける高速掃引法によるEikonal方程式の解法
2025-05-11
このブログ投稿では、JAXで実装された高速掃引法を用いて、界面発展や画像処理において重要なEikonal方程式を解く方法を探ります。レベルセット法とEikonal方程式自体の説明から始まり、高速掃引アルゴリズムの詳細、グリッド設定、反復更新、Godunovアップウインドスキームなどを解説します。NumPyとJAXによるコード実装とベンチマークを提供し、JAXの大きな速度優位性を示します。また、アルゴリズムの並列化の試みとその課題についても議論します。