Résolution de l'équation d'Eikonal avec la méthode de balayage rapide dans JAX

Cet article de blog explore la résolution de l'équation d'Eikonal, cruciale dans l'évolution des interfaces et le traitement d'images, en utilisant la méthode de balayage rapide implémentée dans JAX. Il commence par expliquer les méthodes d'ensembles de niveaux et l'équation d'Eikonal elle-même. Le cœur de l'article détaille l'algorithme de balayage rapide, couvrant la configuration de la grille, les mises à jour itératives et le schéma de différence amont de Godunov. Des implémentations de code NumPy et JAX sont fournies, avec des benchmarks démontrant l'avantage de vitesse significatif de JAX. L'auteur discute également des tentatives de parallélisation de l'algorithme et des défis rencontrés.