Supplementary Material to Improving Inference for Neural Compression S1 Stochastic Annealing