問題タブ [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.
python - vmap(jax)を使用して要素ごとに行列を合計しますか?
vmap の in_axes および out_axes オプションを理解しようとしています。たとえば、2 つの行列を合計して、同じ形状の出力を取得したいとします。
XとYにそれぞれ軸0と1の両方をマッピングしたと思います。出力は、X、Y と同じ形状になります。しかし、私はエラーが発生します、
python - JAXはjitの下の配列のスライスにのみ関数を適用します
私は JAX を使用しており、次のような操作を実行したい
これは では実行できませんjit。またはでこれを行う方法はありますjax.opsかjax.lax？使おうと思ったのですが、同じ問題を繰り返さずjax.ops.index_update(x, idx, y)に計算する方法が見つかりません。y
python - JAXでVJPを使用しているときに前方評価を無効にする方法はありますか?
私は自分のプロジェクトで VJP を頻繁に使用しています。ヤコビアン計算の対象となる関数を実行し、呼び出し可能な vjp 関数とともに primals_out を返します。たとえば、JAX ドキュメントのカスタム VJP 定義は次のようになります。
この例では、VJP を使用する場合に forward 関数の評価が必要であることがわかります。これは、カスタム定義の VJP の代わりに通常の VJP を使用する場合にも当てはまります。ただし、関数の評価コストが高く、その関数をコードのどこかで既に実行しているため、VJP にその関数をもう一度評価させたくありません。
では、VJP を計算するときに関数が評価されないことを示す方法はありますか?
python - scipy.optimize.minimize の NonlinearConstraint でヘシアンを指定する際の問題
で独自の Hessian を指定するのに問題がありscipy.optimize.minimizeます。ダブルチェックするために最小限の問題を作成しましたが、それを機能させることもできません。誰かがたまたま問題が何であるかを知っていますか?
これが私の例です：