Issues

• MI355X (gfx950): scatter-add wrong and nondeterministic with --xla_gpu_enable_scatter_determinism_expander / --xla_gpu_deterministic_ops

  ROCm/jax#896 · Filter by author jbelofamd opened on Oct 21, 2026

• MI355X (gfx950): stable 3-operand lax.sort with in-jit iota returns duplicated iota entries (default flags)

  ROCm/jax#895 · Filter by author jbelofamd opened on Oct 21, 2026

• Unusual memory usage on gfx1151

  bug
  ROCm/jax#871 · Filter by author jrhip opened on Aug 27, 2026

• [gfx950] JAX GPU init silently hangs (>5min) unless AMD_COMGR_NAMESPACE=1

  ROCm/jax#800 · Filter by author ZJLi2013 opened on Jun 13, 2026

• PGLE/FDO profiling broken on all ROCm GPUs — needs XLA pin bump for rocprofiler-sdk v3

  ROCm/jax#791 · Filter by author srinivamd opened on Jun 3, 2026

• rocm-jaxlib-v0.9.2 build failure: stale xla_gpu_cublaslt_algorithm3.patch after XLA pin bump

  ROCm/jax#777 · Filter by author srinivamd opened on May 15, 2026

• Batched Cholesky (potrf) OOM crash due to hipSolver allocating outside XLA memory pool

  bugstatus: triage
  ROCm/jax#718 · Filter by author FlemingH opened on Feb 26, 2026
  2

  

• Convolutions with Float64: Unsupported DNN data type: tf.float64

  bugUnder Investigation
  ROCm/jax#352 · Filter by author PhilipVinc opened on Apr 9, 2025
  3

• Failed build: CI - with Numpy/Scipy nightly wheels (nightly)

  build failed
  ROCm/jax#244 · Filter by author github-actions opened on Feb 27, 2025
  48