Implement constants for minimization (and optimize_kl)
Implement constants for NIFTy.re, i.e. parts of the pytree that are potentially sampled but NOT optimized.
Implement constants for NIFTy.re, i.e. parts of the pytree that are potentially sampled but NOT optimized.