jax
1a33f0ea - Fix v_pages quantization and adjust test params for ROCm compatibilit… (#560)

Commit

91 days ago

Fix v_pages quantization and adjust test params for ROCm compatibilit… (#560)

References

#601 - porting changes from 0.8.0 to 0.8.2

#632 - Add script for running bazel tests on ROCm

#638 - Support wheels as build deps

#710 - fix poor performance of slogdet when input dimension n=1 or 2

#733 - Add ROCm gesdd SVD backend, export back-compat data, and relaxed GPU …

#742 - Fix deprecated mpmath.ctx_mp.mpnumeric import for mpmath >= 1.4.0

#750 - Backport approx_tanh fix (cherry-pick jax-ml/jax#34598)

Author

phambinhfin

phambinhfin

Committer

Ruturaj4

Ruturaj4

Parents

Loading