jax
jax copied to clipboard
[ROCm] Add float8_e4m3fnuz and float8_e5m2fnuz support for Rocm
@jakevdp, @superbobry, @hawkinsp Could you please take a look?