pytest-benchmark~=3.4.0
dm-tree>=0.1.6
tianshou>=0.4.5
jax[cpu]>=0.2.17