jax<0.4.6
jaxlib<0.4.6
grgrjax>=0.4.3
pyyaml
scipy

[linear]
grgrlib>=0.1.22
