File size: 85 Bytes
ee6e328
 
 
 
 
 
1
2
3
4
5
6
7
datasets >= 1.1.3
jax>=0.2.8
jaxlib>=0.1.59
flax>=0.3.5
optax>=0.0.8
evaluate>=0.2.0