# ubuntu seems to have issues with jax v0.4.6. Recheck when v0.4.7 is out
jax>=0.3.20,<0.4.6
jaxlib>=0.3.20,<0.4.6
grgrjax>=0.4.1
scipy
pyyaml
