問題タブ [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.
installation - 特定の JAX jaxlib GPU バージョンをインストールできない
jaxlibCUDA および cuDNN バージョンで動作する特定のバージョンをインストールしようとしています。READMEに従って、私は試しています
pip install --upgrade jax jaxlib==0.1.52+cuda101 -f https://storage.googleapis.com/jax-releases/jax_releases.html
これにより、次のエラーが返されます。
ERROR: Requested jaxlib==0.1.52+cuda101 from https://storage.googleapis.com/jax-releases/cuda101/jaxlib-0.1.52%2Bcuda101-cp37-none-manylinux2010_x86_64.whl has different version in metadata: '0.1.52'
これの原因やエラーを回避する方法を知っている人はいますか?
jax - JIT で JAX reshape を処理する方法
hereで説明されているように、entmax-alpha を実装しようとしています。
これがコードです。
次のコードで呼び出すと：
次のエラーが発生しました。
常にリシェイプ操作に接続されているようです。なぜこれが起こるのかわかりません。助けていただければ幸いです。
問題を再現するために、ここにcolabノートがあります
どうもありがとう。
python - Jax が静的引数を見つけられない
これはこの質問に関連しています。奇妙なことの1つを除いて、私はなんとかコードの作業を最大限に活用しています。
これが変更されたコードです。
このコードは、次のようなエラーを生成します。
これは、このコード行が原因です
関数本体をエンティティ関数だけに置き換えても、エラーは解決しません。これは本当に奇妙な振る舞いです。ただし、ループを展開するのに役立つため、これを静的にすることは非常に重要です。