Corrected bugs in the BNN and moved relevant code from torchbnn to remove its dependency. Split log_sigma per layer, so that one can do proper empirical Bayes.