flax
Support direct quantization for FP8 matmul
#3922
Merged

Loading