mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
4ce6bcc310
* Adds Flax BERT finetuning example * fix traced jax tensor type * Use Optax losses and learning schedulers * Add 1GPU training results * merge into master & make style * fix input * del file * Fix bug in loss and add torch runs * finish bert flax fine-tune * Update examples/flax/text-classification/README.md * Update examples/flax/text-classification/run_flax_glue.py * add requirements * finalize * finalize Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> Co-authored-by: Patrick von Platen <patrick@huggingface.co>
6 lines
125 B
Plaintext
6 lines
125 B
Plaintext
datasets >= 1.1.3
|
|
jax>=0.2.8
|
|
jaxlib>=0.1.59
|
|
git+https://github.com/google/flax.git
|
|
git+https://github.com/deepmind/optax.git
|