# ubuntu seems to have issues with jax v0.4.6. Recheck when v0.4.7 is out
jax
jaxlib
grgrjax>=0.4.3
scipy
pyyaml
