Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
requirements.txt 69 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
Suraj Patil's avatar
[examples/flax] add adafactor optimizer (#12544)  
Suraj Patil committed
5 
optax>=0.0.9