jax
f7f021e5 - Produces better error message on tree mismatch, before we would dump potentially huge subtrees into the log which would often get truncated and would be missing needed information (which keys actually mismatch), now we only include the keys.

Commit

161 days ago

Produces better error message on tree mismatch, before we would dump potentially huge subtrees into the log which would often get truncated and would be missing needed information (which keys actually mismatch), now we only include the keys. PiperOrigin-RevId: 796443223

References

#31073 - Produces better error message on tree mismatch, before we would dump potentially huge subtrees into the log which would often get truncated and would be missing needed information (which keys actually mismatch), now we only include the keys.

#31381 - Remove old ROCm build code

#31720 - Fix ann_test.py numerical bug in target reshape

#31768 - [ROCm] Support lowering through PJRT_Triton_Extension

#32115 - Relax version requirements for ROCm Jax Plugin wheels

#33157 - [ROCm] Resolve undefined behavior in bitshift unit test

#579 - Create rocm-test-requirements.txt

#581 - Fix/pallas tests shared memory

#584 - Use plain bazel to test jax, use hermetic rocm dependency

#585 - update a test for checking zero ROCm GPU event

#34135 - [ROCm] update to test if there are GPU events when doing profiling on…