Frechet inception distance (FID) evaluation in JAX - View it on GitHub