PyTorch 2.1: automatic dynamic shape compilation, distributed checkpointing | Dark Hacker News