AMM: speed up and stabilize test_memory#5737

crusaderky merged 5 commits intodask:maindask/distributed:mainfrom

crusaderky:AMM/test_memorycrusaderky/distributed:AMM/test_memoryCopy head branch name to clipboard