https://github.com/pytorch/PiPPy/blob/ab6c5c9e4135c88126dc293926d5b3880e32e1c8/README.md?plain=1#L331