JAX 中的快速扫描法求解 Eikonal 方程

本文介绍了使用 JAX 实现快速扫描法求解 Eikonal 方程，该方程在界面演化和图像处理等领域应用广泛。文章首先解释了水平集方法和 Eikonal 方程的概念，然后详细阐述了快速扫描法的算法原理，包括网格设置、迭代更新以及 Godunov 上风差分格式的使用。最后，作者提供了基于 NumPy 和 JAX 的代码实现，并进行了性能比较，结果显示 JAX 版本显著提升了计算效率。此外，文章还探讨了并行化快速扫描法的尝试及遇到的挑战。