PyTorch 2.0 теперь stable! Это один из самых мощных апдейтов торча, невероятные улучшения скорости и производительности, один torch.compile, чего стоит, так например я смог ускорить обучение модельки с 40 секунд до 7! Очень рекомендую потестить, но будьте аккуратны, обратной совместимости со старыми версиями - нет. Больше о всех изменениях: https://pytorch.org/get-started/pytorch-2.0/