Just to note that this was done by our intern for our AI platform, its a nice little library that automates a badly documented training process (Python).