JAX - work4ai

JAX

https://gyazo.com/7decfe2111d18dc682c62e441d8dc254

https://github.com/google/jax

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