Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
requirements.txt 69 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
Avital Oliver's avatar
Use a released version of optax rather than installing from Git. (#12173)  
Avital Oliver committed
5 
optax>=0.0.8