jax
dbc860fe - Cherry-pick fixes for JAX 0.9.2 release from JAX 0.9.1 (#740)

Commit

107 days ago

Cherry-pick fixes for JAX 0.9.2 release from JAX 0.9.1 (#740) Co-authored-by: Arati Ganesh <arati.ganesh@amd.com> Co-authored-by: shurale-nkn <shurale.nkn@gmail.com>

References

#740 - Cherry-pick fixes for JAX 0.9.2 release from JAX 0.9.1

#790 - ci: deselect PGLE/FDO profiling tests on ROCm (ROCM-25581) [v0.9.2]

#798 - [ROCm] Deselect crashing FusedAttention bwd8 + add pytest-timeout (ROCM-25626)

Author

magaonka-amd

Parents

a659757d

jax dbc860fe - Cherry-pick fixes for JAX 0.9.2 release from JAX 0.9.1 (#740)

jax
dbc860fe - Cherry-pick fixes for JAX 0.9.2 release from JAX 0.9.1 (#740)