tests/jax/test_custom_call_compute.py · dfeef1a26ba48ccbd690567a19137b2af8aeb7c9 · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[JAX] Address tolerance check for current scaling dact dbias (#2211) · dfeef1a2
jberchtold-nvidia authored Sep 29, 2025
```
Address tolerance check for current scaling dact
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
```
dfeef1a2

test_custom_call_compute.py 57.3 KB