tests/jax/test_custom_call_compute.py · e30c36a30883c49b820096a0bb856c7ea71bebd5 · 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