Add support for Jax v.0.9.0#3717

Merged

JeniferC99 merged 8 commits into

mainfrom

users/jechrist/jax0.9.0

Mar 9, 2026

Commits on Feb 25, 2026

Change JAX version to 0.9.0 (rocm-jaxlib-v0.9.0)

Jenkins
and
cursoragent
committed

Commits on Feb 27, 2026

Conditionally add --jax-source-dir only for jax_ref 0.8.2
Jenkins
committed
add jax path for 0.9.0 too.
Jenkins
committed

Commits on Mar 3, 2026

Commits on Mar 6, 2026

Merge branch 'main' into users/jechrist/jax0.9.0
JeniferC99
authored

Commits on Mar 9, 2026