Add optional JIT to ELBO calculation

Closes #428

See merge request !1011

Model.__init__: Harden against very early attribute references

Closes #427

See merge request !1010

Tweak ELBO arguments and fix #eigenvector handling

See merge request !971

Resolves #427 .

_static_newton_cg: Fix maxiter==0 handling

Closes #426

See merge request !1009

Make CF amplitude models proper NIFTy.re.Models

Closes #418

See merge request !997

Remove jax version constraint as recent version works again

See merge request !976

Jakob Roth authored 3 weeks ago and Gordian Edenhofer committed 3 weeks ago

See: #426

optimze: replace .dot with vdot

Closes #422 and #421

See merge request !1006

OptimizeKlConfig fixups

See merge request !1008

This breaks without the previous patch

Multi gpu

See merge request !993

Previously, it was not possible to overwrite/ set "output directory" and
"save strategy" in OptimizeKLConfig.optimize_kl. It needed to be set
already from the beginning in the constructor.

Now, one can set these two options also in OptimizeKLConfig.optimize_kl
just like all other options.

LikelihoodSum: Join/reduce summands in multi-likelihood sum

Closes #423

See merge request !1007