pytest-xdist
numpy==1.21.5
scipy==1.7.3
autograd==1.3
absl-py==0.15.0
dm-tree==0.1.6
dm-haiku==0.0.5
jax==0.2.28
jaxlib==0.1.74
tensorflow==2.7.0
torch==1.7.1
