04 May 2025 jax, simulation This is going to be a fun blog - we'll explore the intuition behind level sets, the Eikonal equation, and implement a speedy algorithm for solving this equation, called the fast sweeping method, in JAX. The level set method and the eikonal equation Fast sweeping method Code numpy jax In action Benchmarks A note on parallel FSM References Fig 1: FSM in action I was recently researching a problem that involved interface evolution over time. Our interface was represented...