jax>=0.4.20
numpy>=1.18.5
flax>=0.7.5
pandas>=1.1.5
