transformer_engine/jax/cpp_extensions/gemm.py · 355c4e42a7cfd137ea2bca32cee7dea750f73d1c · OpenDAS / TransformerEngine

Phuong Nguyen authored May 28, 2025



* fp8 gemm with direct quant
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

---------
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

355c4e42

gemm.py 17.7 KB