PyTorchチームは1月31日、PyTorch® 2.2のリリースを発表した。新機能や最適化が盛りだくさんのこのバージョンは、エンジニアたちの期待を十分に満たすリリースだ。 PyTorch 2.2では、FlashAttention-v2の統合により、SDPA(_scaled_dot_product_attention_)のパフォーマンスが 約2倍向上 した。これにより、A100 GPUsで最大FLOPs/sの約50-73%まで到達する高速なSDPAが実現された。 新しい機能としては、非Pythonサーバーサイド向けのAOTInductorも導入。これにより、PyTorchプログラムのコンパイルとデプロイメントが可能になり、非Python環境での利用が向上した。 その他にも、Optimizersのサポート向上や新しいログメカニズムTORCH_LOGSの追加など、多くの改良が行われた。一方で
![PyTorch® 2.2リリース ー SDPA性能2倍、AOTInductorによる非Python環境への展開が可能に](https://cdn-ak-scissors.b.st-hatena.com/image/square/1581350b2a61822f06713374f02688f7366a740d/height=288;version=1;width=512/https%3A%2F%2Fres.cloudinary.com%2Ftechfeed%2Fimage%2Ffetch%2Fw_1200%2Ch_630%2Cc_fill%2Fhttps%253A%252F%252Fpytorch.org%252Fassets%252Fimages%252Fsocial-share.jpg)