tests/jax/test_custom_call_compute.py · 8a7ab3ddc17e275fcbcd2eee8688ada265efbcad · OpenDAS / TransformerEngine

jberchtold-nvidia authored Oct 09, 2025


Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
Co-authored-by: Phuong Nguyen <phuonguyen@nvidia.com>

8a7ab3dd

test_custom_call_compute.py 75.3 KB