tests/jax/test_layer.py · cc42a5771431ad29abb21c6a8215928a964717e7 · OpenDAS / TransformerEngine

[JAX] Make test_layer.py tolerances stricter (#2306) · cc42a577

jberchtold-nvidia authored Dec 02, 2025



* wip
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

* wip
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

* revert change to utils.py
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

---------
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

cc42a577

test_layer.py 19.9 KB

Replace test_layer.py