Implementation of Forward Laplacian algorithm in JAX - View it on GitHub