Use Ray to improve speed

[nlsschim]

We should look into using Ray: https://www.ray.io/ to speed up model training and hyperparameter tuning. Similar to dask but not as complex, if I understand it correctly.