Hi, Thanks for sharing the great package! I am curious what would it take to write a simple python wrapper to train and evaluate the models. Thanks