dm-haiku>=0.0.3
gin-config>=0.4.0
jax>=0.2.8
jaxlib==0.1.59+cuda110 -f https://storage.googleapis.com/jax-releases/jax_releases.html
matplotlib>=3.3.3
networkx
pandas
optax>=0.0.2
tensorflow>=2.4.0
tensorflow-datasets>=4.2.0
