transformer_engine/jax/quantize/quantizer.py · ce2e8bd12edfe10647bec8f54fedc394d6287b58 · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[JAX] Add assertion message to amax -> scale computation (#2263) · 76e1af33
jberchtold-nvidia authored Oct 13, 2025
```
assertion check
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
```
76e1af33

quantizer.py 44.7 KB