examples/flax/language-modeling/run_t5_mlm_flax.py · e6f221c8d4829c9a3bca699c18a32043ab21f7a0 · chenpangpang / transformers · GitLab

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

Find file Blame History Permalink

[JAX] Replace all jax.tree_* calls with jax.tree_util.tree_* (#18361) · e6f221c8
Sanchit Gandhi authored Sep 09, 2022
```
* [JAX] Replace all jax.tree_* calls with jax.tree_util.tree_*

* fix double tree_util
```
e6f221c8

run_t5_mlm_flax.py 43.1 KB