Commits · c1a1c04e6341778562a45cff847dcaface8b33cc · OpenDAS / TransformerEngine

13 Nov, 2025 1 commit

[JAX] Shardy rule + QuantizeLayout Rework (#2364) · 9440b76a

Phuong Nguyen authored Nov 13, 2025



* shardy + quantize_layout rework
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

* add assertion for NVFP4 in fused act and fused norm primitive
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

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

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

9440b76a