Skip to content

Draft: JaxOpt style optimization for jifty VI

Philipp Frank requested 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