JAX
https://gyazo.com/7decfe2111d18dc682c62e441d8dc254
https://github.com/google/jax
JAX は CPU、GPU、TPU 上の
NumPy
であり、高性能機械学習研究のための優れた自動微分機能を備えています。