jax

d8179cdf - Enable test_variadic_reduce_window on ROCm (#647)