Release Prep JAX 0.9.1 #735
gulsumgudukbay
wants to merge
4 commits into
rocm-jaxlib-v0.9.1
from
release-prep-091
Implement approx_tanh for ROCm using OCML tanh function
be3564d7
Address review comments for approx_tanh ROCm implementation
6f87beb2
Skip test_topology_jit_serialize on ROCm GPU
7fa83ec1
Update tests/aot_test.py
d2c6ccaa
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub