高性能アレイコンピューティングフレームワークJAXのドキュメント。