FID computation in Flax

[matthias-wright]

Hi all,

I recently ported the awesome PyTorch FID implementation by mseitzer to Flax.
The FID scores are almost identical to the original implementation (absolute difference around 1e-7).

For details, see here: https://github.com/matthias-wright/jax-fid

Cheers!