examples/jax/encoder/test_model_parallel_encoder.py · b6020e3bce7e0a22c6bf988ddee578943c60821f · 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_model_parallel_encoder.py 22.9 KB