Term

JAX

別名: JAX

Overview

JAXは、Googleが開発した数値計算ライブラリで、AutogradとXLA(Accelerated Linear Algebra)を組み合わせることで、高性能な機械学習プログラムの実行を可能にします。NumPyに近いAPIを持ちながら、GPUやTPU上での高速な演算や自動微分をサポートしています。

Mentioned Articles

6 件

External Mentions

10 件