jax
numpy>=2.0
scipy>=1.17.0rc1
crazyflow>=0.4.0
munch
flax
array-api-compat
array-api-extra
