jax
3b39f574 - Skip test_batch_axis_sharding_jvp13 test on 0.8.2 (#707)

Commit

59 days ago

Skip test_batch_axis_sharding_jvp13 test on 0.8.2 (#707)

References

#707 - Skip test_batch_axis_sharding_jvp13 test on 0.8.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

AratiGanesh

AratiGanesh

Parents

Loading