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