dm-haiku>=0.0.3
gin-config>=0.4.0
jax>=0.2.8
jaxlib>=0.1.57
matplotlib>=3.3.3
networkx
pandas
optax>=0.0.2
tensorflow>=2.4.0
tensorflow-datasets>=4.2.0
