examples/jax/encoder/test_multigpu_encoder.py · 9df0c4a347a14ce4d028a16684697a0b38d11a8f · OpenDAS / TransformerEngine

[JAX] Use "nyu-mll/glue" instead of "glue" for encoder datasets to fix 404 error (#2625) · 2104e4c1

jberchtold-nvidia authored Jan 27, 2026



* Use "nyu-mll/glue" instead of "glue" for encoder datasets to fix 404 error
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

* rename mnist dataset path
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

* add dataset manifest
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

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

2104e4c1

test_multigpu_encoder.py 19.7 KB

Replace test_multigpu_encoder.py