Draft: JaxOpt style optimization for jifty VI

Philipp Frankrequested to merge
jaxopt_vi into NIFTy_8

Implements a JaxOpt style optimization for jifty VI inference: OptimizeVI Additionally reimplements optimize_kl using it.

Merge request reports