TODO: - Add a sweep config that supports fancy HPO algorithms like Hyperband correctly, with Hydra and the Orion sweeper. - Might be relevant: https://orion.readthedocs.io/en/stable/auto_tutorials/code_2_hyperband_checkpoint.html