Speed up energy gradient tests

10 jobs for speedup_tests in 13 minutes and 14 seconds