etils
flax>=0.6.3
jax>=0.3.16
matplotlib
numpy>=1.18.0
tensorflow
