> JAX は CPU、GPU、TPU 上の NumPy であり、高性能機械学習研究のための優れた自動微分機能を備えています。