mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 10:56:06 +00:00

A recent XLA change means that certain matrix-vector products may now be lowered as matrix-matrix multiplications. This may mean that we use lower precisions where we previously did not. PiperOrigin-RevId: 633949879