JAX
https://gyazo.com/759f461eaaee8693dce7953997ea50ec
自動微分
に特化した、
GPU
や
TPU
に対応した高速な
NumPy
Autograd
と
XLA
の総称
Google製
github
https://qiita.com/koshian2/items/44a871386576b4f80aff
https://zenn.dev/jellied_unagi/articles/70dbb7ce206c13