This looks really cool. Any chance you'll support pretraining runs as well?