examples/jax/mnist/test_single_gpu_mnist.py · ce2f9fa4632a688d45efc55586be18cd0931ea50 · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[JAX] HuggingFace login in JAX examples if token is available (#2290) · ce2f9fa4
jberchtold-nvidia authored Oct 21, 2025
```
HF login in JAX examples
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
```
ce2f9fa4

test_single_gpu_mnist.py 11.9 KB