問題タブ [jax]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
163 問題
0
投票する
4
に答える
10076
参照
python - jaxlib をインストールできません
ドキュメントで見つけた次のコマンドを使用して、Windows 10 に jaxlib をインストールしようとしています。
pip インストール jaxlib
次のエラーが表示されます
誰でも私を助けることができますか、前もって感謝します!
0
投票する
1
に答える
673
参照
python - Jacobian x Jacobian.T を計算する効率的な方法
Jは、いくつかのパラメーターに関する関数のヤコビアンであると仮定fします。(PyTorch またはおそらく Jax で) 2 つの入力 (x1および) を取り、メモリ内の行列全体をインスタンス化せずx2に計算する関数を持つ効率的な方法はありますか?J(x1)*J(x2).transpose() J
私は何かに出くわしましjvp(f, input, v=vjp(f, input))たが、それをよく理解していません。私が欲しいものかどうかはわかりません。