A nonparametric variational information bottleneck (NVIB) layer in Pytorch - View it on GitHub