Support for Optuna training

1) Limitation to ONE device. Python multithreading is stupid and essentially restarts the whole script to run in parallel. This is not compatible with having hyperparameters chosen for each run
2) Adding an optional callback to allow Optuna to prune (abort) training processes which will not lead to good results
