新しい機能 実験ハイジャックス API (hijax-custom-derivatives) でカスタム派生規則を定義するドキュメントが追加され、VJPHiPrimitive autodiff ルールをjvpまたはlin ルールから導き出すための jax.experimental.hijax ヘルパーも追加されました: apply_derived_linearizationで linearize_from_jvp, transpose_jvpで vjp_fwd_from_jvp, transpose_linearizedで vjp_fwd_from_lin, jvp_from_lin.Jax.custom_r を追加しました。
新しい機能 ヒルバート行列の閉式逆の jax.scipy.linalg.invhilbert が追加されました ({jax-issue}#10144).パスカルの行列の逆 (#10144) のために jax.scipy.linalg.invpascal を追加した。多項式 (#10144) のペンタダイアゴナル フィードラー伴行列を構成するための jax.scipy.linalg.fiedler_companion が追加されました。Jax.ShapeDtypeStruct.likeを追加しました。 Jax.ShapeDを構成するためのショートカットです。
新しい機能は、TensorFlow の AREA サイズ変更 (#20098) に一致する jax.image.resize に ResizeMethod.AREA を追加しました。ハダマード行列 (#10144) を構築するための jax.scipy.linalg.hadamard が追加されました。循環行列 (#10144) を構築するために jax.scipy.linalg.circulantを追加しました。離散なフーリア変換行列 (#10144) を構築するための jax.scipy.linalg.dft が追加されました。Jax.scipy.linalg.leslie を追加しました
新しい機能:PyTorchの2立方体サイズ (#15768) に一致するように jax.image.resize に ResizeMethod.CUBIC_PYTORCH を追加しました。広い行列と full_matrices が True のときの jax.lax.linalg.qr の微分をサポートしますLAPACK 操作は CPU のバッチ次元に沿って並行されます。jax.lax.linalg.tridiagonal_solveに perturb_singular引数を追加しました
JAX 0.9.2 (2026年3月2日) 変更:半プライベートタイプ jax._src.literals.TypedNdArrayは、ダックタイプではなく、np.ndarrayのサブクラスになっています。step が指定されている jax.numpy.arange はもはやホスト上の配列を生成しません利点は、より効率的なコードであるが、狭い幅の浮遊物 (例えばb浮気16).このケースで以前の行動を回復するには。..
変更: Array タイプではない JAX トレイサー (例えば、ref タイプ) は、もはや Array のインスタンスのように報告されません。jax.shard_map を Explicit モード で使用すると、入力 の PartitionSpec が in_specs で指定された PartitionSpec と一致しない場合、エラーが発生します。つまり 暗黙の反論ではなく 断言のように振る舞うのですin_specs はオプションの引数なので、特定の。..
JAX v0.9.0.1は、次の4つのPRからのコミットがパッチされたv0.9.0と同一です: openxla/xla#36579 openxla/xla#36345 openxla/xla#36755 openxla/xla#36696
JAX v0.8.3 は、次の2つのバグ修正を patch した v0.8.2 と同じです: openxla/xla@4bc723d openxla/xla@21552fd
新しい機能: jax.thread_guard が追加され、多コントローラー JAX で複数のスレッドでデバイスが使用されているときを検出するコンテキストマネージャーが追加されました。バグ修正: use_magma=True と pivoting=True を使用するときに MAGMA 2.9.0 でピホットされた QR (magma_zgeqp3_gpu) のワークスペースサイズ計算のエラーが修正されました。#34145 さん削除:フラグ jax_collectives_common_channel_id は削除されました。Jax_pmap_no_rank_reductionの設定は。..
退治は終了しました。Jax.lax.pcast (jax.lax.pcast... to='variating') を代替として使用してください。jax.numpy.arange に渡された複雑な引数は、出力が定義が悪いため、現在、退廃警告を生む。jax.core から、いくつかのシンボルが新しく廃止されています。 例えば: call_impl, get_aval, mapped_aval, subjaxprs, set_current_trace, take_current_trace, traverse_jaxpr_par...
新しい機能: jax.jit は現在、デコレーターファクトリーパターンをサポートしています。 @functools.partial ((jax.jit, static_argnames=['n']) def f ((x, n) の代わりに: ...@jax.jit ((static_argnames=['n']) def f ((x, n) に書くことができます: ...変更: jax.lax.linalg.eigh は、 QR (CPU/GPU), Jacobi (GPU/TPU), QDWH (TPU) の実装を選択するための実装引数を受け入れています。8つの実装リストは。..