jax
6156decd - [indexing] support pallas-like dynamic slicing in jax.numpy indexing

Commit

1 day ago

[indexing] support pallas-like dynamic slicing in jax.numpy indexing

References

#34598 - [ROCm] Implement approx_tanh for ROCm using OCML tanh function

#34600 - [ROCm] Enable scaled_matmul tests on ROCm

#34601 - [ROCm] Enable neural network tests on ROCm

#34602 - [ROCm] Fix and enable Pallas ops tests on ROCm

#34603 - [ROCm] Enable miscellaneous tests on ROCm

#34689 - [ROCm] Fix ROCm GPU architecture detection and route to Triton backend

#34734 - [indexing] support pallas-like dynamic slicing in jax.numpy indexing

#34822 - [ROCm] Added back compatibility test for hipsolver_sytrd

#34829 - [ROCm] Add ROCm LU solver to backward compatibility tests

#34832 - [ROCm] Add ROCm support to annotate_device_placement backward compatibility …

#34862 - [ROCm] Added back compatibility test for hipsolver_geqrf

Author

jakevdp

jakevdp

Committer

jakevdp

jakevdp

Parents

Loading