Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
requirements.txt 56 Bytes
Newer Older
Patrick von Platen's avatar
[Flax MLM] Refactor run mlm with optax (#11745)  
Patrick von Platen committed
1
2
3
4 
datasets >= 1.1.3
jax>=0.2.8
jaxlib>=0.1.59
flax>=0.3.4