refactor

15 jobs for optimize_kl_output in 18 minutes and 21 seconds (queued for 3 seconds)