JAX

JAX はオープンソースの Python 数値計算ライブラリで、自動微分、JIT コンパイル、GPU/TPU による高速化、NumPy 風のインターフェースに対応しています。機械学習の研究、科学計算、最適化問題、数値シミュレーション、高性能計算の実験に向いています。

適用場面：科学計算、自動微分、機械学習、最適化、高性能計算

注意：JAX の学習曲線は通常の NumPy より険しく、Python と数値計算の基礎があるユーザーに向いています。