Clean PyTorch implementation of BatchEnsemble [1] Conv1D, Conv2D and Linear layers.
[1] Y. Wen, D. Tran, and J. Ba. Batchensemble: an alternative approach to efficient ensemble and lifelong learning. In International Conference on Learning Representations, 2020