stress-tests-using-pytorch