numpy
scipy
treegp>=1.2.0
jax
