JAXとPyTorch、どっちが速いのか検証してみた

JAXとPyTorch、どっちが速いのか検証してみた - まったり勉強ノート
https://www.mattari-benkyo-note.com/2021/11/17/ssw-jax-vs-torch/

last-modified: 2021-11-18 09:26:39