examples/jax/encoder/test_single_gpu_encoder.py · 2104e4c1b3dcc7249e9e4de9252bf4fa59a03030 · OpenDAS / TransformerEngine

"vscode:/vscode.git/clone" did not exist on "e3e79e9e8a2224e03a711c3d1ef7a35daa447083"

[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_single_gpu_encoder.py 13.5 KB

Replace test_single_gpu_encoder.py