Draft: JaxOpt style optimization for jifty VI
Compare changes
Beginning with 2nd June, only the "Single Sign On" option for login to the GitLab web interface will be possible. If you don't have an MPCDF wide second factor so far, please get one at our SelfService (https://selfservice.mpcdf.mpg.de). The GitLab internal second factor will not work.
Implements a JaxOpt style optimization for jifty VI inference: OptimizeVI
Additionally reimplements optimize_kl using it.