Additional tweaks: https://gitlab.mpcdf.mpg.de/ift/nifty/-/compare/geometric_kl_pfrank...geomtric_kl_merge