NAM Trainer to be used on Google Colab - View it on GitHub