20x less peak RAM in the new PyTorch memory budget solver