Speed up einsum

10 jobs for speedup_tests in 9 minutes and 22 seconds (queued for 1 second)