This repository implements and studies conditional variance regularization (CoRe) by C. Heinze-Deml and N. Meinshausen (2017).
Original paper and code: https://arxiv.org/abs/1710.11469 and https://github.com/christinaheinze/core/tree/master
General Flax training procedure inspired from MNIST classification example in https://github.com/google/flax