JaxOpt style VI minimization and optimize_kl for nifty.re

Merged Philipp Frank requested to merge jft_minisanity into NIFTy_8

Merge request reports