Hamilton-Jacobi reachability analysis in JAX. - View it on GitHub