JAX

Identity

• JAXは、Google発のオープンソース数値計算ライブラリで、自動微分と高速化（コンパイル）を特徴とする。検証待ち JAX — Documentation

Capability

• JAXは、Google発のオープンソース数値計算ライブラリで、自動微分と高速化（コンパイル）を特徴とする。検証待ち JAX — Documentation
• JAXはNumPy風のAPIと自動微分・XLAによるコンパイルを組み合わせ、研究用途で高速な数値計算を可能にする点が特徴である。検証待ち JAX — Documentation

Credibility

公開Evidence未整備

Use Case

公開Evidence未整備

Constraints / Current Status

• JAXはNumPy風のAPIと自動微分・XLAによるコンパイルを組み合わせ、研究用途で高速な数値計算を可能にする点が特徴である。検証待ち JAX — Documentation

Key References

• JAXとは何ですか？
• JAXはPyTorchやTensorFlowと何が違いますか？
• JAXはどんな場面で使いますか？

Part of

• Google— JAXはGoogle発のオープンソース数値計算ライブラリである。

Related

• Flax— FlaxはJAX上に構築されたニューラルネットワークライブラリであり、JAXと直接関連する。
• Optax— OptaxはJAX向けの勾配処理・最適化ライブラリであり、JAXと直接関連する。