Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
requirements.txt 98 Bytes
Newer Older
Marc van Zee's avatar
Adds Flax BERT finetuning example on GLUE (#11564)  
Marc van Zee committed
1
2
3 
datasets >= 1.1.3
jax>=0.2.8
jaxlib>=0.1.59
Patrick von Platen's avatar
[Flax MLM] Refactor run mlm with optax (#11745)  
Patrick von Platen committed
4 
flax>=0.3.4
Marc van Zee's avatar
Adds Flax BERT finetuning example on GLUE (#11564)  
Marc van Zee committed
5 
git+https://github.com/deepmind/optax.git