Code to plot mutual information The code is written in Pytorch with GPU enabled. The code will generate estimated mutual information for input features.