Variational Information Bottleneck - View it on GitHub