6 ms·PyTorch 2.1: automatic dynamic shape compilation, distributed checkpointing2 points by DreamFlasher 3y ago